latitudesh-python-sdk 1.0.0__py3-none-any.whl → 2.0.0__py3-none-any.whl

latitudesh_python_sdk/operatingsystems_sdk.py

@@ -1,27 +1,32 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from .basesdk import BaseSDK
+from jsonpath import JSONPath
 from latitudesh_python_sdk import models, utils
 from latitudesh_python_sdk._hooks import HookContext
 from latitudesh_python_sdk.types import OptionalNullable, UNSET
 from latitudesh_python_sdk.utils import get_security_from_env
-from typing import Mapping, Optional
+from typing import Any, Dict, List, Mapping, Optional, Union
 
 
 class OperatingSystemsSDK(BaseSDK):
     def list(
         self,
         *,
+        page_size: Optional[int] = 20,
+        page_number: Optional[int] = 1,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.GetPlansOperatingSystemResponseBody:
+    ) -> Optional[models.GetPlansOperatingSystemResponse]:
         r"""List all operating systems available
 
         Lists all operating systems available to deploy and reinstall.
 
 
+        :param page_size: Number of items to return per page
+        :param page_number: Page number to return (starts at 1)
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -36,12 +41,18 @@ class OperatingSystemsSDK(BaseSDK):
             base_url = server_url
         else:
             base_url = self._get_url(base_url, url_variables)
+
+        request = models.GetPlansOperatingSystemRequest(
+            page_size=page_size,
+            page_number=page_number,
+        )
+
         req = self._build_request(
             method="GET",
             path="/plans/operating_systems",
             base_url=base_url,
             url_variables=url_variables,
-            request=None,
+            request=request,
             request_body_required=False,
             request_has_path_params=False,
             request_has_query_params=True,
@@ -62,6 +73,7 @@ class OperatingSystemsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-plans-operating-system",
                 oauth2_scopes=[],
@@ -74,9 +86,32 @@ class OperatingSystemsSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetPlansOperatingSystemResponse]:
+            body = utils.unmarshal_json(http_res.text, Union[Dict[Any, Any], List[Any]])
+            page = request.page_number if not request.page_number is None else 1
+            next_page = page + 1
+
+            if not http_res.text:
+                return None
+            results = JSONPath("$.data").parse(body)
+            if len(results) == 0 or len(results[0]) == 0:
+                return None
+            limit = request.page_size if not request.page_size is None else 20
+            if len(results[0]) < limit:
+                return None
+
+            return self.list(
+                page_size=page_size,
+                page_number=next_page,
+                retries=retries,
+            )
+
         if utils.match_response(http_res, "200", "application/vnd.api+json"):
-            return utils.unmarshal_json(
-                http_res.text, models.GetPlansOperatingSystemResponseBody
+            return models.GetPlansOperatingSystemResponse(
+                result=utils.unmarshal_json(
+                    http_res.text, models.GetPlansOperatingSystemResponseBody
+                ),
+                next=next_func,
             )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
@@ -101,16 +136,20 @@ class OperatingSystemsSDK(BaseSDK):
     async def list_async(
         self,
         *,
+        page_size: Optional[int] = 20,
+        page_number: Optional[int] = 1,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.GetPlansOperatingSystemResponseBody:
+    ) -> Optional[models.GetPlansOperatingSystemResponse]:
         r"""List all operating systems available
 
         Lists all operating systems available to deploy and reinstall.
 
 
+        :param page_size: Number of items to return per page
+        :param page_number: Page number to return (starts at 1)
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -125,12 +164,18 @@ class OperatingSystemsSDK(BaseSDK):
             base_url = server_url
         else:
             base_url = self._get_url(base_url, url_variables)
+
+        request = models.GetPlansOperatingSystemRequest(
+            page_size=page_size,
+            page_number=page_number,
+        )
+
         req = self._build_request_async(
             method="GET",
             path="/plans/operating_systems",
             base_url=base_url,
             url_variables=url_variables,
-            request=None,
+            request=request,
             request_body_required=False,
             request_has_path_params=False,
             request_has_query_params=True,
@@ -151,6 +196,7 @@ class OperatingSystemsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-plans-operating-system",
                 oauth2_scopes=[],
@@ -163,9 +209,32 @@ class OperatingSystemsSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetPlansOperatingSystemResponse]:
+            body = utils.unmarshal_json(http_res.text, Union[Dict[Any, Any], List[Any]])
+            page = request.page_number if not request.page_number is None else 1
+            next_page = page + 1
+
+            if not http_res.text:
+                return None
+            results = JSONPath("$.data").parse(body)
+            if len(results) == 0 or len(results[0]) == 0:
+                return None
+            limit = request.page_size if not request.page_size is None else 20
+            if len(results[0]) < limit:
+                return None
+
+            return self.list(
+                page_size=page_size,
+                page_number=next_page,
+                retries=retries,
+            )
+
         if utils.match_response(http_res, "200", "application/vnd.api+json"):
