samsara-api 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

samsara/trips/raw_client.py

@@ -5,7 +5,7 @@ from json.decoder import JSONDecodeError
 
 from ..core.api_error import ApiError
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from ..core.http_response import AsyncHttpResponse, HttpResponse
+from ..core.pagination import AsyncPager, BaseHttpResponse, SyncPager
 from ..core.pydantic_utilities import parse_obj_as
 from ..core.request_options import RequestOptions
 from ..errors.bad_gateway_error import BadGatewayError
@@ -17,6 +17,7 @@ from ..errors.not_implemented_error import NotImplementedError
 from ..errors.service_unavailable_error import ServiceUnavailableError
 from ..errors.too_many_requests_error import TooManyRequestsError
 from ..errors.unauthorized_error import UnauthorizedError
+from ..types.trip_response_body import TripResponseBody
 from ..types.trips_get_trips_response_body import TripsGetTripsResponseBody
 from .types.trips_stream_request_completion_status import TripsStreamRequestCompletionStatus
 from .types.trips_stream_request_query_by import TripsStreamRequestQueryBy
@@ -37,7 +38,7 @@ class RawTripsClient:
         after: typing.Optional[str] = None,
         ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[TripsGetTripsResponseBody]:
+    ) -> SyncPager[TripResponseBody]:
         """
         This endpoint will return trips that have been collected for your organization based on the time parameters passed in. Results are paginated. Reach out to your Samsara Representative to have this API enabled for your organization.
 
@@ -76,7 +77,7 @@ class RawTripsClient:
 
         Returns
         -------
-        HttpResponse[TripsGetTripsResponseBody]
+        SyncPager[TripResponseBody]
             OK response.
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -95,14 +96,32 @@ class RawTripsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                _data = typing.cast(
+                _parsed_response = typing.cast(
                     TripsGetTripsResponseBody,
                     parse_obj_as(
                         type_=TripsGetTripsResponseBody,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
-                return HttpResponse(response=_response, data=_data)
+                _items = _parsed_response.data
+                _has_next = False
+                _get_next = None
+                if _parsed_response.pagination is not None:
+                    _parsed_next = _parsed_response.pagination.end_cursor
+                    _has_next = _parsed_next is not None and _parsed_next != ""
+                    _get_next = lambda: self.stream(
+                        start_time=start_time,
+                        include_asset=include_asset,
+                        completion_status=completion_status,
+                        end_time=end_time,
+                        query_by=query_by,
+                        after=_parsed_next,
+                        ids=ids,
+                        request_options=request_options,
+                    )
+                return SyncPager(
+                    has_next=_has_next, items=_items, get_next=_get_next, response=BaseHttpResponse(response=_response)
+                )
             if _response.status_code == 401:
                 raise UnauthorizedError(
                     headers=dict(_response.headers),
@@ -223,7 +242,7 @@ class AsyncRawTripsClient:
         after: typing.Optional[str] = None,
         ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[TripsGetTripsResponseBody]:
+    ) -> AsyncPager[TripResponseBody]:
         """
         This endpoint will return trips that have been collected for your organization based on the time parameters passed in. Results are paginated. Reach out to your Samsara Representative to have this API enabled for your organization.
 
@@ -262,7 +281,7 @@ class AsyncRawTripsClient:
 
         Returns
         -------
-        AsyncHttpResponse[TripsGetTripsResponseBody]
+        AsyncPager[TripResponseBody]
             OK response.
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -281,14 +300,35 @@ class AsyncRawTripsClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                _data = typing.cast(
+                _parsed_response = typing.cast(
                     TripsGetTripsResponseBody,
                     parse_obj_as(
                         type_=TripsGetTripsResponseBody,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
-                return AsyncHttpResponse(response=_response, data=_data)
+                _items = _parsed_response.data
+                _has_next = False
+                _get_next = None
+                if _parsed_response.pagination is not None:
+                    _parsed_next = _parsed_response.pagination.end_cursor
+                    _has_next = _parsed_next is not None and _parsed_next != ""
+
+                    async def _get_next():
+                        return await self.stream(
+                            start_time=start_time,
+                            include_asset=include_asset,
+                            completion_status=completion_status,
+                            end_time=end_time,
+                            query_by=query_by,
+                            after=_parsed_next,
+                            ids=ids,
+                            request_options=request_options,
+                        )
+
+                return AsyncPager(
+                    has_next=_has_next, items=_items, get_next=_get_next, response=BaseHttpResponse(response=_response)
+                )
             if _response.status_code == 401:
                 raise UnauthorizedError(
                     headers=dict(_response.headers),