Fix set status

Author: hovnatan

superannotate/db/images.py

@@ -1095,12 +1095,11 @@ def set_images_annotation_statuses(project, image_names, annotation_status):
     project, project_folder = get_project_and_folder_metadata(project)
     params = {"team_id": project["team_id"], "project_id": project["id"]}
     annotation_status = common.annotation_status_str_to_int(annotation_status)
-    data = {
-        "annotation_status": annotation_status,
-        "folder_id": project_folder["id"]
-    }
+    data = {"annotation_status": annotation_status}
+    if project_folder is not None:
+        data["folder_id"] = project_folder["id"]
     if image_names is None:
-        image_names = search_images(project)
+        image_names = search_images((project, project_folder))
     for start_index in range(0, len(image_names), NUM_TO_SEND):
         data["image_names"] = image_names[start_index:start_index + NUM_TO_SEND]
         response = _api.send_request(

superannotate/version.py

@@ -1 +1 @@
-__version__ = "4.1.0b10"
+__version__ = "4.1.0b11"

tests/test_folders.py

@@ -144,7 +144,7 @@ def test_delete_folders(tmpdir):
 
     assert len(sa.search_folders(project)) == 3
 
-    sa.delete_folders(project, "folder1")
+    sa.delete_folders(project, ["folder1"])
     assert len(sa.search_folders(project)) == 2
     sa.delete_folders(project, ["folder2", "folder3"])
     assert len(sa.search_folders(project)) == 0
@@ -494,6 +494,12 @@ def test_folder_image_annotation_status(tmpdir):
         metadata = sa.get_image_metadata(project, image)
         assert metadata["annotation_status"] == "InProgress"
 
+    sa.set_images_annotation_statuses(PROJECT_NAME, None, "QualityCheck")
+
+    for image in sa.search_images(PROJECT_NAME):
+        metadata = sa.get_image_metadata(PROJECT_NAME, image)
+        assert metadata["annotation_status"] == "QualityCheck"
+
 
 def test_folder_misnamed(tmpdir):
     PROJECT_NAME = "test folder misnamed"