cognite-toolkit 0.7.71__py3-none-any.whl → 0.7.72__py3-none-any.whl

cognite_toolkit/_cdf_tk/client/_toolkit_client.py

@@ -11,7 +11,6 @@ from cognite_toolkit._cdf_tk.client.api.legacy.extended_files import ExtendedFil
 from cognite_toolkit._cdf_tk.client.api.legacy.extended_functions import ExtendedFunctionsAPI
 from cognite_toolkit._cdf_tk.client.api.legacy.extended_raw import ExtendedRawAPI
 from cognite_toolkit._cdf_tk.client.api.legacy.extended_timeseries import ExtendedTimeSeriesAPI
-from cognite_toolkit._cdf_tk.client.api.legacy.robotics import RoboticsAPI as RoboticsLegacyAPI
 from cognite_toolkit._cdf_tk.client.http_client import HTTPClient
 
 from .api.assets import AssetsAPI
@@ -80,7 +79,6 @@ class ToolkitClient(CogniteClient):
         self.console = console or Console()
         self.tool = ToolAPI(http_client, self.console)
         self.search = SearchAPI(self._config, self._API_VERSION, self)
-        self.robotics = RoboticsLegacyAPI(self._config, self._API_VERSION, self)
         self.dml = DMLAPI(self._config, self._API_VERSION, self)
         self.verify = VerifyAPI(self._config, self._API_VERSION, self)
         self.lookup = LookUpGroup(self._config, self._API_VERSION, self, self.console)

cognite_toolkit/_cdf_tk/client/api/robotics_capabilities.py

