superb-ai-onprem 0.2.0__py3-none-any.whl → 0.3.1__py3-none-any.whl

spb_onprem/exports/service.py

@@ -0,0 +1,229 @@
+from typing import Optional, List, Union
+
+from spb_onprem.base_service import BaseService
+from spb_onprem.base_types import (
+    Undefined,
+    UndefinedType,
+)
+from spb_onprem.data.params import DataListFilter
+
+from .entities import Export
+from .params import (
+    ExportFilter,
+)
+from .queries import Queries
+
+
+class ExportService(BaseService):
+    """Service class for handling export-related operations."""
+    
+    def create_export(
+        self,
+        dataset_id: str,
+        name: Union[
+            UndefinedType,
+            str
+        ] = Undefined,
+        data_filter: Union[
+            UndefinedType,
+            DataListFilter,
+            dict
+        ] = Undefined,
+        location: Union[
+            UndefinedType,
+            str
+        ] = Undefined,
+        data_count: Union[
+            UndefinedType,
+            int
+        ] = Undefined,
+        frame_count: Union[
+            UndefinedType,
+            int
+        ] = Undefined,
+        annotation_count: Union[
+            UndefinedType,
+            int
+        ] = Undefined,
+        meta: Union[
+            UndefinedType,
+            dict
+        ] = Undefined,
+    ) -> Export:
+        """Create an export.
+        
+        Args:
+            dataset_id (str): The ID of the dataset to create the export for.
+            name (Optional[str]): The name of the export.
+            data_filter (Optional[DataListFilter | dict]): The search filter of the data.
+            data_count (Optional[int]): The number of data items to export.
+            frame_count (Optional[int]): The number of frames to export.
+            annotation_count (Optional[int]): The number of annotations to export.
+            meta (Optional[dict]): The meta information for the export.
+        """
+        response = self.request_gql(
+            Queries.CREATE_EXPORT,
+            Queries.CREATE_EXPORT["variables"](
+                dataset_id=dataset_id,
+                name=name,
+                data_filter=data_filter,
+                location=location,
+                data_count=data_count,
+                frame_count=frame_count,
+                annotation_count=annotation_count,
+                meta=meta,
+            )
+        )
+        return Export.model_validate(response)
+    
+    def get_exports(
+        self,
+        dataset_id: str,
+        export_filter: Optional[ExportFilter] = None,
+        cursor: Optional[str] = None,
+        length: int = 10
+    ) -> tuple[List[Export], Optional[str], int]:
+        """Get exports.
+        
+        Args:
+            dataset_id (str): The ID of the dataset to get exports for.
+            export_filter (Optional[ExportsFilterOptions]): The filter to apply to the exports.
+            cursor (Optional[str]): The cursor to use for pagination.
+            length (int): The number of exports to get.
+        
+        Returns:
+            tuple[List[Export], Optional[str], int]: A tuple containing the exports, the next cursor, and the total count of exports.
+        """
+        response = self.request_gql(
+            Queries.GET_EXPORTS,
+            Queries.GET_EXPORTS["variables"](
+                dataset_id=dataset_id,
+                export_filter=export_filter,
+                cursor=cursor,
+                length=length,
+            )
+        )
+        exports_dict = response.get("exports", [])
+        return (
+            [Export.model_validate(export_dict) for export_dict in exports_dict],
+            response.get("next"),
+            response.get("totalCount"),
+        )
+
+    def get_export(
+        self,
+        dataset_id: str,
+        export_id: str,
+    ) -> Export:
+        """Get an export.
+        
+        Args:
+            dataset_id (str): The ID of the dataset to get the export for.
+            export_id (str): The ID of the export to get.
+        
+        Returns:
+            Export: The export object.
+        """
+        response = self.request_gql(
+            Queries.GET_EXPORT,
+            Queries.GET_EXPORT["variables"](
+                dataset_id=dataset_id,
+                export_id=export_id,
+            )
+        )
+        return Export.model_validate(response)
+
+    def update_export(
+        self,
+        dataset_id: str,
+        export_id: str,
+        location: Union[
+            UndefinedType,
+            str
+        ] = Undefined,
+        name: Union[
+            UndefinedType,
+            str
+        ] = Undefined,
+        data_filter: Union[
+            UndefinedType,
+            DataListFilter,
+            dict
+        ] = Undefined,
+        data_count: Union[
+            UndefinedType,
+            int
+        ] = Undefined,
+        frame_count: Union[
+            UndefinedType,
+            int
+        ] = Undefined,
+        annotation_count: Union[
+            UndefinedType,
+            int
+        ] = Undefined,
+        meta: Union[
+            UndefinedType,
+            dict
+        ] = Undefined,
+        completed_at: Union[
+            UndefinedType,
+            str
+        ] = Undefined,
+    ) -> Export:
+        """Update an export.
+        
+        Args:
+            dataset_id (str): The ID of the dataset to update the export for.
+            export_id (str): The ID of the export to update.
+            location (Optional[str]): The location where the export will be stored.
+            name (Optional[str]): The name of the export.
+            data_filter (Optional[DataListFilter | dict]): The search filter of the data.
+            data_count (Optional[int]): The number of data items to export.
+            frame_count (Optional[int]): The number of frames to export.
+            annotation_count (Optional[int]): The number of annotations to export.
+            meta (Optional[dict]): The meta information for the export.
+            completed_at (Optional[str]): The completed time of the export.
+        
+        Returns:
+            Export: The updated export object.
+        """
+        response = self.request_gql(
+            Queries.UPDATE_EXPORT,
+            Queries.UPDATE_EXPORT["variables"](
+                dataset_id=dataset_id,
+                export_id=export_id,
+                location=location,
+                name=name,
+                data_filter=data_filter,
+                data_count=data_count,
+                frame_count=frame_count,
+                annotation_count=annotation_count,
+                meta=meta,
+                completed_at=completed_at,
+            )
+        )
+        return Export.model_validate(response)
+
+    def delete_export(
+        self,
+        dataset_id: str,
+        export_id: str,
+    ) -> bool:
+        """Delete an export.
+        
+        Args:
+            dataset_id (str): The ID of the dataset to delete the export for.
+            export_id (str): The ID of the export to delete.
+        
+        Returns:
+            bool: True if the export was deleted, False otherwise.
+        """
+        response = self.request_gql(
+            Queries.DELETE_EXPORT,
+            Queries.DELETE_EXPORT["variables"](
+                dataset_id=dataset_id,
+                export_id=export_id,
+            )
+        )
+        return response

