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

cognite_toolkit/_cdf_tk/client/api/legacy/robotics/locations.py

@@ -1,136 +0,0 @@
-from collections.abc import Iterator, Sequence
-from typing import overload
-
-from cognite.client._api_client import APIClient
-from cognite.client.utils._identifier import IdentifierSequence
-from cognite.client.utils.useful_types import SequenceNotStr
-
-from cognite_toolkit._cdf_tk.client.resource_classes.legacy.robotics import (
-    Location,
-    LocationList,
-    LocationWrite,
-    _LocationUpdate,
-)
-
-from .utlis import tmp_disable_gzip
-
-
-class LocationsAPI(APIClient):
-    _RESOURCE_PATH = "/robotics/locations"
-
-    @overload
-    def __call__(self) -> Iterator[Location]: ...
-
-    @overload
-    def __call__(self, chunk_size: int) -> Iterator[LocationList]: ...
-
-    def __call__(self, chunk_size: int | None = None) -> Iterator[Location] | Iterator[LocationList]:
-        """Iterate over robot locations.
-
-        Args:
-            chunk_size: The number of robot locations to return in each chunk. None will return all robot locations.
-
-        Yields:
-            Location or LocationList
-
-        """
-        return self._list_generator(method="GET", resource_cls=Location, list_cls=LocationList, chunk_size=chunk_size)
-
-    def __iter__(self) -> Iterator[Location]:
-        return self.__call__()
-
-    @overload
-    def create(self, location: LocationWrite) -> Location: ...
-
-    @overload
-    def create(self, location: Sequence[LocationWrite]) -> LocationList: ...
-
-    def create(self, location: LocationWrite | Sequence[LocationWrite]) -> Location | LocationList:
-        """Create a new robot location.
-
-        Args:
-            location: LocationWrite or list of LocationWrite.
-
-        Returns:
-            Location object.
-
-        """
-        with tmp_disable_gzip():
-            return self._create_multiple(
-                list_cls=LocationList,
-                resource_cls=Location,
-                items=location,
-                input_resource_cls=LocationWrite,
-            )
-
-    @overload
-    def retrieve(self, external_id: str) -> Location | None: ...
-
-    @overload
-    def retrieve(self, external_id: SequenceNotStr[str]) -> LocationList: ...
-
-    def retrieve(self, external_id: str | SequenceNotStr[str]) -> Location | None | LocationList:
-        """Retrieve a robot location.
-
-        Args:
-            external_id: External id of the robot location.
-
-        Returns:
-            Location object.
-
-        """
-        identifiers = IdentifierSequence.load(external_ids=external_id)
-        with tmp_disable_gzip():
-            return self._retrieve_multiple(
-                identifiers=identifiers,
-                resource_cls=Location,
-                list_cls=LocationList,
-            )
-
-    @overload
-    def update(self, location: LocationWrite) -> Location: ...
-
-    @overload
-    def update(self, location: Sequence[LocationWrite]) -> LocationList: ...
-
-    def update(self, location: LocationWrite | Sequence[LocationWrite]) -> Location | LocationList:
-        """Update a robot location.
-
-        Args:
-            location: LocationWrite or list of LocationWrite.
-
-        Returns:
-            Location object.
-
-        """
-        with tmp_disable_gzip():
-            return self._update_multiple(
-                items=location,
-                resource_cls=Location,
-                list_cls=LocationList,
-                update_cls=_LocationUpdate,
-            )
-
-    def delete(self, external_id: str | SequenceNotStr[str]) -> None:
-        """Delete a robot location.
-
-        Args:
-            external_id: External id of the robot location.
-
-        Returns:
-            None
-
-        """
-        identifiers = IdentifierSequence.load(external_ids=external_id)
-        with tmp_disable_gzip():
-            self._delete_multiple(identifiers=identifiers, wrap_ids=True)
-
-    def list(self) -> LocationList:
-        """List robot locations.
-
-        Returns:
-            LocationList
-
-        """
-        with tmp_disable_gzip():
-            return self._list(method="GET", resource_cls=Location, list_cls=LocationList)

cognite_toolkit/_cdf_tk/client/api/legacy/robotics/maps.py

