Fix folders download

Author: VaghinakDev

src/superannotate/lib/core/usecases/annotations.py

@@ -1443,8 +1443,6 @@ def execute(self):
                 folders = self._service_provider.folders.list(
                     Condition("project_id", self._project.id, EQ)
                 ).data
-                folders.append(self._folder)
-
             if not folders:
                 folders.append(self._folder)
             with concurrent.futures.ThreadPoolExecutor(max_workers=5) as executor:

src/superannotate/lib/core/usecases/projects.py

@@ -254,7 +254,7 @@ def execute(self):
                     ] = self._service_provider.annotation_classes.create_multiple(
                         entity, [annotation_class]
                     )
-                self._response.data.classes = self._project.classes
+                data["classes"] = self._project.classes
             if self._project.workflows:
                 set_workflow_use_case = SetWorkflowUseCase(
                     service_provider=self._service_provider,

tests/integration/annotations/test_download_annotations.py

@@ -1,5 +1,6 @@
 import json
 import os
+import glob
 import tempfile
 from pathlib import Path
 
@@ -70,4 +71,28 @@ def test_download_empty_annotations_from_folders(self):
         )
         with tempfile.TemporaryDirectory() as temp_dir:
             annotations_path = sa.download_annotations(f"{self.PROJECT_NAME}", temp_dir)
-            self.assertEqual(len(os.listdir(annotations_path)), 1)
+            self.assertEqual(len(os.listdir(annotations_path)), 1)
+
+    @pytest.mark.flaky(reruns=3)
+    def test_download_annotations_from_folders(self):
+        sa.create_folder(self.PROJECT_NAME, self.FOLDER_NAME)
+        sa.create_folder(self.PROJECT_NAME, self.FOLDER_NAME_2)
+        sa.create_annotation_classes_from_classes_json(
+            self.PROJECT_NAME, f"{self.folder_path}/classes/classes.json"
+        )
+        sa.attach_items(
+            f'{self.PROJECT_NAME}/{self.FOLDER_NAME}',
+            [{"name": f"example_image_{i}.jpg", "url": f"url_{i}"} for i in range(1, 5)]  # noqa
+        )
+        sa.attach_items(
+            self.PROJECT_NAME,
+            [{"name": f"example_image_{i}.jpg", "url": f"url_{i}"} for i in range(1, 19)]  # noqa
+        )
+        sa.attach_items(
+            f'{self.PROJECT_NAME}/{self.FOLDER_NAME_2}',
+            [{"name": f"example_image_{i}.jpg", "url": f"url_{i}"} for i in range(1, 10)]  # noqa
+        )
+        with tempfile.TemporaryDirectory() as temp_dir:
+            annotations_path = sa.download_annotations(self.PROJECT_NAME, temp_dir, recursive=True)
+            count = len([i for i in glob.iglob(annotations_path + '**/**', recursive=True)])
+            assert count == 31 + 5  # folder names and classes