Fix tests

Author: dshabin

src/superannotate/lib/app/analytics/class_analytics.py

@@ -3,8 +3,8 @@
 
 import pandas as pd
 import plotly.express as px
-from superannotate.lib.app.mixp.decorators import Trackable
 from superannotate.lib.app.exceptions import AppException
+from superannotate.lib.app.mixp.decorators import Trackable
 
 from .common import aggregate_annotations_as_df
 
@@ -26,7 +26,11 @@ def class_distribution(export_root, project_names, visualize=False):
     """
 
     json_paths = list(Path(str(export_root)).glob("*.json"))
-    if json_paths and "___pixel.json" not in json_paths[0].name and "___objects.json" not in json_paths[0].name:
+    if (
+        json_paths
+        and "___pixel.json" not in json_paths[0].name
+        and "___objects.json" not in json_paths[0].name
+    ):
         raise AppException(
             "The function does not support projects containing videos attached with URLs"
         )

src/superannotate/lib/app/analytics/common.py

@@ -173,7 +173,11 @@ def aggregate_annotations_as_df(
     """
 
     json_paths = list(Path(str(project_root)).glob("*.json"))
-    if json_paths and "___pixel.json" not in json_paths[0].name and "___objects.json" not in json_paths[0].name:
+    if (
+        json_paths
+        and "___pixel.json" not in json_paths[0].name
+        and "___objects.json" not in json_paths[0].name
+    ):
         raise AppException(
             "The function does not support projects containing videos attached with URLs"
         )

src/superannotate/lib/app/interface/sdk_interface.py

@@ -977,7 +977,7 @@ def get_image_metadata(project, image_name, *_, **__):
         raise AppValidationException(response.errors)
 
     res_data = response.data
-    res_data.data["annotation_status"] = constances.AnnotationStatus.get_name(
+    res_data["annotation_status"] = constances.AnnotationStatus.get_name(
         res_data["annotation_status"]
     )
     return res_data
@@ -2389,7 +2389,48 @@ def attach_video_urls_to_project(project, attachments, annotation_status="NotSta
     :return: attached videos, failed videos, skipped videos
     :rtype: (list, list, list)
     """
-    return attach_image_urls_to_project(project, attachments, annotation_status)
+    project_name, folder_name = extract_project_folder(project)
+    project = controller.get_project_metadata(project_name).data
+    if project["project"].project_type != constances.ProjectType.VIDEO.value:
+        raise AppValidationException("The function does not support")
+
+    image_data = pd.read_csv(attachments, dtype=str)
+    image_data = image_data[~image_data["url"].isnull()]
+    if "name" in image_data.columns:
+        image_data["name"] = (
+            image_data["name"]
+            .fillna("")
+            .apply(lambda cell: cell if str(cell).strip() else str(uuid.uuid4()))
+        )
+    else:
+        image_data["name"] = [str(uuid.uuid4()) for _ in range(len(image_data.index))]
+
+    image_data = pd.DataFrame(image_data, columns=["name", "url"])
+    img_names_urls = image_data.rename(columns={"url": "path"}).to_dict(
+        orient="records"
+    )
+    list_of_not_uploaded = []
+    duplicate_images = []
+    for i in range(0, len(img_names_urls), 500):
+        response = controller.attach_urls(
+            project_name=project_name,
+            folder_name=folder_name,
+            files=ImageSerializer.deserialize(
+                img_names_urls[i : i + 500]  # noqa: E203
+            ),
+            annotation_status=annotation_status,
+        )
+        if response.errors:
+            list_of_not_uploaded.append(response.data[0])
+            duplicate_images.append(response.data[1])
+
+    list_of_uploaded = [
+        image["name"]
+        for image in img_names_urls
+        if image["name"] not in list_of_not_uploaded
+    ]
+
+    return list_of_uploaded, list_of_not_uploaded, duplicate_images
 
 
 @Trackable

src/superannotate/lib/core/usecases.py

@@ -381,11 +381,11 @@ def __init__(
         self._annotation_status = annotation_status
         self._image_name_prefix = image_name_prefix
 
-    def validate_project_type(self):
-        if self._project.project_type == constances.ProjectType.VIDEO.value:
-            raise AppValidationException(
-                "The function does not support projects containing videos attached with URLs"
-            )
+    # def validate_project_type(self):
+    #     if self._project.project_type == constances.ProjectType.VIDEO.value:
+    #         raise AppValidationException(
+    #             "The function does not support projects containing videos attached with URLs"
+    #         )
 
     def execute(self):
         if self.is_valid():
@@ -1847,7 +1847,9 @@ def settings_use_case(self):
     @property
     def work_flow_use_case(self):
         return GetWorkflowsUseCase(
-            workflows=self._workflows, annotation_classes=self._annotation_classes,
+            project=self._project,
+            workflows=self._workflows,
+            annotation_classes=self._annotation_classes,
         )
 
     def execute(self):

tests/integration/test_attach_image_urls.py

@@ -12,7 +12,7 @@ class TestImageUrls(BaseTestCase):
     PROJECT_TYPE = "Vector"
 
     def test_attach_image_urls(self):
-        uploaded, could_not_upload, existing_images = sa.attach_video_urls_to_project(
+        uploaded, could_not_upload, existing_images = sa.attach_image_urls_to_project(
             self.PROJECT_NAME,
             os.path.join(dirname(dirname(__file__)), self.PATH_TO_URLS),
         )

tests/integration/test_attach_video_urls.py

@@ -20,8 +20,7 @@ def test_attach_video_urls(self):
         self.assertEqual(len(uploaded), 8)
         self.assertEqual(len(could_not_upload), 0)
         self.assertEqual(len(existing_images), 0)
-        images = sa.search_images(project=self.PROJECT_NAME, return_metadata=True)
-        self.assertTrue(all([image["name"] for image in images]))
+
 
     def test_attach_video_urls_without_name_column(self):
         uploaded, could_not_upload, existing_images = sa.attach_video_urls_to_project(
@@ -31,5 +30,3 @@ def test_attach_video_urls_without_name_column(self):
         self.assertEqual(len(uploaded), 8)
         self.assertEqual(len(could_not_upload), 0)
         self.assertEqual(len(existing_images), 0)
-        images = sa.search_images(project=self.PROJECT_NAME, return_metadata=True)
-        self.assertTrue(all([image["name"] for image in images]))

tests/integration/test_filter_instances.py

@@ -7,7 +7,7 @@
 from tests.integration.base import BaseTestCase
 
 
-class TestFolders(BaseTestCase):
+class TestFilterInstances(BaseTestCase):
     PROJECT_NAME = "test filter instances"
     PROJECT_TYPE = "Vector"
     TEST_FOLDER_PATH = "data_set/sample_project_vector"