Merge pull request #698 from superannotateai/FRIDAY-2895

Added export csv format

Author: nareksa

pytest.ini

@@ -3,5 +3,5 @@ minversion = 3.7
 log_cli=true
 python_files = test_*.py
 ;pytest_plugins = ['pytest_profiling']
-addopts = -n auto --dist=loadscope
+;addopts = -n auto --dist=loadscope
 

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

@@ -1187,8 +1187,10 @@ def prepare_export(
         :param only_pinned: enable only pinned output in export. This option disables all other types of output.
         :type only_pinned: bool
 
-        :param kwargs: Arbitrary kwarg ``integration_name``
-            can be provided which will be used as a storage to store export file
+        :param kwargs:
+            Arbitrary kwargs::
+                 * integration_name: can be provided which will be used as a storage to store export file
+                 * export_type: can be CSV for the Gen AI projects
 
         :return: metadata object of the prepared export
         :rtype: dict
@@ -1216,13 +1218,18 @@ def prepare_export(
                     break
             else:
                 raise AppException("Integration not found.")
+        _export_type = None
+        export_type = kwargs.get("export_type")
+        if export_type and export_type == "csv":
+            _export_type = 3
         response = self.controller.prepare_export(
             project_name=project_name,
             folder_names=folders,
             include_fuse=include_fuse,
             only_pinned=only_pinned,
             annotation_statuses=annotation_statuses,
             integration_id=integration_id,
+            export_type=_export_type,
         )
         if response.errors:
             raise AppException(response.errors)

src/superannotate/lib/core/serviceproviders.py

@@ -571,6 +571,7 @@ def prepare_export(
         include_fuse: bool,
         only_pinned: bool,
         integration_id: int,
+        export_type: int = None,
     ) -> ServiceResponse:
         raise NotImplementedError
 

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

@@ -46,6 +46,7 @@ def __init__(
         only_pinned: bool,
         annotation_statuses: List[str] = None,
         integration_id: int = None,
+        export_type: int = None,
     ):
         super().__init__(),
         self._project = project
@@ -55,6 +56,7 @@ def __init__(
         self._include_fuse = include_fuse
         self._only_pinned = only_pinned
         self._integration_id = integration_id
+        self._export_type = export_type
 
     def validate_only_pinned(self):
         if (
@@ -102,15 +104,17 @@ def execute(self):
                     constances.AnnotationStatus.NOT_STARTED.name,
                     constances.AnnotationStatus.SKIPPED.name,
                 ]
-
-            response = self._service_provider.prepare_export(
+            kwargs = dict(
                 project=self._project,
                 folders=self._folder_names,
                 annotation_statuses=self._annotation_statuses,
                 include_fuse=self._include_fuse,
                 only_pinned=self._only_pinned,
                 integration_id=self._integration_id,
             )
+            if self._export_type:
+                kwargs["export_type"] = self._export_type
+            response = self._service_provider.prepare_export(**kwargs)
             if not response.ok:
                 raise AppException(response.error)
 

src/superannotate/lib/infrastructure/controller.py

@@ -1013,6 +1013,7 @@ def prepare_export(
         only_pinned: bool,
         annotation_statuses: List[str] = None,
         integration_id: int = None,
+        export_type: int = None,
     ):
         project = self.get_project(project_name)
         use_case = usecases.PrepareExportUseCase(
@@ -1023,6 +1024,7 @@ def prepare_export(
             only_pinned=only_pinned,
             annotation_statuses=annotation_statuses,
             integration_id=integration_id,
+            export_type=export_type,
         )
         return use_case.execute()
 

src/superannotate/lib/infrastructure/serviceprovider.py

@@ -152,6 +152,7 @@ def prepare_export(
         include_fuse: bool,
         only_pinned: bool,
         integration_id: int = None,
+        export_type: int = None,
     ):
         annotation_statuses = ",".join(
             [str(constants.AnnotationStatus.get_value(i)) for i in annotation_statuses]
@@ -164,6 +165,8 @@ def prepare_export(
             "coco": 0,
             "time": datetime.datetime.now().strftime("%b %d %Y %H:%M"),
         }
+        if export_type:
+            data["export_format"] = export_type
         if folders:
             data["folder_names"] = folders
         if integration_id is not None: