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

latitudesh_python_sdk/firewalls_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 OptionalNullable, UNSET
 from latitudesh_python_sdk.utils import get_security_from_env
-from typing import Any, Mapping, Optional, Union
+from typing import Any, Dict, List, Mapping, Optional, Union
 
 
 class FirewallsSDK(BaseSDK):
@@ -74,6 +75,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create-firewall",
                 oauth2_scopes=[],
@@ -177,6 +179,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create-firewall",
                 oauth2_scopes=[],
@@ -219,16 +222,20 @@ class FirewallsSDK(BaseSDK):
         self,
         *,
         filter_project: 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.Firewalls:
+    ) -> Optional[models.ListFirewallsResponse]:
         r"""List firewalls
 
         List firewalls
 
         :param filter_project:
+        :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
@@ -246,6 +253,8 @@ class FirewallsSDK(BaseSDK):
 
         request = models.ListFirewallsRequest(
             filter_project=filter_project,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request(
@@ -274,6 +283,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="list-firewalls",
                 oauth2_scopes=[],
@@ -286,8 +296,32 @@ class FirewallsSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.ListFirewallsResponse]:
+            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_project=filter_project,
+                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.Firewalls)
+            return models.ListFirewallsResponse(
+                result=utils.unmarshal_json(http_res.text, models.Firewalls),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise models.APIError(
@@ -312,16 +346,20 @@ class FirewallsSDK(BaseSDK):
         self,
         *,
         filter_project: 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.Firewalls:
+    ) -> Optional[models.ListFirewallsResponse]:
         r"""List firewalls
 
         List firewalls
 
         :param filter_project:
