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

latitudesh_python_sdk/projects_sdk.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 ProjectsSDK(BaseSDK):
@@ -19,11 +20,13 @@ class ProjectsSDK(BaseSDK):
         filter_environment: Optional[str] = None,
         filter_tags: Optional[str] = None,
         extra_fields_projects: 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.Projects:
+    ) -> Optional[models.GetProjectsResponse]:
         r"""List all Projects
 
         Returns a list of all projects for the current team
@@ -36,6 +39,8 @@ class ProjectsSDK(BaseSDK):
         :param filter_environment: The environment to filter by
         :param filter_tags: The tags ids to filter by, separated by comma, e.g. `filter[tags]=tag_1,tag_2`will return projects with `tag_1` AND `tag_2`
         :param extra_fields_projects: The `last_renewal_date` and `next_renewal_date` are provided as extra attributes that show previous and future billing cycle dates. To request it, just set `extra_fields[projects]=last_renewal_date,next_renewal_date` in the query string.
+        :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
@@ -59,6 +64,8 @@ class ProjectsSDK(BaseSDK):
             filter_environment=filter_environment,
             filter_tags=filter_tags,
             extra_fields_projects=extra_fields_projects,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request(
@@ -87,6 +94,7 @@ class ProjectsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-projects",
                 oauth2_scopes=[],
@@ -99,8 +107,38 @@ class ProjectsSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetProjectsResponse]:
+            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(
+                filter_name=filter_name,
+                filter_slug=filter_slug,
+                filter_description=filter_description,
+                filter_billing_type=filter_billing_type,
+                filter_environment=filter_environment,
+                filter_tags=filter_tags,
+                extra_fields_projects=extra_fields_projects,
+                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.Projects)
+            return models.GetProjectsResponse(
+                result=utils.unmarshal_json(http_res.text, models.Projects),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise models.APIError(
@@ -131,11 +169,13 @@ class ProjectsSDK(BaseSDK):
         filter_environment: Optional[str] = None,
         filter_tags: Optional[str] = None,
         extra_fields_projects: 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.Projects:
+    ) -> Optional[models.GetProjectsResponse]:
         r"""List all Projects
 
         Returns a list of all projects for the current team
@@ -148,6 +188,8 @@ class ProjectsSDK(BaseSDK):
         :param filter_environment: The environment to filter by
         :param filter_tags: The tags ids to filter by, separated by comma, e.g. `filter[tags]=tag_1,tag_2`will return projects with `tag_1` AND `tag_2`
         :param extra_fields_projects: The `last_renewal_date` and `next_renewal_date` are provided as extra attributes that show previous and future billing cycle dates. To request it, just set `extra_fields[projects]=last_renewal_date,next_renewal_date` in the query string.