-            return utils.unmarshal_json(
-                http_res.text, models.GetPlansOperatingSystemResponseBody
+            return models.GetPlansOperatingSystemResponse(
+                result=utils.unmarshal_json(
+                    http_res.text, models.GetPlansOperatingSystemResponseBody
+                ),
+                next=next_func,
             )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)

latitudesh_python_sdk/plans.py

@@ -1,11 +1,12 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from .basesdk import BaseSDK
+from jsonpath import JSONPath
 from latitudesh_python_sdk import models, utils
 from latitudesh_python_sdk._hooks import HookContext
-from latitudesh_python_sdk.types import BaseModel, OptionalNullable, UNSET
+from latitudesh_python_sdk.types import OptionalNullable, UNSET
 from latitudesh_python_sdk.utils import get_security_from_env
-from typing import Any, Mapping, Optional, Union, cast
+from typing import Any, Dict, List, Mapping, Optional, Union
 
 
 class Plans(BaseSDK):
@@ -90,6 +91,7 @@ class Plans(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-plans",
                 oauth2_scopes=[],
@@ -205,6 +207,7 @@ class Plans(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-plans",
                 oauth2_scopes=[],
@@ -296,6 +299,7 @@ class Plans(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-plan",
                 oauth2_scopes=[],
@@ -391,6 +395,7 @@ class Plans(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-plan",
                 oauth2_scopes=[],
@@ -434,17 +439,21 @@ class Plans(BaseSDK):
         *,
         api_version: Optional[str] = "2023-06-01",
         filter_id: Optional[str] = None,
+        page_size: Optional[int] = 20,
+        page_number: Optional[int] = 1,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.BandwidthPlans:
+    ) -> Optional[models.GetBandwidthPlansResponse]:
         r"""List all bandwidth plans
 
         Lists all bandwidth plans.
 
         :param api_version:
         :param filter_id: The plan ID to filter by
+        :param page_size: Number of items to return per page
+        :param page_number: Page number to return (starts at 1)
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -463,6 +472,8 @@ class Plans(BaseSDK):
         request = models.GetBandwidthPlansRequest(
             api_version=api_version,
             filter_id=filter_id,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request(
@@ -491,6 +502,7 @@ class Plans(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-bandwidth-plans",
                 oauth2_scopes=[],
@@ -503,8 +515,33 @@ class Plans(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetBandwidthPlansResponse]:
+            body = utils.unmarshal_json(http_res.text, Union[Dict[Any, Any], List[Any]])
+            page = request.page_number if not request.page_number is None else 1
+            next_page = page + 1
+
+            if not http_res.text:
+                return None
+            results = JSONPath("$.data").parse(body)
+            if len(results) == 0 or len(results[0]) == 0:
+                return None
+            limit = request.page_size if not request.page_size is None else 20
+            if len(results[0]) < limit:
+                return None
+
+            return self.list_bandwidth(
+                api_version=api_version,
+                filter_id=filter_id,
+                page_size=page_size,
+                page_number=next_page,
+                retries=retries,
+            )
+
         if utils.match_response(http_res, "200", "application/vnd.api+json"):
-            return utils.unmarshal_json(http_res.text, models.BandwidthPlans)
+            return models.GetBandwidthPlansResponse(
+                result=utils.unmarshal_json(http_res.text, models.BandwidthPlans),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise models.APIError(
@@ -530,17 +567,21 @@ class Plans(BaseSDK):
         *,
         api_version: Optional[str] = "2023-06-01",
         filter_id: Optional[str] = None,
+        page_size: Optional[int] = 20,
+        page_number: Optional[int] = 1,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.BandwidthPlans:
+    ) -> Optional[models.GetBandwidthPlansResponse]:
         r"""List all bandwidth plans
 
         Lists all bandwidth plans.
 
         :param api_version:
         :param filter_id: The plan ID to filter by
+        :param page_size: Number of items to return per page
+        :param page_number: Page number to return (starts at 1)
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -559,6 +600,8 @@ class Plans(BaseSDK):
         request = models.GetBandwidthPlansRequest(
             api_version=api_version,
             filter_id=filter_id,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request_async(
@@ -587,6 +630,7 @@ class Plans(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-bandwidth-plans",
                 oauth2_scopes=[],
@@ -599,8 +643,33 @@ class Plans(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetBandwidthPlansResponse]:
+            body = utils.unmarshal_json(http_res.text, Union[Dict[Any, Any], List[Any]])
+            page = request.page_number if not request.page_number is None else 1
+            next_page = page + 1
+
+            if not http_res.text:
+                return None
+            results = JSONPath("$.data").parse(body)
+            if len(results) == 0 or len(results[0]) == 0:
+                return None
+            limit = request.page_size if not request.page_size is None else 20
+            if len(results[0]) < limit:
+                return None
+
+            return self.list_bandwidth(
+                api_version=api_version,
+                filter_id=filter_id,
+                page_size=page_size,
+                page_number=next_page,
+                retries=retries,
+            )
+
         if utils.match_response(http_res, "200", "application/vnd.api+json"):
-            return utils.unmarshal_json(http_res.text, models.BandwidthPlans)
+            return models.GetBandwidthPlansResponse(
+                result=utils.unmarshal_json(http_res.text, models.BandwidthPlans),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise models.APIError(
@@ -624,10 +693,12 @@ class Plans(BaseSDK):
     def update_bandwidth(
         self,
         *,
-        request: Union[
-            models.UpdatePlansBandwidthPlansRequestBody,
-            models.UpdatePlansBandwidthPlansRequestBodyTypedDict,
-        ] = models.UpdatePlansBandwidthPlansRequestBody(),
+        data: Optional[
+            Union[
+                models.UpdatePlansBandwidthPlansData,
+                models.UpdatePlansBandwidthPlansDataTypedDict,
+            ]
+        ] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -638,7 +709,7 @@ class Plans(BaseSDK):
         Allow to increase or decrease bandwidth packages. Only admins and owners can request.
 
 
-        :param request: The request object to send.
+        :param data:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -654,11 +725,11 @@ class Plans(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        if not isinstance(request, BaseModel):
-            request = utils.unmarshal(
-                request, models.UpdatePlansBandwidthPlansRequestBody
-            )
-        request = cast(models.UpdatePlansBandwidthPlansRequestBody, request)
+        request = models.UpdatePlansBandwidthPlansRequestBody(
+            data=utils.get_pydantic_model(
+                data, Optional[models.UpdatePlansBandwidthPlansData]
+            ),
+        )
 
         req = self._build_request(
             method="POST",
@@ -676,9 +747,9 @@ class Plans(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request,
                 False,
-                True,
+                False,
                 "json",
-                Optional[models.UpdatePlansBandwidthPlansRequestBody],
+                models.UpdatePlansBandwidthPlansRequestBody,
             ),
             timeout_ms=timeout_ms,
         )
@@ -693,6 +764,7 @@ class Plans(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update-plans-bandwidth",
                 oauth2_scopes=[],
@@ -734,10 +806,12 @@ class Plans(BaseSDK):
     async def update_bandwidth_async(
         self,
         *,
-        request: Union[
-            models.UpdatePlansBandwidthPlansRequestBody,
-            models.UpdatePlansBandwidthPlansRequestBodyTypedDict,
-        ] = models.UpdatePlansBandwidthPlansRequestBody(),
+        data: Optional[
+            Union[
+                models.UpdatePlansBandwidthPlansData,
+                models.UpdatePlansBandwidthPlansDataTypedDict,
+            ]
+        ] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -748,7 +822,7 @@ class Plans(BaseSDK):
         Allow to increase or decrease bandwidth packages. Only admins and owners can request.
 
 
-        :param request: The request object to send.
+        :param data:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -764,11 +838,11 @@ class Plans(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        if not isinstance(request, BaseModel):
-            request = utils.unmarshal(
-                request, models.UpdatePlansBandwidthPlansRequestBody
-            )
-        request = cast(models.UpdatePlansBandwidthPlansRequestBody, request)
+        request = models.UpdatePlansBandwidthPlansRequestBody(
+            data=utils.get_pydantic_model(
+                data, Optional[models.UpdatePlansBandwidthPlansData]
+            ),
+        )
 
         req = self._build_request_async(
             method="POST",
@@ -786,9 +860,9 @@ class Plans(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request,
                 False,
-                True,
+                False,
                 "json",
-                Optional[models.UpdatePlansBandwidthPlansRequestBody],
+                models.UpdatePlansBandwidthPlansRequestBody,
             ),
             timeout_ms=timeout_ms,
         )
@@ -803,6 +877,7 @@ class Plans(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update-plans-bandwidth",
                 oauth2_scopes=[],
@@ -891,6 +966,7 @@ class Plans(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-storage-plans",
                 oauth2_scopes=[],
@@ -975,6 +1051,7 @@ class Plans(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-storage-plans",
                 oauth2_scopes=[],
@@ -1059,6 +1136,7 @@ class Plans(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-vm-plans",
                 oauth2_scopes=[],
@@ -1143,6 +1221,7 @@ class Plans(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-vm-plans",
                 oauth2_scopes=[],