@@ -1,136 +0,0 @@
-from collections.abc import Iterator, Sequence
-from typing import overload
-
-from cognite.client._api_client import APIClient
-from cognite.client.utils._identifier import IdentifierSequence
-from cognite.client.utils.useful_types import SequenceNotStr
-
-from cognite_toolkit._cdf_tk.client.resource_classes.legacy.robotics import (
-    Map,
-    MapList,
-    MapWrite,
-    _MapUpdate,
-)
-
-from .utlis import tmp_disable_gzip
-
-
-class MapsAPI(APIClient):
-    _RESOURCE_PATH = "/robotics/maps"
-
-    @overload
-    def __call__(self) -> Iterator[Map]: ...
-
-    @overload
-    def __call__(self, chunk_size: int) -> Iterator[MapList]: ...
-
-    def __call__(self, chunk_size: int | None = None) -> Iterator[Map] | Iterator[MapList]:
-        """Iterate over robot maps.
-
-        Args:
-            chunk_size: The number of robot maps to return in each chunk. None will return all robot maps.
-
-        Yields:
-            Map or MapList
-
-        """
-        return self._list_generator(method="GET", resource_cls=Map, list_cls=MapList, chunk_size=chunk_size)
-
-    def __iter__(self) -> Iterator[Map]:
-        return self.__call__()
-
-    @overload
-    def create(self, map: MapWrite) -> Map: ...
-
-    @overload
-    def create(self, map: Sequence[MapWrite]) -> MapList: ...
-
-    def create(self, map: MapWrite | Sequence[MapWrite]) -> Map | MapList:
-        """Create a new robot map.
-
-        Args:
-            map: MapWrite or list of MapWrite.
-
-        Returns:
-            Map object.
-
-        """
-        with tmp_disable_gzip():
-            return self._create_multiple(
-                list_cls=MapList,
-                resource_cls=Map,
-                items=map,
-                input_resource_cls=MapWrite,
-            )
-
-    @overload
-    def retrieve(self, external_id: str) -> Map | None: ...
-
-    @overload
-    def retrieve(self, external_id: SequenceNotStr[str]) -> MapList: ...
-
-    def retrieve(self, external_id: str | SequenceNotStr[str]) -> Map | None | MapList:
-        """Retrieve a robot map.
-
-        Args:
-            external_id: External id of the robot map.
-
-        Returns:
-            Map object.
-
-        """
-        identifiers = IdentifierSequence.load(external_ids=external_id)
-        with tmp_disable_gzip():
-            return self._retrieve_multiple(
-                identifiers=identifiers,
-                resource_cls=Map,
-                list_cls=MapList,
-            )
-
-    @overload
-    def update(self, map: MapWrite) -> Map: ...
-
-    @overload
-    def update(self, map: Sequence[MapWrite]) -> MapList: ...
-
-    def update(self, map: MapWrite | Sequence[MapWrite]) -> Map | MapList:
-        """Update a robot map.
-
-        Args:
-            map: MapWrite or list of MapWrite.
-
-        Returns:
-            Map object.
-
-        """
-        with tmp_disable_gzip():
-            return self._update_multiple(
-                items=map,
-                resource_cls=Map,
-                list_cls=MapList,
-                update_cls=_MapUpdate,
-            )
-
-    def delete(self, external_id: str | SequenceNotStr[str]) -> None:
-        """Delete a robot map.
-
-        Args:
-            external_id: External id of the robot map.
-
-        Returns:
-            None
-
-        """
-        identifiers = IdentifierSequence.load(external_ids=external_id)
-        with tmp_disable_gzip():
-            self._delete_multiple(identifiers=identifiers, wrap_ids=True)
-
-    def list(self) -> MapList:
-        """List robot maps.
-
-        Returns:
-            MapList
-
-        """
-        with tmp_disable_gzip():
-            return self._list(method="GET", resource_cls=Map, list_cls=MapList)

cognite_toolkit/_cdf_tk/client/api/legacy/robotics/robots.py

