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

latitudesh_python_sdk/privatenetworks.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 PrivateNetworks(BaseSDK):
@@ -15,11 +16,13 @@ class PrivateNetworks(BaseSDK):
         filter_location: Optional[str] = None,
         filter_project: Optional[str] = None,
         filter_tags: 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.VirtualNetworks:
+    ) -> Optional[models.GetVirtualNetworksResponse]:
         r"""List all Virtual Networks
 
         Lists virtual networks assigned to a project
@@ -28,6 +31,8 @@ class PrivateNetworks(BaseSDK):
         :param filter_location: The location slug to filter by
         :param filter_project: The project id or slug 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 ssh keys with `tag_1` AND `tag_2`
+        :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
@@ -47,6 +52,8 @@ class PrivateNetworks(BaseSDK):
             filter_location=filter_location,
             filter_project=filter_project,
             filter_tags=filter_tags,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request(
@@ -75,6 +82,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-virtual-networks",
                 oauth2_scopes=[],
@@ -87,8 +95,34 @@ class PrivateNetworks(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetVirtualNetworksResponse]:
+            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_location=filter_location,
+                filter_project=filter_project,
+                filter_tags=filter_tags,
+                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.VirtualNetworks)
+            return models.GetVirtualNetworksResponse(
+                result=utils.unmarshal_json(http_res.text, models.VirtualNetworks),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise models.APIError(
@@ -115,11 +149,13 @@ class PrivateNetworks(BaseSDK):
         filter_location: Optional[str] = None,
         filter_project: Optional[str] = None,
         filter_tags: 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.VirtualNetworks:
+    ) -> Optional[models.GetVirtualNetworksResponse]:
         r"""List all Virtual Networks
 
         Lists virtual networks assigned to a project
@@ -128,6 +164,8 @@ class PrivateNetworks(BaseSDK):
         :param filter_location: The location slug to filter by
         :param filter_project: The project id or slug 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 ssh keys with `tag_1` AND `tag_2`
