@@ -264,42 +264,7 @@ def upload_data(
264264 if (filename and content_type )
265265 else content
266266 }
267- headers = self .connection .headers .copy ()
268- headers .pop ("Content-Type" , None )
269- request = requests .Request (
270- "POST" ,
271- self .endpoint ,
272- headers = headers ,
273- data = request_data ,
274- files = files ,
275- )
276-
277- prepped : requests .PreparedRequest = request .prepare ()
278-
279- response = self .connection .send (prepped )
280-
281- if response .status_code == 502 :
282- error_502 = "502 Bad Gateway"
283- raise InternalServerError (error_502 )
284- elif response .status_code == 503 :
285- raise InternalServerError (response .text )
286- elif response .status_code == 520 :
287- raise InternalServerError (response .text )
288-
289- try :
290- file_data = response .json ().get ("data" , None )
291- except ValueError as e : # response is not valid JSON
292- raise LabelboxError ("Failed to upload, unknown cause" , e )
293-
294- if not file_data or not file_data .get ("uploadFile" , None ):
295- try :
296- errors = response .json ().get ("errors" , [])
297- error_msg = next (iter (errors ), {}).get (
298- "message" , "Unknown error"
299- )
300- except Exception :
301- error_msg = "Unknown error"
302- raise LabelboxError ("Failed to upload, message: %s" % error_msg )
267+ file_data = self .execute (data = request_data , files = files )
303268
304269 return file_data ["uploadFile" ]["url" ]
305270
0 commit comments