+        :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
@@ -339,6 +377,8 @@ class FirewallsSDK(BaseSDK):
 
         request = models.ListFirewallsRequest(
             filter_project=filter_project,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request_async(
@@ -367,6 +407,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="list-firewalls",
                 oauth2_scopes=[],
@@ -379,8 +420,32 @@ class FirewallsSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.ListFirewallsResponse]:
+            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_project=filter_project,
+                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.Firewalls)
+            return models.ListFirewallsResponse(
+                result=utils.unmarshal_json(http_res.text, models.Firewalls),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise models.APIError(
@@ -460,6 +525,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-firewall",
                 oauth2_scopes=[],
@@ -557,6 +623,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-firewall",
                 oauth2_scopes=[],
@@ -669,6 +736,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update-firewall",
                 oauth2_scopes=[],
@@ -781,6 +849,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update-firewall",
                 oauth2_scopes=[],
@@ -878,6 +947,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete-firewall",
                 oauth2_scopes=[],
@@ -975,6 +1045,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete-firewall",
                 oauth2_scopes=[],
@@ -1089,6 +1160,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create-firewall-assignment",
                 oauth2_scopes=[],
@@ -1205,6 +1277,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create-firewall-assignment",
                 oauth2_scopes=[],
@@ -1249,16 +1322,20 @@ class FirewallsSDK(BaseSDK):
         self,
         *,
         firewall_id: str,
+        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.FirewallServer:
+    ) -> Optional[models.GetFirewallAssignmentsResponse]:
         r"""Firewall Assignments
 
         List servers assigned to a firewall
 
         :param firewall_id: The Firewall ID
+        :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
@@ -1276,6 +1353,8 @@ class FirewallsSDK(BaseSDK):
 
         request = models.GetFirewallAssignmentsRequest(
             firewall_id=firewall_id,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request(
@@ -1304,6 +1383,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-firewall-assignments",
                 oauth2_scopes=[],
@@ -1316,9 +1396,33 @@ class FirewallsSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetFirewallAssignmentsResponse]:
+            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(
+                firewall_id=firewall_id,
+                page_size=page_size,
+                page_number=next_page,
+                retries=retries,
+            )
+
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/vnd.api+json"):
-            return utils.unmarshal_json(http_res.text, models.FirewallServer)
+            return models.GetFirewallAssignmentsResponse(
+                result=utils.unmarshal_json(http_res.text, models.FirewallAssignments),
+                next=next_func,
+            )
         if utils.match_response(http_res, "404", "application/vnd.api+json"):
             response_data = utils.unmarshal_json(http_res.text, models.ErrorObjectData)
             raise models.ErrorObject(data=response_data)
@@ -1346,16 +1450,20 @@ class FirewallsSDK(BaseSDK):
         self,
         *,
         firewall_id: str,
+        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.FirewallServer:
+    ) -> Optional[models.GetFirewallAssignmentsResponse]:
         r"""Firewall Assignments
 
         List servers assigned to a firewall
 
         :param firewall_id: The Firewall ID
+        :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
@@ -1373,6 +1481,8 @@ class FirewallsSDK(BaseSDK):
 
         request = models.GetFirewallAssignmentsRequest(
             firewall_id=firewall_id,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request_async(
@@ -1401,6 +1511,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-firewall-assignments",
                 oauth2_scopes=[],
@@ -1413,9 +1524,33 @@ class FirewallsSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetFirewallAssignmentsResponse]:
+            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(
+                firewall_id=firewall_id,
+                page_size=page_size,
+                page_number=next_page,
+                retries=retries,
+            )
+
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/vnd.api+json"):
-            return utils.unmarshal_json(http_res.text, models.FirewallServer)
+            return models.GetFirewallAssignmentsResponse(
+                result=utils.unmarshal_json(http_res.text, models.FirewallAssignments),
+                next=next_func,
+            )
         if utils.match_response(http_res, "404", "application/vnd.api+json"):
             response_data = utils.unmarshal_json(http_res.text, models.ErrorObjectData)
             raise models.ErrorObject(data=response_data)
@@ -1501,6 +1636,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete-firewall-assignment",
                 oauth2_scopes=[],
@@ -1601,6 +1737,7 @@ class FirewallsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete-firewall-assignment",
                 oauth2_scopes=[],

latitudesh_python_sdk/httpclient.py

@@ -2,7 +2,6 @@
 
 # pyright: reportReturnType = false
 import asyncio
-from concurrent.futures import ThreadPoolExecutor
 from typing_extensions import Protocol, runtime_checkable
 import httpx
 from typing import Any, Optional, Union
@@ -116,21 +115,12 @@ def close_clients(
             pass
 
     if async_client is not None and not async_client_supplied:
-        is_async = False
         try:
-            asyncio.get_running_loop()
-            is_async = True
+            loop = asyncio.get_running_loop()
+            asyncio.run_coroutine_threadsafe(async_client.aclose(), loop)
         except RuntimeError:
-            pass
-
-        try:
-            # If this function is called in an async loop then start another
-            # loop in a separate thread to close the async http client.
-            if is_async:
-                with ThreadPoolExecutor(max_workers=1) as executor:
-                    future = executor.submit(asyncio.run, async_client.aclose())
-                    future.result()
-            else:
+            try:
                 asyncio.run(async_client.aclose())
-        except Exception:
-            pass
+            except RuntimeError:
+                # best effort
+                pass

latitudesh_python_sdk/ipaddresses_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 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 IPAddressesSDK(BaseSDK):
@@ -19,11 +20,13 @@ class IPAddressesSDK(BaseSDK):
         filter_location: Optional[str] = None,
         filter_address: Optional[str] = None,
         extra_fields_ip_addresses: 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.IPAddresses:
+    ) -> Optional[models.GetIpsResponse]:
         r"""List IPs
 
         List all Management and Additional IP Addresses.
@@ -39,6 +42,8 @@ class IPAddressesSDK(BaseSDK):
         :param filter_location: The site slug to filter by
         :param filter_address: The address of IP to filter by starts_with
         :param extra_fields_ip_addresses: The `region` and `server` are provided as extra attributes that is lazy loaded. To request it, just set `extra_fields[ip_addresses]=region,server` 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
@@ -62,6 +67,8 @@ class IPAddressesSDK(BaseSDK):
             filter_location=filter_location,
             filter_address=filter_address,
             extra_fields_ip_addresses=extra_fields_ip_addresses,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request(
@@ -90,6 +97,7 @@ class IPAddressesSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-ips",
                 oauth2_scopes=[],
@@ -102,9 +110,39 @@ class IPAddressesSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetIpsResponse]:
+            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_server=filter_server,
+                filter_project=filter_project,
+                filter_family=filter_family,
+                filter_type=filter_type,
+                filter_location=filter_location,
+                filter_address=filter_address,
+                extra_fields_ip_addresses=extra_fields_ip_addresses,
+                page_size=page_size,
+                page_number=next_page,
+                retries=retries,
+            )
+
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/vnd.api+json"):
-            return utils.unmarshal_json(http_res.text, models.IPAddresses)
+            return models.GetIpsResponse(
+                result=utils.unmarshal_json(http_res.text, models.IPAddresses),
+                next=next_func,
+            )
         if utils.match_response(http_res, "422", "application/vnd.api+json"):
             response_data = utils.unmarshal_json(http_res.text, models.ErrorObjectData)
             raise models.ErrorObject(data=response_data)
@@ -138,11 +176,13 @@ class IPAddressesSDK(BaseSDK):
         filter_location: Optional[str] = None,
         filter_address: Optional[str] = None,
         extra_fields_ip_addresses: 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.IPAddresses:
+    ) -> Optional[models.GetIpsResponse]:
         r"""List IPs
 
         List all Management and Additional IP Addresses.
@@ -158,6 +198,8 @@ class IPAddressesSDK(BaseSDK):
         :param filter_location: The site slug to filter by
         :param filter_address: The address of IP to filter by starts_with
         :param extra_fields_ip_addresses: The `region` and `server` are provided as extra attributes that is lazy loaded. To request it, just set `extra_fields[ip_addresses]=region,server` 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
@@ -181,6 +223,8 @@ class IPAddressesSDK(BaseSDK):
             filter_location=filter_location,
             filter_address=filter_address,
             extra_fields_ip_addresses=extra_fields_ip_addresses,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request_async(
@@ -209,6 +253,7 @@ class IPAddressesSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-ips",
                 oauth2_scopes=[],
@@ -221,9 +266,39 @@ class IPAddressesSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetIpsResponse]:
+            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_server=filter_server,
+                filter_project=filter_project,
+                filter_family=filter_family,
+                filter_type=filter_type,
+                filter_location=filter_location,
+                filter_address=filter_address,
+                extra_fields_ip_addresses=extra_fields_ip_addresses,
+                page_size=page_size,
+                page_number=next_page,
+                retries=retries,
+            )
+
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/vnd.api+json"):
-            return utils.unmarshal_json(http_res.text, models.IPAddresses)
+            return models.GetIpsResponse(
+                result=utils.unmarshal_json(http_res.text, models.IPAddresses),
+                next=next_func,
+            )
         if utils.match_response(http_res, "422", "application/vnd.api+json"):
             response_data = utils.unmarshal_json(http_res.text, models.ErrorObjectData)
             raise models.ErrorObject(data=response_data)
@@ -309,6 +384,7 @@ class IPAddressesSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-ip",
                 oauth2_scopes=[],
@@ -409,6 +485,7 @@ class IPAddressesSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-ip",
                 oauth2_scopes=[],