@@ -1,132 +0,0 @@
-from collections.abc import Iterator, Sequence
-from typing import overload
-
-from cognite.client._api_client import APIClient
-
-from cognite_toolkit._cdf_tk.client.api.legacy.robotics.utlis import tmp_disable_gzip
-from cognite_toolkit._cdf_tk.client.resource_classes.legacy.robotics import Robot, RobotList, RobotWrite, _RobotUpdate
-
-
-class RobotsAPI(APIClient):
-    _RESOURCE_PATH = "/robotics/robots"
-
-    @overload
-    def __call__(self) -> Iterator[Robot]: ...
-
-    @overload
-    def __call__(self, chunk_size: int) -> Iterator[RobotList]: ...
-
-    def __call__(self, chunk_size: int | None = None) -> Iterator[Robot] | Iterator[RobotList]:
-        """Iterate over robots.
-
-        Args:
-            chunk_size: The number of robots to return in each chunk. None will return all robots.
-
-        Yields:
-            Robot or RobotList
-
-        """
-        return self._list_generator(method="GET", resource_cls=Robot, list_cls=RobotList, chunk_size=chunk_size)
-
-    def __iter__(self) -> Iterator[Robot]:
-        return self.__call__()
-
-    @overload
-    def create(self, robot: RobotWrite) -> Robot: ...
-
-    @overload
-    def create(self, robot: Sequence[RobotWrite]) -> RobotList: ...
-
-    def create(self, robot: RobotWrite | Sequence[RobotWrite]) -> Robot | RobotList:
-        """Create a new robot.
-
-        Args:
-            robot: RobotWrite or list of RobotWrite.
-
-        Returns:
-            Robot object.
-
-        """
-        with tmp_disable_gzip():
-            return self._create_multiple(
-                list_cls=RobotList,
-                resource_cls=Robot,
-                items=robot,
-                input_resource_cls=RobotWrite,
-            )
-
-    def retrieve(self, data_set_id: int | Sequence[int]) -> RobotList:
-        """Retrieve a robot.
-
-        Args:
-            data_set_id: Data set id of the robot.
-
-        Returns:
-            Robot object.
-
-        """
-        body = self._create_body(data_set_id)
-        res = self._post(url_path=self._RESOURCE_PATH + "/byids", json=body)
-        return RobotList._load(res.json()["items"], cognite_client=self._cognite_client)
-
-    @staticmethod
-    def _create_body(data_set_id: int | Sequence[int]) -> dict:
-        ids = [data_set_id] if isinstance(data_set_id, int) else data_set_id
-        body = {"items": [{"dataSetId": external_id} for external_id in ids]}
-        return body
-
-    @overload
-    def update(self, robot: RobotWrite) -> Robot: ...
-
-    @overload
-    def update(self, robot: Sequence[RobotWrite]) -> RobotList: ...
-
-    def update(self, robot: RobotWrite | Sequence[RobotWrite]) -> Robot | RobotList:
-        """Update a robot.
-
-        Args:
-            robot: RobotWrite or list of RobotWrite.
-
-        Returns:
-            Robot object.
-
-        """
-        is_single = False
-        if isinstance(robot, RobotWrite):
-            robots = [robot]
-            is_single = True
-        elif isinstance(robot, Sequence):
-            robots = list(robot)
-        else:
-            raise ValueError("robot must be a RobotWrite or a list of RobotWrite")
-
-        property_spec = _RobotUpdate._get_update_properties()
-        update = [
-            {"dataSetId": r.data_set_id, **self._convert_resource_to_patch_object(r, property_spec)} for r in robots
-        ]
-        res = self._post(url_path=self._RESOURCE_PATH + "/update", json={"items": update})
-        loaded = RobotList._load(res.json()["items"], cognite_client=self._cognite_client)
-        return loaded[0] if is_single else loaded
-
-    def delete(self, data_set_id: int | Sequence[int]) -> None:
-        """Delete a robot.
-
-        Args:
-            data_set_id: Data set id of the robot.
-
-        Returns:
-            None
-
-        """
-        body = self._create_body(data_set_id)
-        self._post(url_path=self._RESOURCE_PATH + "/delete", json=body)
-
-    def list(self) -> RobotList:
-        """List robots.
-
-        Returns:
-            RobotList
-
-        """
-        with tmp_disable_gzip():
-            return self._list(method="GET", resource_cls=Robot, list_cls=RobotList)

cognite_toolkit/_cdf_tk/client/api/legacy/robotics/utlis.py

@@ -1,13 +0,0 @@
-import contextlib
-from collections.abc import Iterator
-
-
-@contextlib.contextmanager
-def tmp_disable_gzip() -> Iterator[None]:
-    from cognite.client.config import global_config
-
-    _current_value = global_config.disable_gzip
-
-    global_config.disable_gzip = True
-    yield None
-    global_config.disable_gzip = _current_value