+        :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
@@ -147,6 +185,8 @@ class PrivateNetworks(BaseSDK):
             filter_location=filter_location,
             filter_project=filter_project,
             filter_tags=filter_tags,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request_async(
@@ -175,6 +215,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-virtual-networks",
                 oauth2_scopes=[],
@@ -187,8 +228,34 @@ class PrivateNetworks(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetVirtualNetworksResponse]:
+            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_location=filter_location,
+                filter_project=filter_project,
+                filter_tags=filter_tags,
+                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.VirtualNetworks)
+            return models.GetVirtualNetworksResponse(
+                result=utils.unmarshal_json(http_res.text, models.VirtualNetworks),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise models.APIError(
@@ -281,6 +348,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create-virtual-network",
                 oauth2_scopes=[],
@@ -391,6 +459,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create-virtual-network",
                 oauth2_scopes=[],
@@ -509,6 +578,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update-virtual-network",
                 oauth2_scopes=[],
@@ -629,6 +699,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update-virtual-network",
                 oauth2_scopes=[],
@@ -729,6 +800,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="destroy-virtual-network",
                 oauth2_scopes=[],
@@ -827,6 +899,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="destroy-virtual-network",
                 oauth2_scopes=[],
@@ -925,6 +998,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-virtual-network",
                 oauth2_scopes=[],
@@ -1021,6 +1095,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-virtual-network",
                 oauth2_scopes=[],
@@ -1063,11 +1138,13 @@ class PrivateNetworks(BaseSDK):
         filter_server: Optional[str] = None,
         filter_vid: Optional[str] = None,
         filter_virtual_network_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.VirtualNetworkAssignments:
+    ) -> Optional[models.GetVirtualNetworksAssignmentsResponse]:
         r"""List all servers assigned to virtual networks
 
         Returns a list of all servers assigned to virtual networks.
@@ -1076,6 +1153,8 @@ class PrivateNetworks(BaseSDK):
         :param filter_server: The server ID to filter by
         :param filter_vid: The vlan ID to filter by
         :param filter_virtual_network_id: The virtual network 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
@@ -1095,6 +1174,8 @@ class PrivateNetworks(BaseSDK):
             filter_server=filter_server,
             filter_vid=filter_vid,
             filter_virtual_network_id=filter_virtual_network_id,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request(
@@ -1123,6 +1204,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-virtual-networks-assignments",
                 oauth2_scopes=[],
@@ -1135,8 +1217,36 @@ class PrivateNetworks(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetVirtualNetworksAssignmentsResponse]:
+            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_assignments(
+                filter_server=filter_server,
+                filter_vid=filter_vid,
+                filter_virtual_network_id=filter_virtual_network_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.VirtualNetworkAssignments)
+            return models.GetVirtualNetworksAssignmentsResponse(
+                result=utils.unmarshal_json(
+                    http_res.text, models.VirtualNetworkAssignments
+                ),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise models.APIError(
@@ -1163,11 +1273,13 @@ class PrivateNetworks(BaseSDK):
         filter_server: Optional[str] = None,
         filter_vid: Optional[str] = None,
         filter_virtual_network_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.VirtualNetworkAssignments:
+    ) -> Optional[models.GetVirtualNetworksAssignmentsResponse]:
         r"""List all servers assigned to virtual networks
 
         Returns a list of all servers assigned to virtual networks.
@@ -1176,6 +1288,8 @@ class PrivateNetworks(BaseSDK):
         :param filter_server: The server ID to filter by
         :param filter_vid: The vlan ID to filter by
         :param filter_virtual_network_id: The virtual network 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
@@ -1195,6 +1309,8 @@ class PrivateNetworks(BaseSDK):
             filter_server=filter_server,
             filter_vid=filter_vid,
             filter_virtual_network_id=filter_virtual_network_id,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request_async(
@@ -1223,6 +1339,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-virtual-networks-assignments",
                 oauth2_scopes=[],
@@ -1235,8 +1352,36 @@ class PrivateNetworks(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetVirtualNetworksAssignmentsResponse]:
+            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_assignments(
+                filter_server=filter_server,
+                filter_vid=filter_vid,
+                filter_virtual_network_id=filter_virtual_network_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.VirtualNetworkAssignments)
+            return models.GetVirtualNetworksAssignmentsResponse(
+                result=utils.unmarshal_json(
+                    http_res.text, models.VirtualNetworkAssignments
+                ),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise models.APIError(
@@ -1260,10 +1405,12 @@ class PrivateNetworks(BaseSDK):
     def assign(
         self,
         *,
-        request: Union[
-            models.AssignServerVirtualNetworkPrivateNetworksRequestBody,
-            models.AssignServerVirtualNetworkPrivateNetworksRequestBodyTypedDict,
-        ] = models.AssignServerVirtualNetworkPrivateNetworksRequestBody(),
+        data: Optional[
+            Union[
+                models.AssignServerVirtualNetworkPrivateNetworksData,
+                models.AssignServerVirtualNetworkPrivateNetworksDataTypedDict,
+            ]
+        ] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -1271,7 +1418,7 @@ class PrivateNetworks(BaseSDK):
     ) -> models.VirtualNetworkAssignment:
         r"""Assign Virtual network
 
-        :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
@@ -1287,12 +1434,10 @@ class PrivateNetworks(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        if not isinstance(request, BaseModel):
-            request = utils.unmarshal(
-                request, models.AssignServerVirtualNetworkPrivateNetworksRequestBody
-            )
-        request = cast(
-            models.AssignServerVirtualNetworkPrivateNetworksRequestBody, request
+        request = models.AssignServerVirtualNetworkPrivateNetworksRequestBody(
+            data=utils.get_pydantic_model(
+                data, Optional[models.AssignServerVirtualNetworkPrivateNetworksData]
+            ),
         )
 
         req = self._build_request(
@@ -1311,9 +1456,9 @@ class PrivateNetworks(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request,
                 False,
-                True,
+                False,
                 "json",
-                Optional[models.AssignServerVirtualNetworkPrivateNetworksRequestBody],
+                models.AssignServerVirtualNetworkPrivateNetworksRequestBody,
             ),
             timeout_ms=timeout_ms,
         )
@@ -1328,6 +1473,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="assign-server-virtual-network",
                 oauth2_scopes=[],
@@ -1369,10 +1515,12 @@ class PrivateNetworks(BaseSDK):
     async def assign_async(
         self,
         *,
-        request: Union[
-            models.AssignServerVirtualNetworkPrivateNetworksRequestBody,
-            models.AssignServerVirtualNetworkPrivateNetworksRequestBodyTypedDict,
-        ] = models.AssignServerVirtualNetworkPrivateNetworksRequestBody(),
+        data: Optional[
+            Union[
+                models.AssignServerVirtualNetworkPrivateNetworksData,
+                models.AssignServerVirtualNetworkPrivateNetworksDataTypedDict,
+            ]
+        ] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -1380,7 +1528,7 @@ class PrivateNetworks(BaseSDK):
     ) -> models.VirtualNetworkAssignment:
         r"""Assign Virtual network
 
-        :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
@@ -1396,12 +1544,10 @@ class PrivateNetworks(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        if not isinstance(request, BaseModel):
-            request = utils.unmarshal(
-                request, models.AssignServerVirtualNetworkPrivateNetworksRequestBody
-            )
-        request = cast(
-            models.AssignServerVirtualNetworkPrivateNetworksRequestBody, request
+        request = models.AssignServerVirtualNetworkPrivateNetworksRequestBody(
+            data=utils.get_pydantic_model(
+                data, Optional[models.AssignServerVirtualNetworkPrivateNetworksData]
+            ),
         )
 
         req = self._build_request_async(
@@ -1420,9 +1566,9 @@ class PrivateNetworks(BaseSDK):
             get_serialized_body=lambda: utils.serialize_request_body(
                 request,
                 False,
-                True,
+                False,
                 "json",
-                Optional[models.AssignServerVirtualNetworkPrivateNetworksRequestBody],
+                models.AssignServerVirtualNetworkPrivateNetworksRequestBody,
             ),
             timeout_ms=timeout_ms,
         )
@@ -1437,6 +1583,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="assign-server-virtual-network",
                 oauth2_scopes=[],
@@ -1535,6 +1682,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete-virtual-networks-assignments",
                 oauth2_scopes=[],
@@ -1633,6 +1781,7 @@ class PrivateNetworks(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete-virtual-networks-assignments",
                 oauth2_scopes=[],