Skip to content

Commit 83eadf2

Browse files
author
Val Brodsky
committed
Reimplement client.upload_data using client.execute
1 parent 9eb0311 commit 83eadf2

File tree

1 file changed

+1
-36
lines changed

1 file changed

+1
-36
lines changed

libs/labelbox/src/labelbox/client.py

Lines changed: 1 addition & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)