Merge pull request #42 from superannotateai/SAS-3432

Fix image unassign

Author: dshabin

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,

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):