spb_onprem/searches.py

@@ -16,6 +16,10 @@ from .activities.params.activities import (
     ActivitiesFilter,
     ActivitiesFilterOptions,
 )
+from .exports.params.exports import (
+    ExportFilter,
+    ExportFilterOptions,
+)
 
 __all__ = [
     "AnnotationFilter",
@@ -27,4 +31,6 @@ __all__ = [
     "SlicesFilterOptions",
     "ActivitiesFilter",
     "ActivitiesFilterOptions",
+    "ExportFilter",
+    "ExportFilterOptions",
 ]

{superb_ai_onprem-0.2.0.dist-info → superb_ai_onprem-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: superb-ai-onprem
-Version: 0.2.0
+Version: 0.3.1
 Summary: Python SDK for Superb AI On-premise
 Home-page: https://github.com/Superb-AI-Suite/superb-ai-onprem-python
 Author: Superb AI

{superb_ai_onprem-0.2.0.dist-info → superb_ai_onprem-0.3.1.dist-info}/RECORD

@@ -1,11 +1,11 @@
-spb_onprem/__init__.py,sha256=X2jDAWU5Dp2lEfA6fCSHZbcwEyzMZJ0gS92Um9QnZlU,1586
-spb_onprem/_version.py,sha256=iB5DfB5V6YB5Wo4JmvS-txT42QtmGaWcWp3udRT7zCI,511
+spb_onprem/__init__.py,sha256=Iucr68oU5pQ3lhHlyI36HUaReA7gsanGuLCSAVZh7GI,1761
+spb_onprem/_version.py,sha256=lOWWIGJeBi0KkFopWU_n3GH71C1PsaZ-ZYDfxFkne6c,511
 spb_onprem/base_model.py,sha256=XLtyoxRBs68LrvbFH8V4EvQGPc2W17koC310MnS37jc,442
 spb_onprem/base_service.py,sha256=dPfr3mGXYlqadOXycu6RBFX1HcZ1qzEsskLoOxERLOU,5737
 spb_onprem/base_types.py,sha256=5HO6uy6qf08b4KSElwIaGy7UkoQG2KqVO6gcHbsqqSo,269
-spb_onprem/entities.py,sha256=Z3MA-W8jw-qYaCXbiMiYJxpqLUBoGVVtZpMcKsz2UJk,776
+spb_onprem/entities.py,sha256=VFwaSdHIVB6NhxX5-pFRNPzCqVLNEZS-O2JETIYMopk,823
 spb_onprem/exceptions.py,sha256=jx5rTGsVZ5shOdbgQzk8GcSyMWFtb_3xhPq6Sylwc5o,478
-spb_onprem/searches.py,sha256=JxeVkASLFtVsJYY7lmxBoDDAXcFgrj-B6HPOT2MY2tA,620
+spb_onprem/searches.py,sha256=VRj9vpvKU8iOdBJUA6BwfvOjUbWrKBo9fPJvNlRmKr0,750
 spb_onprem/activities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spb_onprem/activities/queries.py,sha256=iXuNVsJuw8yt9QNw8jEBXpUnGLD6LDC1v2_jBBgsmXs,5626
 spb_onprem/activities/service.py,sha256=DI68qkDrvpm1NW_9BzWc37kSzyedfx4xLHqMeyDjp3A,10554
@@ -37,7 +37,7 @@ spb_onprem/data/entities/prediction.py,sha256=Eb2ldNSezeYDnaLQOfC65XWoDGJ0snlvlc
 spb_onprem/data/entities/scene.py,sha256=SJgr5UnGxktyaKjU8FYDaIQmsu7xSJftJOiCgq9uSmo,446
 spb_onprem/data/enums/__init__.py,sha256=IJWaapwesyIiIjuAykZc5fXdMXK2_IiOBa7qNY5cCNk,213
 spb_onprem/data/enums/data_meta_type.py,sha256=9rd12-7C1udbbIGvnuGURKmd5-lndtW7oWQAQwKSf_Q,335
-spb_onprem/data/enums/data_type.py,sha256=S7sbKHtJC_pvhNxLt8xtSFO3edo0Q9c7pZ3UyEeCeVA,178
+spb_onprem/data/enums/data_type.py,sha256=BXcmeuWL-y4yIAswI_zsYqH9ozEmgl6uSki7hgbOZ_g,197
 spb_onprem/data/enums/scene_type.py,sha256=ed8fAKfDk9PNG8YgYv3jI59IR9oCr_gkooexAe0448I,187
 spb_onprem/data/params/__init__.py,sha256=IApFFU6t3lHclvbivLSFdUxhrj1BjO50c3OMG6zP2iY,1311
 spb_onprem/data/params/create_data.py,sha256=8DOfbEchf9GSybiYY1cZoNzNYwILokoAkxXRFybJUAU,2038
@@ -67,6 +67,17 @@ spb_onprem/datasets/params/create_dataset.py,sha256=YGhLvY4arthjZwKQ28HLv7ch0Gd2
 spb_onprem/datasets/params/dataset.py,sha256=WTOUl5M5cc6rtTwhLw_z31Cs209LkBq8Ja4LJGzrmGE,668
 spb_onprem/datasets/params/datasets.py,sha256=Hx4YlLxfb-Qwi4Y5AFl5pyyjupvuoVcCtxLPGjIV7UY,1580
 spb_onprem/datasets/params/update_dataset.py,sha256=1oaj2qB9hvnypl4-WtcTNCa4iSuEkJjEalq2JsTm5Ro,924
+spb_onprem/exports/__init__.py,sha256=l_eUjnrFJXs-vdOeMNiWYK-UdJkxArv7hxmKDG4Dfio,181
+spb_onprem/exports/queries.py,sha256=sFT6VX2UwAYyVNkiBteQ_JtKYnhMrt64ww_NuXhUhLM,4084
+spb_onprem/exports/service.py,sha256=dYdi-X03A8YuKxxvQhMsyzJ8r95DxXQr5KZ818g0vSk,7118
+spb_onprem/exports/entities/__init__.py,sha256=_w5Qs_9Dvmy8_tweOmEpGmlMHx8m70rDSl94o7oTfmk,94
+spb_onprem/exports/entities/export.py,sha256=awE2xASCarLcmxNwvjjs6CqWXKptz2M-sGE-AUf74bI,1084
+spb_onprem/exports/params/__init__.py,sha256=F4X2go6V1vZ_pts5thKwW8Gal8otgv6FlLYSMDmPaMg,471
+spb_onprem/exports/params/create_export.py,sha256=vC6qmGETQNQ9PIbe7ayarEe0KuBwylWupBqQOlsDD8E,2594
+spb_onprem/exports/params/delete_export.py,sha256=EusUB86HNLtFYu4gIDJqZsODRETtTYhgxznjFHfxywc,664
+spb_onprem/exports/params/export.py,sha256=0EP6nkQc6vFI-f8218Yq4NxfFcw8MQtHMNkYlGOXqo4,799
+spb_onprem/exports/params/exports.py,sha256=oOQo-2Cqsm3Th3s-0gNcVGjbOyZqB6ujPRLXipulB4Y,2417
+spb_onprem/exports/params/update_export.py,sha256=iOlZoHEN2iiY83hCZWdgTn-O9J8hcSUqd5K7_gEzvP4,3057
 spb_onprem/slices/__init__.py,sha256=xgpNGYzqgwQ8C-Bgw9AZWMAgBW38UU-U4Wube8hkodI,69
 spb_onprem/slices/queries.py,sha256=dl_q6Uc2_oeuItgSD6gUL7a3H5VrOW9Ig5Epte7sl78,2732
 spb_onprem/slices/service.py,sha256=byhB9CdxNKV7uLIiL9yI674UVSlgaAwLaYoGQGBCawE,4988
@@ -81,13 +92,19 @@ spb_onprem/slices/params/update_slice.py,sha256=kryOmCnRTQ_OU0qDKgugppLrpeUpuLwm
 spb_onprem/users/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spb_onprem/users/entities/__init__.py,sha256=X8HZsCTlQnuPszok3AwI-i7bsQi0Ehul5L_2jZaol5E,57
 spb_onprem/users/entities/auth.py,sha256=_KP-7yUErBxhJMm-dE3ObprPEG6e0JI2qNg6g8aK1qM,3371
-superb_ai_onprem-0.2.0.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
+superb_ai_onprem-0.3.1.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
 tests/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 tests/activities/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 tests/activities/real_test.py,sha256=0gQHg7rIEuZndGZyNHMWSD5nUZPMsUGigfCjWFxMthQ,1786
 tests/activities/test_params.py,sha256=L3mcnrN2c8dT0AVTjnGedu6LHF3eZ4sSDzz8eaMHJmg,2369
 tests/activities/test_service.py,sha256=AbWFwjaUjQRs5oDZ2ba0C23XJxCqsFNrxH1LamMtWmU,4698
-superb_ai_onprem-0.2.0.dist-info/METADATA,sha256=GYkPH8YMVeOq_dEtkWRgHUOpVg6WaHLHaZLO_MmcJmE,5817
-superb_ai_onprem-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-superb_ai_onprem-0.2.0.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
-superb_ai_onprem-0.2.0.dist-info/RECORD,,
+tests/exports/__init__.py,sha256=iBqr781TtnT1Tm5wLTCsdftpuRgQifIibxWrB_I7tiw,23
+tests/exports/real_test.py,sha256=KeHWvQP-vHCvFWvD4_7qFmreAX9o_Dd_YowaGFogZu8,3765
+tests/exports/test_entities.py,sha256=hG7G4kVkyHKT3mv4lvrzUqOW8ILeHiYj87QZjQcmg9E,8836
+tests/exports/test_integration.py,sha256=cCcEgwIIHyQRlc04EAXSKz7RcblQvhI2GBR3uVaOOq8,6201
+tests/exports/test_params.py,sha256=oRRa6nEru_FImlB3TrmFiBidz6ZstCx4rVaCK-EMGfQ,11070
+tests/exports/test_service.py,sha256=IDcKxrmByh00jk9142P2ThuGureMoijTHNdU0rERGG8,14628
+superb_ai_onprem-0.3.1.dist-info/METADATA,sha256=wRZVOqPK9ul6lYcUgGgFd9Yaf9zWvnfqEhJND6JACOA,5817
+superb_ai_onprem-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+superb_ai_onprem-0.3.1.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
+superb_ai_onprem-0.3.1.dist-info/RECORD,,

tests/exports/__init__.py

@@ -0,0 +1 @@
+# Export tests package 

tests/exports/real_test.py

@@ -0,0 +1,130 @@
+from spb_onprem import (
+    DatasetService,
+    Dataset,
+    ExportService,
+    Export,
+    ExportFilter,
+    ExportFilterOptions,
+)
+from spb_onprem.data.params import DataListFilter, DataFilterOptions
+from spb_onprem.data.enums import DataType
+
+
+def test_export_service():
+    # Initialize services
+    dataset_service = DatasetService()
+    dataset = dataset_service.get_dataset(
+        dataset_id="01JPM6NR1APMBXJNC0YW72S1FN"
+    )
+
+    print(f"Dataset: {dataset}")
+    
+    export_service = ExportService()
+    
+    # Test 1: Create an export with DataListFilter
+    print("\n=== Creating Export ===")
+    data_filter = DataListFilter(
+        must_filter=DataFilterOptions(key_contains="validation")
+    )
+    
+    new_export = export_service.create_export(
+        dataset_id=dataset.id,
+        name="SDK Test Export",
+        data_filter=data_filter,
+        meta={
+            "created_by": "sdk_test",
+            "purpose": "real_test"
+        }
+    )
+    print(f"Created export: {new_export}")
+    
+    # Test 2: Get exports with pagination
+    print("\n=== Getting Exports ===")
+    cursor = None
+    all_exports = []
+    while True:
+        exports, cursor, total_count = export_service.get_exports(
+            dataset_id=dataset.id,
+            cursor=cursor,
+            length=10
+        )
+        all_exports.extend(exports)
+        print(f"Fetched {len(exports)} exports, total: {total_count}")
+        
+        if cursor is None:
+            break
+    
+    print(f"Total exports found: {len(all_exports)}")
+    
+    # Test 3: Get exports with filter
+    print("\n=== Getting Exports with Filter ===")
+    export_filter = ExportFilter(
+        must_filter=ExportFilterOptions(
+            name_contains="SDK Test"
+        )
+    )
+    
+    filtered_exports, _, filtered_count = export_service.get_exports(
+        dataset_id=dataset.id,
+        export_filter=export_filter,
+        length=50
+    )
+    print(f"Filtered exports: {len(filtered_exports)}, total: {filtered_count}")
+    
+    # Test 4: Get specific export
+    if all_exports:
+        print("\n=== Getting Specific Export ===")
+        specific_export = export_service.get_export(
+            dataset_id=dataset.id,
+            export_id=all_exports[0].id
+        )
+        print(f"Specific export: {specific_export}")
+    
+    # Test 5: Update the created export with complex DataListFilter
+    print("\n=== Updating Export ===")
+    complex_data_filter = DataListFilter(
+        must_filter=DataFilterOptions(
+            key_contains="validation",
+            type_in=[DataType.SUPERB_IMAGE]
+        ),
+        not_filter=DataFilterOptions(
+            key_contains="test"
+        )
+    )
+    
+    updated_export = export_service.update_export(
+        dataset_id=dataset.id,
+        export_id=new_export.id,
+        name="SDK Test Export - Updated",
+        data_filter=complex_data_filter,
+        meta={
+            "created_by": "sdk_test",
+            "purpose": "real_test",
+            "updated": True,
+            "status": "completed"
+        }
+    )
+    print(f"Updated export: {updated_export}")
+    
+    # Test 6: Delete the created export
+    print("\n=== Deleting Export ===")
+    delete_result = export_service.delete_export(
+        dataset_id=dataset.id,
+        export_id=new_export.id
+    )
+    print(f"Delete result: {delete_result}")
+    
+    # Test 7: Verify deletion
+    print("\n=== Verifying Deletion ===")
+    try:
+        deleted_export = export_service.get_export(
+            dataset_id=dataset.id,
+            export_id=new_export.id
+        )
+        print(f"Export still exists: {deleted_export}")
+    except Exception as e:
+        print(f"Export successfully deleted (expected error): {e}")
+
+
+if __name__ == "__main__":
+    test_export_service()