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

samsara/maintenance/work_orders/raw_client.py

@@ -7,6 +7,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 ...core.serialization import convert_and_respect_annotation_metadata
@@ -22,6 +23,7 @@ from ...errors.unauthorized_error import UnauthorizedError
 from ...types.service_task_instance_input_object_request_body import ServiceTaskInstanceInputObjectRequestBody
 from ...types.work_order_discount_object_request_body import WorkOrderDiscountObjectRequestBody
 from ...types.work_order_item_object_request_body import WorkOrderItemObjectRequestBody
+from ...types.work_order_object_response_body import WorkOrderObjectResponseBody
 from ...types.work_order_tax_object_request_body import WorkOrderTaxObjectRequestBody
 from ...types.work_orders_get_work_orders_response_body import WorkOrdersGetWorkOrdersResponseBody
 from ...types.work_orders_patch_work_orders_response_body import WorkOrdersPatchWorkOrdersResponseBody
@@ -48,7 +50,7 @@ class RawWorkOrdersClient:
         ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         after: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[WorkOrdersGetWorkOrdersResponseBody]:
+    ) -> SyncPager[WorkOrderObjectResponseBody]:
         """
         Gets work orders.
 
@@ -72,7 +74,7 @@ class RawWorkOrdersClient:
 
         Returns
         -------
-        HttpResponse[WorkOrdersGetWorkOrdersResponseBody]
+        SyncPager[WorkOrderObjectResponseBody]
             OK response.
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -86,14 +88,27 @@ class RawWorkOrdersClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                _data = typing.cast(
+                _parsed_response = typing.cast(
                     WorkOrdersGetWorkOrdersResponseBody,
                     parse_obj_as(
                         type_=WorkOrdersGetWorkOrdersResponseBody,  # 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.list(
+                        ids=ids,
+                        after=_parsed_next,
+                        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),
@@ -807,7 +822,7 @@ class RawWorkOrdersClient:
         asset_ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         assigned_user_ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[WorkOrdersStreamWorkOrdersResponseBody]:
+    ) -> SyncPager[WorkOrderObjectResponseBody]:
         """
         Stream work orders.
 
@@ -843,7 +858,7 @@ class RawWorkOrdersClient:
 
         Returns
         -------
-        HttpResponse[WorkOrdersStreamWorkOrdersResponseBody]
+        SyncPager[WorkOrderObjectResponseBody]
             OK response.
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -861,14 +876,31 @@ class RawWorkOrdersClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                _data = typing.cast(
+                _parsed_response = typing.cast(
                     WorkOrdersStreamWorkOrdersResponseBody,
                     parse_obj_as(
                         type_=WorkOrdersStreamWorkOrdersResponseBody,  # 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,
+                        after=_parsed_next,
+                        end_time=end_time,
+                        work_order_statuses=work_order_statuses,
+                        asset_ids=asset_ids,
+                        assigned_user_ids=assigned_user_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),
@@ -984,7 +1016,7 @@ class AsyncRawWorkOrdersClient:
         ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         after: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[WorkOrdersGetWorkOrdersResponseBody]:
+    ) -> AsyncPager[WorkOrderObjectResponseBody]:
         """
         Gets work orders.
 
@@ -1008,7 +1040,7 @@ class AsyncRawWorkOrdersClient:
 
         Returns
         -------
-        AsyncHttpResponse[WorkOrdersGetWorkOrdersResponseBody]
+        AsyncPager[WorkOrderObjectResponseBody]
             OK response.
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1022,14 +1054,30 @@ class AsyncRawWorkOrdersClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                _data = typing.cast(
+                _parsed_response = typing.cast(
                     WorkOrdersGetWorkOrdersResponseBody,
                     parse_obj_as(
                         type_=WorkOrdersGetWorkOrdersResponseBody,  # 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.list(
+                            ids=ids,
+                            after=_parsed_next,
+                            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),
@@ -1745,7 +1793,7 @@ class AsyncRawWorkOrdersClient:
         asset_ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         assigned_user_ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[WorkOrdersStreamWorkOrdersResponseBody]:
+    ) -> AsyncPager[WorkOrderObjectResponseBody]:
         """
         Stream work orders.
 
@@ -1781,7 +1829,7 @@ class AsyncRawWorkOrdersClient:
 
         Returns
         -------
-        AsyncHttpResponse[WorkOrdersStreamWorkOrdersResponseBody]
+        AsyncPager[WorkOrderObjectResponseBody]
             OK response.
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1799,14 +1847,34 @@ class AsyncRawWorkOrdersClient:
         )
         try:
             if 200 <= _response.status_code < 300:
-                _data = typing.cast(
+                _parsed_response = typing.cast(
                     WorkOrdersStreamWorkOrdersResponseBody,
                     parse_obj_as(
                         type_=WorkOrdersStreamWorkOrdersResponseBody,  # 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,
+                            after=_parsed_next,
+                            end_time=end_time,
+                            work_order_statuses=work_order_statuses,
+                            asset_ids=asset_ids,
+                            assigned_user_ids=assigned_user_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),