Added setting serializer

Author: Vaghinak Basentsyan

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

@@ -36,6 +36,7 @@
 from lib.app.serializers import BaseSerializers
 from lib.app.serializers import ImageSerializer
 from lib.app.serializers import ProjectSerializer
+from lib.app.serializers import SettingsSerializer
 from lib.app.serializers import TeamSerializer
 from lib.core.enums import ImageQuality
 from lib.core.exceptions import AppException
@@ -856,8 +857,12 @@ def get_project_metadata(
     ).data
 
     metadata = ProjectSerializer(response["project"]).serialize()
+    if response.get("settings"):
+        metadata["settings"] = [
+            SettingsSerializer(setting).serialize() for setting in response["settings"]
+        ]
 
-    for elem in "settings", "classes", "workflows", "contributors":
+    for elem in "classes", "workflows", "contributors":
         if response.get(elem):
             metadata[elem] = [
                 BaseSerializers(attribute).serialize() for attribute in response[elem]

src/superannotate/lib/app/serializers.py

@@ -62,3 +62,11 @@ def deserialize(data):
         if isinstance(data, list):
             return [ImageEntity(**image) for image in data]
         return ImageEntity(**data)
+
+
+class SettingsSerializer(BaseSerializers):
+    def serialize(self):
+        data = super().serialize()
+        if data["attribute"] == "ImageQuality":
+            data["value"] = constance.ImageQuality.get_name(data["value"])
+        return data

src/superannotate/lib/core/__init__.py

@@ -1,6 +1,7 @@
 from pathlib import Path
 
 from superannotate.lib.core.enums import AnnotationStatus
+from superannotate.lib.core.enums import ImageQuality
 from superannotate.lib.core.enums import ProjectType
 from superannotate.lib.core.enums import SegmentationStatus
 from superannotate.lib.core.enums import TrainingStatus
@@ -63,6 +64,7 @@
     TrainingStatus,
     SegmentationStatus,
     TrainingTask,
+    ImageQuality,
     AnnotationStatus,
     CONFIG_FILE_LOCATION,
     BACKEND_URL,

src/superannotate/lib/core/usecases.py

@@ -1377,7 +1377,7 @@ def execute(self):
 
         new_settings_to_update = []
         for new_setting in self._to_update:
-            if new_setting["attribute"] == 'ImageQuality':
+            if new_setting["attribute"] == "ImageQuality":
                 new_setting["value"] = ImageQuality.get_value(new_setting["value"])
             new_settings_to_update.append(
                 {
@@ -4159,7 +4159,9 @@ def s3_repository(self):
         return self._s3_repo_instance
 
     def _upload_image(self, image_path: str):
-        ProcessedImage = namedtuple("ProcessedImage", ["uploaded", "path", "entity", "name"])
+        ProcessedImage = namedtuple(
+            "ProcessedImage", ["uploaded", "path", "entity", "name"]
+        )
         if self._from_s3_bucket:
             image_bytes = (
                 GetS3ImageUseCase(s3_bucket=self._from_s3_bucket, image_path=image_path)
@@ -4180,9 +4182,16 @@ def _upload_image(self, image_path: str):
 
         if not upload_response.errors and upload_response.data:
             entity = upload_response.data
-            return ProcessedImage(uploaded=True, path=entity.path, entity=entity, name=Path(image_path).name)
+            return ProcessedImage(
+                uploaded=True,
+                path=entity.path,
+                entity=entity,
+                name=Path(image_path).name,
+            )
         else:
-            return ProcessedImage(uploaded=False, path=image_path, entity=None, name=Path(image_path).name)
+            return ProcessedImage(
+                uploaded=False, path=image_path, entity=None, name=Path(image_path).name
+            )
 
     @property
     def paths(self):

src/superannotate/lib/infrastructure/services.py

@@ -7,11 +7,10 @@
 from urllib.parse import urljoin
 
 import lib.core as constance
-import requests
+import requests.packages.urllib3
 from lib.core.exceptions import AppException
 from lib.core.serviceproviders import SuerannotateServiceProvider
 from requests.exceptions import HTTPError
-import requests.packages.urllib3
 
 requests.packages.urllib3.disable_warnings()
 

tests/integration/test_interface.py

@@ -18,5 +18,5 @@ def test_get_project_default_image_quality_in_editor(self):
         self.assertIsNotNone(sa.get_project_default_image_quality_in_editor(self.PROJECT_NAME))
 
     def test_get_project_metadata(self):
-        metadata = sa.upload_images_from_folder_to_project(self.PROJECT_NAME, self.folder_path)
-        pass
+        metadata = sa.get_project_metadata(self.PROJECT_NAME, include_settings=True)
+        print(metadata)