Add folder_id in the item metadata

Author: VaghinakDev

src/superannotate/__init__.py

@@ -3,7 +3,7 @@
 import sys
 
 
-__version__ = "4.4.32"
+__version__ = "4.4.33b1"
 
 os.environ.update({"sa_version": __version__})
 sys.path.append(os.path.split(os.path.realpath(__file__))[0])

src/superannotate/lib/core/entities/base.py

@@ -57,6 +57,7 @@ class TimedBaseModel(BaseModel):
 class BaseItemEntity(TimedBaseModel):
     id: Optional[int]
     name: Optional[str]
+    folder_id: Optional[int]
     path: Optional[str] = Field(
         None, description="Item’s path in SuperAnnotate project"
     )

src/superannotate/lib/infrastructure/annotation_adapter.py

@@ -44,7 +44,9 @@ def get_component_value(self, component_id: str):
         return None
 
     def set_component_value(self, component_id: str, value: Any):
-        self.annotation.setdefault("data", {}).setdefault(component_id, {})["value"] = value
+        self.annotation.setdefault("data", {}).setdefault(component_id, {})[
+            "value"
+        ] = value
         return self
 
 

tests/integration/items/test_get_item_metadata.py

@@ -20,6 +20,7 @@ class TestGetEntityMetadataVector(BaseTestCase):
     EXPECTED_ITEM_METADATA = {
         "name": "example_image_1.jpg",
         "path": "TestGetEntityMetadataVector",
+        "folder_id": None,
         "url": None,
         "annotator_email": None,
         "qa_email": None,