add pause resume

Author: PacificDou

roboflow/adapters/deploymentapi.py

@@ -9,10 +9,11 @@ class DeploymentApiError(Exception):
     pass
 
 
-def add_deployment(api_key, machine_type, duration, delete_on_expiration, deployment_name, inference_version):
+def add_deployment(api_key, creator_email, machine_type, duration, delete_on_expiration, deployment_name, inference_version):
     url = f"{DEDICATED_DEPLOYMENT_URL}/add"
     params = {
         "api_key": api_key,
+        "creator_email": creator_email,
         # "security_level": security_level,
         "duration": duration,
         "delete_on_expiration": delete_on_expiration,
@@ -69,6 +70,22 @@ def get_deployment_usage(api_key, deployment_name, from_timestamp, to_timestamp)
     return response.status_code, response.json()
 
 
+def pause_deployment(api_key, deployment_name):
+    url = f"{DEDICATED_DEPLOYMENT_URL}/pause"
+    response = requests.post(url, json={"api_key": api_key, "deployment_name": deployment_name})
+    if response.status_code != 200:
+        return response.status_code, response.text
+    return response.status_code, response.json()
+
+
+def resume_deployment(api_key, deployment_name):
+    url = f"{DEDICATED_DEPLOYMENT_URL}/resume"
+    response = requests.post(url, json={"api_key": api_key, "deployment_name": deployment_name})
+    if response.status_code != 200:
+        return response.status_code, response.text
+    return response.status_code, response.json()
+
+
 def delete_deployment(api_key, deployment_name):
     url = f"{DEDICATED_DEPLOYMENT_URL}/delete"
     response = requests.post(url, json={"api_key": api_key, "deployment_name": deployment_name})

roboflow/deployment.py

@@ -61,6 +61,8 @@ def add_deployment_parser(subparsers):
     deployment_usage_deployment_parser = deployment_subparsers.add_parser(
         "usage_deployment", help="get usage of a specific dedicated deployments"
     )
+    deployment_pause_parser = deployment_subparsers.add_parser("pause", help="pause a dedicated deployment")
+    deployment_resume_parser = deployment_subparsers.add_parser("resume", help="resume a dedicated deployment")
     deployment_delete_parser = deployment_subparsers.add_parser("delete", help="delete a dedicated deployment")
     deployment_log_parser = deployment_subparsers.add_parser("log", help="show log info for a dedicated deployment")
 
@@ -77,7 +79,11 @@ def add_deployment_parser(subparsers):
     #     "-s", "--security_level", help="security level (protected)", default="protected"
     # )
     deployment_add_parser.add_argument(
-        "-m", "--machine_type", help="machine type, run `roboflow deployment machine_type` to see available options"
+        "-m", "--machine_type", help="machine type, run `roboflow deployment machine_type` to see available options",
+        required=True
+    )
+    deployment_add_parser.add_argument(
+        "-e", "--creator_email", help="your email address", required=True
     )
     deployment_add_parser.add_argument(
         "-t",
@@ -87,7 +93,7 @@ def add_deployment_parser(subparsers):
         default=3,
     )
     deployment_add_parser.add_argument(
-        "-e", "--no_delete_on_expiration", help="keep when expired (default: False)", action="store_true"
+        "-nodel", "--no_delete_on_expiration", help="keep when expired (default: False)", action="store_true"
     )
     deployment_add_parser.add_argument(
         "-v",
@@ -127,6 +133,14 @@ def add_deployment_parser(subparsers):
     deployment_usage_deployment_parser.add_argument(
         "-t", "--to_timestamp", help="end time stamp in ISO8601 format (YYYY-MM-DD HH:MM:SS)", default=None
     )
+    
+    deployment_pause_parser.set_defaults(func=pause_deployment)
+    deployment_pause_parser.add_argument("-a", "--api_key", help="api key")
+    deployment_pause_parser.add_argument("deployment_name", help="deployment name")
+    
+    deployment_resume_parser.set_defaults(func=resume_deployment)
+    deployment_resume_parser.add_argument("-a", "--api_key", help="api key")
+    deployment_resume_parser.add_argument("deployment_name", help="deployment name")
 
     deployment_delete_parser.set_defaults(func=delete_deployment)
     deployment_delete_parser.add_argument("-a", "--api_key", help="api key")
@@ -163,6 +177,7 @@ def add_deployment(args):
         exit(1)
     status_code, msg = deploymentapi.add_deployment(
         api_key,
+        args.creator_email,
         # args.security_level,
         args.machine_type,
         args.duration,
@@ -241,6 +256,30 @@ def get_deployment_usage(args):
     print(json.dumps(msg, indent=2))
 
 
+def pause_deployment(args):
+    api_key = args.api_key or load_roboflow_api_key(None)
+    if api_key is None:
+        print("Please provide an api key")
+        exit(1)
+    status_code, msg = deploymentapi.pause_deployment(api_key, args.deployment_name)
+    if status_code != 200:
+        print(f"{status_code}: {msg}")
+        exit(status_code)
+    print(json.dumps(msg, indent=2))
+
+
+def resume_deployment(args):
+    api_key = args.api_key or load_roboflow_api_key(None)
+    if api_key is None:
+        print("Please provide an api key")
+        exit(1)
+    status_code, msg = deploymentapi.resume_deployment(api_key, args.deployment_name)
+    if status_code != 200:
+        print(f"{status_code}: {msg}")
+        exit(status_code)
+    print(json.dumps(msg, indent=2))
+
+
 def delete_deployment(args):
     api_key = args.api_key or load_roboflow_api_key(None)
     if api_key is None: