Skip to content

Commit 9eb0311

Browse files
author
Val Brodsky
committed
Reimplement upload_members using client.execute
1 parent 67f91fd commit 9eb0311

File tree

1 file changed

+2
-36
lines changed

1 file changed

+2
-36
lines changed

libs/labelbox/src/labelbox/schema/user_group_v2.py

Lines changed: 2 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,7 @@
44
from io import BytesIO
55
from typing import List, Optional
66

7-
import requests
87
from lbox.exceptions import (
9-
InternalServerError,
108
LabelboxError,
119
ResourceNotFoundError,
1210
)
@@ -151,42 +149,10 @@ def upload_members(
151149
), # Remove the unnecessary (None, ...) tuple
152150
}
153151

154-
client = self.client
155-
headers = dict(client.connection.headers)
156-
headers.pop("Content-Type", None)
157-
request = requests.Request(
158-
"POST",
159-
client.endpoint,
160-
headers=headers,
161-
data=request_data,
162-
files=files,
163-
)
164-
165-
prepped: requests.PreparedRequest = request.prepare()
166-
167-
response = client.connection.send(prepped)
168-
169-
if response.status_code == 502:
170-
error_502 = "502 Bad Gateway"
171-
raise InternalServerError(error_502)
172-
elif response.status_code == 503:
173-
raise InternalServerError(response.text)
174-
elif response.status_code == 520:
175-
raise InternalServerError(response.text)
176-
177-
try:
178-
file_data = response.json().get("data", None)
179-
except ValueError as e: # response is not valid JSON
180-
raise LabelboxError("Failed to upload, unknown cause", e)
152+
file_data = self.client.execute(data=request_data, files=files)
181153

182154
if not file_data or not file_data.get("importUsersAsCsvToGroup", None):
183-
try:
184-
errors = response.json().get("errors", [])
185-
error_msg = "Unknown error"
186-
if errors:
187-
error_msg = errors[0].get("message", "Unknown error")
188-
except Exception:
189-
error_msg = "Unknown error"
155+
error_msg = "Unknown error"
190156
raise LabelboxError("Failed to upload, message: %s" % error_msg)
191157

192158
csv_report = file_data["importUsersAsCsvToGroup"]["csvReport"]

0 commit comments

Comments
 (0)