From 7649db54558e9dcba60fdd6e80a29824eb01aa5c Mon Sep 17 00:00:00 2001 From: Roshanlal Date: Fri, 7 Oct 2022 23:53:11 +0530 Subject: [PATCH] Update LabelStudioClient.py Refactored nested IF --- src/yolov7_reid/src/LabelStudioClient.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/src/yolov7_reid/src/LabelStudioClient.py b/src/yolov7_reid/src/LabelStudioClient.py index eed41d4..c346150 100644 --- a/src/yolov7_reid/src/LabelStudioClient.py +++ b/src/yolov7_reid/src/LabelStudioClient.py @@ -25,12 +25,12 @@ def upload_file(file_path): file_id = None if file_ids is not None and len(file_ids) > 0: file_id = file_ids[0] - if file_id is not None: + if file_id is not None and response.ok: get_upload_file_url = f"{labelstudio_url}/api/import/file-upload/{file_id}" response = requests.get(get_upload_file_url,headers=auth_header) - if response.ok: - logging.debug(response.json()) - return response.json() + logging.debug(response.json()) + return response.json() + else: logging.error('cant get uploaded info') else: @@ -39,7 +39,7 @@ def upload_file(file_path): @staticmethod def create_task_with_file(file_path): resp = LabelStudioClient.upload_file(file_path) - if resp != None: + if resp != None and response.ok: fileurl = '/data/' + resp['file'] task_json = [{ "data": {"image": fileurl}, @@ -50,10 +50,10 @@ def create_task_with_file(file_path): auth_header = {'Authorization' : f'Token {labelstudio_token}'} task_url = f"{labelstudio_url}/api/projects/{labelstudio_pid}/import" response = requests.post(task_url, json=task_json, headers=auth_header) - if response.ok: - created_task = response.json() - logging.debug(created_task) - return True + created_task = response.json() + logging.debug(created_task) + return True + else: logging.error('Failed to create Task {}'.format(response)) @@ -63,4 +63,4 @@ def create_task_with_file(file_path): # print(resp) resp =LabelStudioClient.create_task_with_file('./img.jpg') - print(resp) \ No newline at end of file + print(resp)