@@ -93,18 +93,25 @@ def create_jobs(self, jobs):
9393 "jobs" : jobs ,
9494 }
9595
96- r = requests .post ("%s/create_jobs" % self .api_server ,
97- headers = self .get_headers (),
98- json = payload , timeout = 60 , verify = self .verify )
96+ while True :
97+ r = requests .post ("%s/create_jobs" % self .api_server ,
98+ headers = self .get_headers (),
99+ json = payload , timeout = 60 , verify = self .verify )
99100
100- if r .status_code != 200 :
101- msg = r .text
102- try :
103- msg = r .json ()['message' ]
104- except :
105- pass
101+ if r .status_code == 200 :
102+ return
106103
107- raise Failure (msg )
104+ if r .status_code == 400 :
105+ msg = r .text
106+ try :
107+ msg = r .json ()['message' ]
108+ except :
109+ pass
110+
111+ raise Failure (msg )
112+
113+ self .console .collect ('Failed to connect to API, retrying.' , show = True )
114+ time .sleep (3 )
108115
109116 def post_api_server (self , endpoint , data = None ):
110117 while True :
@@ -119,6 +126,7 @@ def post_api_server(self, endpoint, data=None):
119126 except Exception as e :
120127 print e
121128
129+ self .console .collect ('Failed to connect to API, retrying.' , show = True )
122130 time .sleep (1 )
123131
124132 def post_stats (self , stat ):
@@ -165,7 +173,7 @@ def _get_file_from_api_server(self, url, path):
165173 message = None
166174
167175 r = None
168- for _ in xrange (0 , 5 ):
176+ for _ in xrange (0 , 20 ):
169177 try :
170178 message = None
171179 r = requests .get ("%s%s" % (self .api_server , url ),
@@ -176,6 +184,8 @@ def _get_file_from_api_server(self, url, path):
176184 return
177185
178186 if r .status_code != 200 :
187+ self .console .collect ('Failed to download file (%s), retrying' % r .status_code , show = True )
188+ time .sleep (10 )
179189 continue
180190
181191 with open (path , 'wb' ) as f :
@@ -185,6 +195,7 @@ def _get_file_from_api_server(self, url, path):
185195
186196 except Exception as e :
187197 message = str (e )
198+ self .console .collect ('Failed to download file (%s), retrying' % message , show = True )
188199 time .sleep (10 )
189200 continue
190201
@@ -199,7 +210,7 @@ def _get_file_from_api_server(self, url, path):
199210 except :
200211 pass
201212
202- raise Failure ('Failed to download file: %s' % msg )
213+ raise Failure ('Failed to download file(%s) : %s' % ( r . status_code , msg ) )
203214
204215 def post_file_to_api_server (self , url , path , filename = None , split = False ):
205216 if not filename :
0 commit comments