Merge pull request #212 from superannotateai/328_315

fix bug

Author: VaghinakDev

requirements.txt

@@ -7,7 +7,7 @@ pillow>=7.2.0
 numpy>=1.19.0
 matplotlib>=3.3.1
 xmltodict==0.12.0
-opencv-python==4.4.0.42
+opencv-python>=4.4.0.42
 wheel==0.35.1
 packaging>=20.4
 pandas>=1.1.4

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

@@ -2173,10 +2173,12 @@ def download_export(
                 total=use_case.get_upload_files_count(), desc="Uploading"
             ) as progress_bar:
                 for _ in use_case.execute():
-                    progress_bar.update(1)
+                    progress_bar.update()
+            progress_bar.close()
         else:
             for _ in use_case.execute():
                 continue
+        logger.info(use_case.response.data)
 
 
 @Trackable

src/superannotate/lib/core/usecases.py

@@ -2508,12 +2508,6 @@ def __init__(
         self._project = project
         self._return_metadata = return_metadata
 
-    def validate_project_type(self):
-        if self._project.project_type in constances.LIMITED_FUNCTIONS:
-            raise AppValidationException(
-                constances.LIMITED_FUNCTIONS[self._project.project_type]
-            )
-
     def execute(self):
         if self.is_valid():
             data = self._service.get_exports(
@@ -3999,7 +3993,7 @@ def execute(self):
         return self._response
 
 
-class DownloadExportUseCase(BaseUseCase):
+class DownloadExportUseCase(BaseInteractiveUseCase):
     def __init__(
         self,
         service: SuerannotateServiceProvider,
@@ -4090,20 +4084,24 @@ def get_upload_files_count(self):
 
     def execute(self):
         if self.is_valid():
+            report = []
             if self._to_s3_bucket:
                 self.get_upload_files_count()
                 yield from self.upload_to_s3_from_folder(self._temp_dir.name)
-                logger.info(f"Exported to AWS {self._to_s3_bucket}/{self._folder_path}")
+                report.append(
+                    f"Exported to AWS {self._to_s3_bucket}/{self._folder_path}"
+                )
                 self._temp_dir.cleanup()
             else:
                 export_id, filepath, destination = self.download_to_local_storage(
                     self._folder_path
                 )
                 if self._extract_zip_contents:
-                    logger.info(f"Extracted {filepath} to folder {destination}")
+                    report.append(f"Extracted {filepath} to folder {destination}")
                 else:
-                    logger.info(f"Downloaded export ID {export_id} to {filepath}")
+                    report.append(f"Downloaded export ID {export_id} to {filepath}")
                 yield
+            self._response.data = "\n".join(report)
         return self._response
 
 

tests/integration/test_attach_video_urls.py

@@ -37,3 +37,11 @@ def test_attach_video_urls_without_name_column(self):
         self.assertEqual(len(uploaded), 7)
         self.assertEqual(len(could_not_upload), 0)
         self.assertEqual(len(existing_images), 1)
+
+    def test_get_exports(self):
+        sa.attach_video_urls_to_project(
+            self.PROJECT_NAME,
+            self.csv_path_without_name_column
+        )
+        sa.prepare_export(self.PROJECT_NAME)
+        self.assertEqual(len(sa.get_exports(self.PROJECT_NAME)),1)