From 17c933dddab2ba578d105a891480c026d06633b7 Mon Sep 17 00:00:00 2001 From: shab Date: Mon, 7 Jun 2021 17:20:32 +0400 Subject: [PATCH] Fix image unassign --- superannotate/db/project_images.py | 5 ++--- tests/test_assign_images.py | 22 ++++++++++++++++++++++ 2 files changed, 24 insertions(+), 3 deletions(-) diff --git a/superannotate/db/project_images.py b/superannotate/db/project_images.py index 593e775d0..6592b6748 100644 --- a/superannotate/db/project_images.py +++ b/superannotate/db/project_images.py @@ -688,14 +688,13 @@ def unassign_images(project, image_names): :param image_names: list of image unassign :type image_names: list of str """ - project_meta = get_project_metadata(project) project, folder = get_project_and_folder_metadata(project) folder_name = 'root' if folder: folder_name = folder['name'] params = { - "project_id": project_meta['id'], - "team_id": project_meta["team_id"] + "project_id": project['id'], + "team_id": project["team_id"] } json_req = { "image_names": image_names, diff --git a/tests/test_assign_images.py b/tests/test_assign_images.py index aa70c0091..af5551c4d 100644 --- a/tests/test_assign_images.py +++ b/tests/test_assign_images.py @@ -147,6 +147,28 @@ def test_unassign_images(tmpdir): assert im1_metadata["qa_id"] == None assert im2_metadata["qa_id"] == None + FOLDER_NAME = "test_folder" + sa.create_folder(project, FOLDER_NAME) + project = PROJECT_NAME_VECTOR1 + "/" + FOLDER_NAME + sa.move_images( + PROJECT_NAME_VECTOR1, ["example_image_1.jpg", "example_image_2.jpg"], project + ) + sa.assign_images( + project, ["example_image_1.jpg", "example_image_2.jpg"], email + ) + time.sleep(1) + sa.unassign_images( + project, + ["example_image_1.jpg", "example_image_2.jpg"], + ) + + sa.search_images(project) + im1_metadata = sa.get_image_metadata(project, "example_image_1.jpg") + + im2_metadata = sa.get_image_metadata(project, "example_image_2.jpg") + + assert im1_metadata["qa_id"] == None + assert im2_metadata["qa_id"] == None def test_assign_folder(tmpdir):