@@ -53,7 +53,7 @@ class CapabilitiesAPI(CDFResourceAPI[ExternalId, RobotCapabilityRequest, RobotCa
         """
         return self._request_item_response(items, "create")
 
-    def retrieve(self, items: Sequence[ExternalId]) -> list[RobotCapabilityResponse]:
+    def retrieve(self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False) -> list[RobotCapabilityResponse]:
         """Retrieve capabilities from CDF.
 
         Args:
@@ -61,6 +61,8 @@ class CapabilitiesAPI(CDFResourceAPI[ExternalId, RobotCapabilityRequest, RobotCa
         Returns:
             List of retrieved RobotCapabilityResponse objects.
         """
+        if ignore_unknown_ids:
+            return self._request_item_split_retries(items, method="retrieve")
         return self._request_item_response(items, method="retrieve")
 
     def update(
@@ -77,13 +79,17 @@ class CapabilitiesAPI(CDFResourceAPI[ExternalId, RobotCapabilityRequest, RobotCa
         """
         return self._update(items, mode=mode)
 
-    def delete(self, items: Sequence[ExternalId]) -> None:
+    def delete(self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False) -> None:
         """Delete capabilities from CDF.
 
         Args:
             items: List of ExternalId objects to delete.
+            ignore_unknown_ids: If True, ignores unknown IDs during deletion.
         """
-        self._request_no_response(items, "delete")
+        if ignore_unknown_ids:
+            self._request_item_split_retries_no_response(items, "delete")
+        else:
+            self._request_no_response(items, "delete")
 
     def paginate(
         self,

cognite_toolkit/_cdf_tk/client/api/robotics_data_postprocessing.py

@@ -64,14 +64,19 @@ class DataPostProcessingAPI(
         """
         return self._request_item_response(items, "create")
 
-    def retrieve(self, items: Sequence[ExternalId]) -> list[RobotDataPostProcessingResponse]:
+    def retrieve(
+        self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False
+    ) -> list[RobotDataPostProcessingResponse]:
         """Retrieve data post-processing configurations from CDF.
 
         Args:
             items: List of ExternalId objects to retrieve.
+            ignore_unknown_ids: If True, ignores unknown IDs during retrieval.
         Returns:
             List of retrieved RobotDataPostProcessingResponse objects.
         """
+        if ignore_unknown_ids:
+            return self._request_item_split_retries(items, method="retrieve")
         return self._request_item_response(items, method="retrieve")
 
     def update(
@@ -88,13 +93,17 @@ class DataPostProcessingAPI(
         """
         return self._update(items, mode=mode)
 
-    def delete(self, items: Sequence[ExternalId]) -> None:
+    def delete(self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False) -> None:
         """Delete data post-processing configurations from CDF.
 
         Args:
             items: List of ExternalId objects to delete.
+            ignore_unknown_ids: If True, ignores unknown IDs during deletion.
         """
-        self._request_no_response(items, "delete")
+        if ignore_unknown_ids:
+            self._request_item_split_retries_no_response(items, "delete")
+        else:
+            self._request_no_response(items, "delete")
 
     def paginate(
         self,

cognite_toolkit/_cdf_tk/client/api/robotics_frames.py

@@ -48,14 +48,17 @@ class FramesAPI(CDFResourceAPI[ExternalId, RobotFrameRequest, RobotFrameResponse
         """
         return self._request_item_response(items, "create")
 
-    def retrieve(self, items: Sequence[ExternalId]) -> list[RobotFrameResponse]:
+    def retrieve(self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False) -> list[RobotFrameResponse]:
         """Retrieve frames from CDF.
 
         Args:
             items: List of ExternalId objects to retrieve.
+            ignore_unknown_ids: If True, ignores unknown IDs during retrieval.
         Returns:
             List of retrieved RobotFrameResponse objects.
         """
+        if ignore_unknown_ids:
+            return self._request_item_split_retries(items, method="retrieve")
         return self._request_item_response(items, method="retrieve")
 
     def update(
@@ -72,13 +75,17 @@ class FramesAPI(CDFResourceAPI[ExternalId, RobotFrameRequest, RobotFrameResponse
         """
         return self._update(items, mode=mode)
 
-    def delete(self, items: Sequence[ExternalId]) -> None:
+    def delete(self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False) -> None:
         """Delete frames from CDF.
 
         Args:
             items: List of ExternalId objects to delete.
+            ignore_unknown_ids: If True, ignores unknown IDs during deletion.
         """
-        self._request_no_response(items, "delete")
+        if ignore_unknown_ids:
+            self._request_item_split_retries_no_response(items, "delete")
+        else:
+            self._request_no_response(items, "delete")
 
     def paginate(
         self,

cognite_toolkit/_cdf_tk/client/api/robotics_locations.py

@@ -53,14 +53,17 @@ class LocationsAPI(CDFResourceAPI[ExternalId, RobotLocationRequest, RobotLocatio
         """
         return self._request_item_response(items, "create")
 
-    def retrieve(self, items: Sequence[ExternalId]) -> list[RobotLocationResponse]:
+    def retrieve(self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False) -> list[RobotLocationResponse]:
         """Retrieve locations from CDF.
 
         Args:
             items: List of ExternalId objects to retrieve.
+            ignore_unknown_ids: If True, ignores unknown IDs during retrieval.
         Returns:
             List of retrieved RobotLocationResponse objects.
         """
+        if ignore_unknown_ids:
+            return self._request_item_split_retries(items, method="retrieve")
         return self._request_item_response(items, method="retrieve")
 
     def update(
@@ -77,13 +80,17 @@ class LocationsAPI(CDFResourceAPI[ExternalId, RobotLocationRequest, RobotLocatio
         """
         return self._update(items, mode=mode)
 
-    def delete(self, items: Sequence[ExternalId]) -> None:
+    def delete(self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False) -> None:
         """Delete locations from CDF.
 
         Args:
             items: List of ExternalId objects to delete.
+            ignore_unknown_ids: If True, ignores unknown IDs during deletion.
         """
-        self._request_no_response(items, "delete")
+        if ignore_unknown_ids:
+            self._request_item_split_retries_no_response(items, "delete")
+        else:
+            self._request_no_response(items, "delete")
 
     def paginate(
         self,

cognite_toolkit/_cdf_tk/client/api/robotics_maps.py

@@ -48,14 +48,17 @@ class MapsAPI(CDFResourceAPI[ExternalId, RobotMapRequest, RobotMapResponse]):
         """
         return self._request_item_response(items, "create")
 
-    def retrieve(self, items: Sequence[ExternalId]) -> list[RobotMapResponse]:
+    def retrieve(self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False) -> list[RobotMapResponse]:
         """Retrieve maps from CDF.
 
         Args:
             items: List of ExternalId objects to retrieve.
+            ignore_unknown_ids: If True, ignores unknown IDs during retrieval.
         Returns:
             List of retrieved RobotMapResponse objects.
         """
+        if ignore_unknown_ids:
+            return self._request_item_split_retries(items, method="retrieve")
         return self._request_item_response(items, method="retrieve")
 
     def update(
@@ -72,13 +75,17 @@ class MapsAPI(CDFResourceAPI[ExternalId, RobotMapRequest, RobotMapResponse]):
         """
         return self._update(items, mode=mode)
 
-    def delete(self, items: Sequence[ExternalId]) -> None:
+    def delete(self, items: Sequence[ExternalId], ignore_unknown_ids: bool = False) -> None:
         """Delete maps from CDF.
 
         Args:
             items: List of ExternalId objects to delete.
+            ignore_unknown_ids: If True, ignores unknown IDs during deletion.
         """
-        self._request_no_response(items, "delete")
+        if ignore_unknown_ids:
+            self._request_item_split_retries_no_response(items, "delete")
+        else:
+            self._request_no_response(items, "delete")
 
     def paginate(
         self,

cognite_toolkit/_cdf_tk/client/api/robotics_robots.py

@@ -46,14 +46,17 @@ class RobotsAPI(CDFResourceAPI[DataSetId, RobotRequest, RobotResponse]):
         """
         return self._request_item_response(items, "create")
 
-    def retrieve(self, items: Sequence[DataSetId]) -> list[RobotResponse]:
+    def retrieve(self, items: Sequence[DataSetId], ignore_unknown_ids: bool = False) -> list[RobotResponse]:
         """Retrieve robots from CDF.
 
         Args:
             items: List of DataSetId objects to retrieve.
+            ignore_unknown_ids: If True, ignores unknown IDs during retrieval.
         Returns:
             List of retrieved RobotResponse objects.
         """
+        if ignore_unknown_ids:
+            return self._request_item_split_retries(items, method="retrieve")
         return self._request_item_response(items, method="retrieve")
 
     def update(
@@ -70,13 +73,17 @@ class RobotsAPI(CDFResourceAPI[DataSetId, RobotRequest, RobotResponse]):
         """
         return self._update(items, mode=mode)
 
-    def delete(self, items: Sequence[DataSetId]) -> None:
+    def delete(self, items: Sequence[DataSetId], ignore_unknown_ids: bool = False) -> None:
         """Delete robots from CDF.
 
         Args:
             items: List of DataSetId objects to delete.
+            ignore_unknown_ids: If True, ignores unknown IDs during deletion.
         """
-        self._request_no_response(items, "delete")
+        if ignore_unknown_ids:
+            self._request_item_split_retries_no_response(items, "delete")
+        else:
+            self._request_no_response(items, "delete")
 
     def paginate(
         self,

cognite_toolkit/_cdf_tk/client/http_client/_client.py

@@ -65,7 +65,7 @@ class HTTPClient:
         pool_connections: int = 10,
         pool_maxsize: int = 20,
         retry_status_codes: Set[int] = frozenset({408, 429, 502, 503, 504}),
-        split_items_status_codes: Set[int] = frozenset({400, 408, 409, 422, 502, 503, 504}),
+        split_items_status_codes: Set[int] = frozenset({400, 404, 408, 409, 422, 502, 503, 504}),
         console: Console | None = None,
     ):
         self.config = config

cognite_toolkit/_cdf_tk/client/testing.py

@@ -23,15 +23,14 @@ from cognite_toolkit._cdf_tk.client.api.legacy.extended_functions import Extende
 from cognite_toolkit._cdf_tk.client.api.legacy.extended_raw import ExtendedRawAPI
 from cognite_toolkit._cdf_tk.client.api.legacy.extended_timeseries import ExtendedTimeSeriesAPI
 from cognite_toolkit._cdf_tk.client.api.legacy.location_filters import LocationFiltersAPI
-from cognite_toolkit._cdf_tk.client.api.legacy.robotics import (
-    CapabilitiesAPI,
-    DataPostProcessingAPI,
-    FramesAPI,
-    MapsAPI,
-    RoboticsAPI,
-)
-from cognite_toolkit._cdf_tk.client.api.legacy.robotics import LocationsAPI as RoboticsLocationsAPI
 from cognite_toolkit._cdf_tk.client.api.legacy.search_config import SearchConfigurationsAPI
+from cognite_toolkit._cdf_tk.client.api.robotics import RoboticsAPI
+from cognite_toolkit._cdf_tk.client.api.robotics_capabilities import CapabilitiesAPI
+from cognite_toolkit._cdf_tk.client.api.robotics_data_postprocessing import DataPostProcessingAPI
+from cognite_toolkit._cdf_tk.client.api.robotics_frames import FramesAPI
+from cognite_toolkit._cdf_tk.client.api.robotics_locations import LocationsAPI
+from cognite_toolkit._cdf_tk.client.api.robotics_maps import MapsAPI
+from cognite_toolkit._cdf_tk.client.api.robotics_robots import RobotsAPI
 
 from ._toolkit_client import ToolAPI
 from .api.assets import AssetsAPI
@@ -139,14 +138,6 @@ class ToolkitClientMock(CogniteClientMock):
         self.raw.rows = MagicMock(spec_set=RawRowsAPI)
         self.raw.tables = MagicMock(spec_set=RawTablesAPI)
 
-        self.robotics = MagicMock()
-        self.robotics.robots = MagicMock(spec=RoboticsAPI)
-        self.robotics.data_postprocessing = MagicMock(spec_set=DataPostProcessingAPI)
-        self.robotics.locations = MagicMock(spec_set=RoboticsLocationsAPI)
-        self.robotics.frames = MagicMock(spec_set=FramesAPI)
-        self.robotics.maps = MagicMock(spec_set=MapsAPI)
-        self.robotics.capabilities = MagicMock(spec_set=CapabilitiesAPI)
-
         self.data_modeling.instances = MagicMock(spec_set=ExtendedInstancesAPI)
 
         self.time_series = MagicMock(spec=ExtendedTimeSeriesAPI)
@@ -171,6 +162,13 @@ class ToolkitClientMock(CogniteClientMock):
         self.tool.hosted_extractors.destinations = MagicMock(spec_set=HostedExtractorDestinationsAPI)
         self.tool.hosted_extractors.mappings = MagicMock(spec_set=HostedExtractorMappingsAPI)
         self.tool.labels = MagicMock(spec_set=LabelsAPI)
+        self.tool.robotics = MagicMock(spec=RoboticsAPI)
+        self.tool.robotics.capabilities = MagicMock(spec_set=CapabilitiesAPI)
+        self.tool.robotics.data_postprocessing = MagicMock(spec_set=DataPostProcessingAPI)
+        self.tool.robotics.frames = MagicMock(spec_set=FramesAPI)
+        self.tool.robotics.locations = MagicMock(spec_set=LocationsAPI)
+        self.tool.robotics.maps = MagicMock(spec_set=MapsAPI)
+        self.tool.robotics.robots = MagicMock(spec_set=RobotsAPI)
         self.tool.security_categories = MagicMock(spec_set=SecurityCategoriesAPI)
         self.tool.sequences = MagicMock(spec_set=SequencesAPI)
         self.tool.transformations = MagicMock(spec_set=TransformationsAPI)