1
1
import json
2
+ import time
3
+ from datetime import datetime
2
4
3
5
from roboflow .adapters import deploymentapi
4
6
from roboflow .config import load_roboflow_api_key
5
- import time
6
- from datetime import datetime
7
7
8
8
9
9
def add_deployment_parser (subparsers ):
@@ -26,7 +26,8 @@ def add_deployment_parser(subparsers):
26
26
deployment_add_parser .set_defaults (func = add_deployment )
27
27
deployment_add_parser .add_argument ("-a" , "--api_key" , help = "api key" )
28
28
deployment_add_parser .add_argument (
29
- "deployment_name" , help = "deployment name, must contain 5-15 lowercase characters, first character must be a letter"
29
+ "deployment_name" ,
30
+ help = "deployment name, must contain 5-15 lowercase characters, first character must be a letter" ,
30
31
)
31
32
# deployment_add_parser.add_argument(
32
33
# "-s", "--security_level", help="security level (protected)", default="protected"
@@ -35,21 +36,31 @@ def add_deployment_parser(subparsers):
35
36
"-m" , "--machine_type" , help = "machine type, run `roboflow deployment machine_type` to see available options"
36
37
)
37
38
deployment_add_parser .add_argument (
38
- "-t" , "--duration" , help = "duration, how long you want to keep the deployment (unit: hour, default: 3)" ,
39
- type = float , default = 3
39
+ "-t" ,
40
+ "--duration" ,
41
+ help = "duration, how long you want to keep the deployment (unit: hour, default: 3)" ,
42
+ type = float ,
43
+ default = 3 ,
40
44
)
41
45
deployment_add_parser .add_argument (
42
46
"-e" , "--no_delete_on_expiration" , help = "keep when expired (default: False)" , action = "store_true"
43
47
)
44
48
deployment_add_parser .add_argument (
45
- "-v" , "--inference_version" , help = "inference server version (default: latest)" , default = "latest" ,
49
+ "-v" ,
50
+ "--inference_version" ,
51
+ help = "inference server version (default: latest)" ,
52
+ default = "latest" ,
53
+ )
54
+ deployment_add_parser .add_argument (
55
+ "-w" , "--wait_on_pending" , help = "wait if deployment is pending" , action = "store_true"
46
56
)
47
- deployment_add_parser .add_argument ("-w" , "--wait_on_pending" , help = "wait if deployment is pending" , action = "store_true" )
48
57
49
58
deployment_get_parser .set_defaults (func = get_deployment )
50
59
deployment_get_parser .add_argument ("-a" , "--api_key" , help = "api key" )
51
60
deployment_get_parser .add_argument ("deployment_name" , help = "deployment name" )
52
- deployment_get_parser .add_argument ("-w" , "--wait_on_pending" , help = "wait if deployment is pending" , action = "store_true" )
61
+ deployment_get_parser .add_argument (
62
+ "-w" , "--wait_on_pending" , help = "wait if deployment is pending" , action = "store_true"
63
+ )
53
64
54
65
deployment_list_parser .set_defaults (func = list_deployment )
55
66
deployment_list_parser .add_argument ("-a" , "--api_key" , help = "api key" )
@@ -85,7 +96,7 @@ def add_deployment(args):
85
96
args .deployment_name ,
86
97
args .inference_version ,
87
98
)
88
-
99
+
89
100
if status_code != 200 :
90
101
print (f"{ status_code } : { msg } " )
91
102
return
@@ -107,12 +118,14 @@ def get_deployment(args):
107
118
if status_code != 200 :
108
119
print (f"{ status_code } : { msg } " )
109
120
return
110
-
121
+
111
122
if (not args .wait_on_pending ) or msg ["status" ] != "pending" :
112
123
print (json .dumps (msg , indent = 2 ))
113
124
break
114
-
115
- print (f'{ datetime .now ().strftime ("%Y-%m-%d %H:%M:%S" )} Waiting for deployment { args .deployment_name } to be ready...\n ' )
125
+
126
+ print (
127
+ f'{ datetime .now ().strftime ("%Y-%m-%d %H:%M:%S" )} Waiting for deployment { args .deployment_name } to be ready...\n '
128
+ )
116
129
time .sleep (30 )
117
130
118
131
@@ -138,4 +151,3 @@ def delete_deployment(args):
138
151
print (f"{ status_code } : { msg } " )
139
152
return
140
153
print (json .dumps (msg , indent = 2 ))
141
-
0 commit comments