image copy

Author: hovnatan

superannotate/db/project_images.py

@@ -134,9 +134,27 @@ def upload_image_to_project(
             break
 
 
+def copy_image_endpoint(img_metadatas, destination_project_folder_id):
+    params = {
+        "team_id": _api.team_id,
+        "project_id": img_metadatas[0]["project_id"]
+    }
+    json_req = {
+        "image_ids": [x["id"] for x in img_metadatas],
+        "destination_folder_id": destination_project_folder_id
+    }
+    response = _api.send_request(
+        req_type='POST', path='/image/copy', params=params, json_req=json_req
+    )
+    if not response.ok:
+        raise SABaseException(
+            response.status_code, "Couldn't copy images " + response.text
+        )
+
+
 def copy_image(
     source_project,
-    image_name,
+    image_names,
     destination_project,
     include_annotations=False,
     copy_annotation_status=False,
@@ -159,12 +177,22 @@ def copy_image(
     :param copy_pin: enables image pin status copy
     :type copy_pin: bool
     """
-    if not isinstance(source_project, dict):
-        source_project = get_project_metadata_bare(source_project)
-    if not isinstance(destination_project, dict):
-        destination_project = get_project_metadata_bare(destination_project)
-    img_b = get_image_bytes(source_project, image_name)
-    img_metadata = get_image_metadata(source_project, image_name)
+    source_project, source_project_folder = get_project_project_folder_metadata(
+        source_project
+    )
+    destination_project, destination_project_folder = get_project_project_folder_metadata(
+        destination_project
+    )
+    img_metadatas = get_image_metadata(
+        (source_project, source_project_folder), image_names
+    )
+    print(source_project, destination_project)
+    if source_project["id"] == destination_project["id"]:
+        copy_image_endpoint(img_metadatas, destination_project_folder["id"])
+        return
+    img_b = get_image_bytes(
+        (source_project, source_project_folder), image_names
+    )
     new_name = image_name
     extension = Path(image_name).suffix
     p = re.compile(r"_\([0-9]+\)\.")

tests/test_image_copy_move.py

@@ -5,6 +5,7 @@
 PROJECT_NAME_CPY = "test image copy 1"
 PROJECT_NAME_CPY_MULT = "test image copy mult 2"
 PROJECT_NAME_MOVE = "test image move 3"
+PROJECT_NAME_CPY_FOLDER = "test image copy folder 4"
 
 
 def test_image_copy_mult(tmpdir):
@@ -158,3 +159,37 @@ def test_image_move(tmpdir):
 
     si = sa.search_images(project)
     assert len(si) == 1
+
+
+def test_image_copy_folders(tmpdir):
+    tmpdir = Path(tmpdir)
+
+    projects_found = sa.search_projects(
+        PROJECT_NAME_CPY_FOLDER, return_metadata=True
+    )
+    for pr in projects_found:
+        sa.delete_project(pr)
+
+    project = sa.create_project(PROJECT_NAME_CPY_FOLDER, "test", "Vector")
+
+    sa.upload_image_to_project(
+        project,
+        "./tests/sample_project_vector/example_image_1.jpg",
+        annotation_status="InProgress"
+    )
+    sa.upload_image_to_project(
+        project,
+        "./tests/sample_project_vector/example_image_2.jpg",
+        annotation_status="InProgress"
+    )
+
+    sa.create_folder(project, "folder1")
+
+    sa.copy_image(
+        project, ["example_image_1.jpg", "example_image_2.jpg"],
+        project["name"] + "/folder1"
+    )
+    sa.copy_image(
+        project, ["example_image_1.jpg", "example_image_2.jpg"],
+        project["name"] + "/folder1"
+    )