+        :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
@@ -171,6 +213,8 @@ class ProjectsSDK(BaseSDK):
             filter_environment=filter_environment,
             filter_tags=filter_tags,
             extra_fields_projects=extra_fields_projects,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request_async(
@@ -199,6 +243,7 @@ class ProjectsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-projects",
                 oauth2_scopes=[],
@@ -211,8 +256,38 @@ class ProjectsSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetProjectsResponse]:
+            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(
+                filter_name=filter_name,
+                filter_slug=filter_slug,
+                filter_description=filter_description,
+                filter_billing_type=filter_billing_type,
+                filter_environment=filter_environment,
+                filter_tags=filter_tags,
+                extra_fields_projects=extra_fields_projects,
+                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.Projects)
+            return models.GetProjectsResponse(
+                result=utils.unmarshal_json(http_res.text, models.Projects),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise models.APIError(
@@ -236,10 +311,12 @@ class ProjectsSDK(BaseSDK):
     def create(
         self,
         *,
-        request: Union[
-            models.CreateProjectProjectsRequestBody,
-            models.CreateProjectProjectsRequestBodyTypedDict,
-        ] = models.CreateProjectProjectsRequestBody(),
+        data: Optional[
+            Union[
+                models.CreateProjectProjectsData,
+                models.CreateProjectProjectsDataTypedDict,
+            ]
+        ] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -247,7 +324,7 @@ class ProjectsSDK(BaseSDK):
     ) -> models.CreateProjectResponseBody:
         r"""Create a Project
 
-        :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
@@ -263,9 +340,11 @@ class ProjectsSDK(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        if not isinstance(request, BaseModel):
-            request = utils.unmarshal(request, models.CreateProjectProjectsRequestBody)
-        request = cast(models.CreateProjectProjectsRequestBody, request)
+        request = models.CreateProjectProjectsRequestBody(
+            data=utils.get_pydantic_model(
+                data, Optional[models.CreateProjectProjectsData]
+            ),
+        )
 
         req = self._build_request(
             method="POST",
@@ -281,11 +360,7 @@ class ProjectsSDK(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request,
-                False,
-                True,
-                "json",
-                Optional[models.CreateProjectProjectsRequestBody],
+                request, False, False, "json", models.CreateProjectProjectsRequestBody
             ),
             timeout_ms=timeout_ms,
         )
@@ -300,6 +375,7 @@ class ProjectsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create-project",
                 oauth2_scopes=[],
@@ -343,10 +419,12 @@ class ProjectsSDK(BaseSDK):
     async def create_async(
         self,
         *,
-        request: Union[
-            models.CreateProjectProjectsRequestBody,
-            models.CreateProjectProjectsRequestBodyTypedDict,
-        ] = models.CreateProjectProjectsRequestBody(),
+        data: Optional[
+            Union[
+                models.CreateProjectProjectsData,
+                models.CreateProjectProjectsDataTypedDict,
+            ]
+        ] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -354,7 +432,7 @@ class ProjectsSDK(BaseSDK):
     ) -> models.CreateProjectResponseBody:
         r"""Create a Project
 
-        :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
@@ -370,9 +448,11 @@ class ProjectsSDK(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        if not isinstance(request, BaseModel):
-            request = utils.unmarshal(request, models.CreateProjectProjectsRequestBody)
-        request = cast(models.CreateProjectProjectsRequestBody, request)
+        request = models.CreateProjectProjectsRequestBody(
+            data=utils.get_pydantic_model(
+                data, Optional[models.CreateProjectProjectsData]
+            ),
+        )
 
         req = self._build_request_async(
             method="POST",
@@ -388,11 +468,7 @@ class ProjectsSDK(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request,
-                False,
-                True,
-                "json",
-                Optional[models.CreateProjectProjectsRequestBody],
+                request, False, False, "json", models.CreateProjectProjectsRequestBody
             ),
             timeout_ms=timeout_ms,
         )
@@ -407,6 +483,7 @@ class ProjectsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create-project",
                 oauth2_scopes=[],
@@ -518,6 +595,7 @@ class ProjectsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update-project",
                 oauth2_scopes=[],
@@ -629,6 +707,7 @@ class ProjectsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update-project",
                 oauth2_scopes=[],
@@ -726,6 +805,7 @@ class ProjectsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete-project",
                 oauth2_scopes=[],
@@ -823,6 +903,7 @@ class ProjectsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete-project",
                 oauth2_scopes=[],

latitudesh_python_sdk/regions_sdk.py

@@ -1,28 +1,33 @@
 """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 Any, Mapping, Optional
+from typing import Any, Dict, List, Mapping, Optional, Union
 
 
 class RegionsSDK(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.Regions:
+    ) -> Optional[models.GetRegionsResponse]:
         r"""List all Regions
 
         Lists all [available locations](https://latitude.sh/locations). For server availability by location, please see the [Plans API](/reference/get-plans).
 
 
 
+        :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
@@ -37,12 +42,18 @@ class RegionsSDK(BaseSDK):
             base_url = server_url
         else:
             base_url = self._get_url(base_url, url_variables)
+
+        request = models.GetRegionsRequest(
+            page_size=page_size,
+            page_number=page_number,
+        )
+
         req = self._build_request(
             method="GET",
             path="/regions",
             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,
@@ -63,6 +74,7 @@ class RegionsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-regions",
                 oauth2_scopes=[],
@@ -75,8 +87,31 @@ class RegionsSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetRegionsResponse]:
+            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.Regions)
+            return models.GetRegionsResponse(
+                result=utils.unmarshal_json(http_res.text, models.Regions),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise models.APIError(
@@ -100,17 +135,21 @@ class RegionsSDK(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.Regions:
+    ) -> Optional[models.GetRegionsResponse]:
         r"""List all Regions
 
         Lists all [available locations](https://latitude.sh/locations). For server availability by location, please see the [Plans API](/reference/get-plans).
 
 
 
+        :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 RegionsSDK(BaseSDK):
             base_url = server_url
         else:
             base_url = self._get_url(base_url, url_variables)
+
+        request = models.GetRegionsRequest(
+            page_size=page_size,
+            page_number=page_number,
+        )
+
         req = self._build_request_async(
             method="GET",
             path="/regions",
             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 RegionsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-regions",
                 oauth2_scopes=[],
@@ -163,8 +209,31 @@ class RegionsSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetRegionsResponse]:
+            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.Regions)
+            return models.GetRegionsResponse(
+                result=utils.unmarshal_json(http_res.text, models.Regions),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise models.APIError(
@@ -242,6 +311,7 @@ class RegionsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-region",
                 oauth2_scopes=[],
@@ -337,6 +407,7 @@ class RegionsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-region",
                 oauth2_scopes=[],