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

latitudesh_python_sdk/roles.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 Any, Mapping, Optional
+from typing import Any, Dict, List, Mapping, Optional, Union
 
 
 class Roles(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.GetRolesResponseBody:
+    ) -> Optional[models.GetRolesResponse]:
         r"""List all Roles
 
         Returns a list of all roles that can be assigned to users
 
 
+        :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 Roles(BaseSDK):
             base_url = server_url
         else:
             base_url = self._get_url(base_url, url_variables)
+
+        request = models.GetRolesRequest(
+            page_size=page_size,
+            page_number=page_number,
+        )
+
         req = self._build_request(
             method="GET",
             path="/roles",
             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 Roles(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-roles",
                 oauth2_scopes=[],
@@ -74,8 +86,31 @@ class Roles(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetRolesResponse]:
+            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.GetRolesResponseBody)
+            return models.GetRolesResponse(
+                result=utils.unmarshal_json(http_res.text, models.GetRolesResponseBody),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise models.APIError(
@@ -99,16 +134,20 @@ class Roles(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.GetRolesResponseBody:
+    ) -> Optional[models.GetRolesResponse]:
         r"""List all Roles
 
         Returns a list of all roles that can be assigned to users
 
 
+        :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
@@ -123,12 +162,18 @@ class Roles(BaseSDK):
             base_url = server_url
         else:
             base_url = self._get_url(base_url, url_variables)
+
+        request = models.GetRolesRequest(
+            page_size=page_size,
+            page_number=page_number,
+        )
+
         req = self._build_request_async(
             method="GET",
             path="/roles",
             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,
@@ -149,6 +194,7 @@ class Roles(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-roles",
                 oauth2_scopes=[],
@@ -161,8 +207,31 @@ class Roles(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetRolesResponse]:
+            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.GetRolesResponseBody)
+            return models.GetRolesResponse(
+                result=utils.unmarshal_json(http_res.text, models.GetRolesResponseBody),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise models.APIError(
@@ -240,6 +309,7 @@ class Roles(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-role-id",
                 oauth2_scopes=[],
@@ -335,6 +405,7 @@ class Roles(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-role-id",
                 oauth2_scopes=[],

latitudesh_python_sdk/sdk.py

@@ -6,60 +6,96 @@ from .sdkconfiguration import SDKConfiguration
 from .utils.logger import Logger, get_default_logger
 from .utils.retries import RetryConfig
 import httpx
+import importlib
 from latitudesh_python_sdk import models, utils
 from latitudesh_python_sdk._hooks import SDKHooks
-from latitudesh_python_sdk.apikeys import APIKeys
-from latitudesh_python_sdk.billing import Billing
-from latitudesh_python_sdk.events_sdk import EventsSDK
-from latitudesh_python_sdk.firewalls_sdk import FirewallsSDK
-from latitudesh_python_sdk.ipaddresses_sdk import IPAddressesSDK
-from latitudesh_python_sdk.operatingsystems_sdk import OperatingSystemsSDK
-from latitudesh_python_sdk.plans import Plans
-from latitudesh_python_sdk.privatenetworks import PrivateNetworks
-from latitudesh_python_sdk.projects_sdk import ProjectsSDK
-from latitudesh_python_sdk.regions_sdk import RegionsSDK
-from latitudesh_python_sdk.roles import Roles
-from latitudesh_python_sdk.servers_sdk import ServersSDK
-from latitudesh_python_sdk.sshkeys import SSHKeys
-from latitudesh_python_sdk.storage import Storage
-from latitudesh_python_sdk.tags import Tags
-from latitudesh_python_sdk.teams_sdk import TeamsSDK
-from latitudesh_python_sdk.teamsmembers import TeamsMembers
-from latitudesh_python_sdk.traffic_sdk import TrafficSDK
 from latitudesh_python_sdk.types import OptionalNullable, UNSET
-from latitudesh_python_sdk.userdata_sdk import UserDataSDK
-from latitudesh_python_sdk.userprofile import UserProfile
-from latitudesh_python_sdk.virtualmachines import VirtualMachines
-from latitudesh_python_sdk.vpnsessions import VpnSessions
-from typing import Any, Callable, Dict, List, Optional, Union, cast
+from typing import Any, Callable, Dict, List, Optional, TYPE_CHECKING, Union, cast
 import weakref
 
+if TYPE_CHECKING:
+    from latitudesh_python_sdk.apikeys import APIKeys
+    from latitudesh_python_sdk.billing import Billing
+    from latitudesh_python_sdk.events_sdk import EventsSDK
+    from latitudesh_python_sdk.firewalls_sdk import FirewallsSDK
+    from latitudesh_python_sdk.ipaddresses_sdk import IPAddressesSDK
+    from latitudesh_python_sdk.operatingsystems_sdk import OperatingSystemsSDK
+    from latitudesh_python_sdk.plans import Plans
+    from latitudesh_python_sdk.privatenetworks import PrivateNetworks
+    from latitudesh_python_sdk.projects_sdk import ProjectsSDK
+    from latitudesh_python_sdk.regions_sdk import RegionsSDK
+    from latitudesh_python_sdk.roles import Roles
+    from latitudesh_python_sdk.servers_sdk import ServersSDK
+    from latitudesh_python_sdk.sshkeys_sdk import SSHKeysSDK
+    from latitudesh_python_sdk.storage import Storage
+    from latitudesh_python_sdk.tags import Tags
+    from latitudesh_python_sdk.teams_sdk import TeamsSDK
+    from latitudesh_python_sdk.teamsmembers import TeamsMembers
+    from latitudesh_python_sdk.traffic_sdk import TrafficSDK
+    from latitudesh_python_sdk.userdata_sdk import UserDataSDK
+    from latitudesh_python_sdk.userprofile import UserProfile
+    from latitudesh_python_sdk.virtualmachines import VirtualMachines
+    from latitudesh_python_sdk.vpnsessions import VpnSessions
+
 
 class Latitudesh(BaseSDK):
     r"""Latitude.sh API: The Latitude.sh API is a RESTful API to manage your Latitude.sh account. It allows you to perform the same actions as the Latitude.sh dashboard."""
 
-    api_keys: APIKeys
-    billing: Billing
-    events: EventsSDK
-    firewalls: FirewallsSDK
-    ip_addresses: IPAddressesSDK
-    teams_members: TeamsMembers
-    operating_systems: OperatingSystemsSDK
-    plans: Plans
-    projects: ProjectsSDK
-    regions: RegionsSDK
-    roles: Roles
-    servers: ServersSDK
-    ssh_keys: SSHKeys
-    storage: Storage
-    tags: Tags
-    teams: TeamsSDK
-    traffic: TrafficSDK
-    user_data: UserDataSDK
-    user_profile: UserProfile
-    virtual_machines: VirtualMachines
-    private_networks: PrivateNetworks
-    vpn_sessions: VpnSessions
+    api_keys: "APIKeys"
+    billing: "Billing"
+    events: "EventsSDK"
+    firewalls: "FirewallsSDK"
+    ip_addresses: "IPAddressesSDK"
+    teams_members: "TeamsMembers"
+    operating_systems: "OperatingSystemsSDK"
+    plans: "Plans"
+    projects: "ProjectsSDK"
+    regions: "RegionsSDK"
+    roles: "Roles"
+    servers: "ServersSDK"
+    ssh_keys: "SSHKeysSDK"
+    storage: "Storage"
+    tags: "Tags"
+    teams: "TeamsSDK"
+    traffic: "TrafficSDK"
+    user_data: "UserDataSDK"
+    user_profile: "UserProfile"
+    virtual_machines: "VirtualMachines"
+    private_networks: "PrivateNetworks"
+    vpn_sessions: "VpnSessions"
+    _sub_sdk_map = {
+        "api_keys": ("latitudesh_python_sdk.apikeys", "APIKeys"),
+        "billing": ("latitudesh_python_sdk.billing", "Billing"),
+        "events": ("latitudesh_python_sdk.events_sdk", "EventsSDK"),
+        "firewalls": ("latitudesh_python_sdk.firewalls_sdk", "FirewallsSDK"),
+        "ip_addresses": ("latitudesh_python_sdk.ipaddresses_sdk", "IPAddressesSDK"),
+        "teams_members": ("latitudesh_python_sdk.teamsmembers", "TeamsMembers"),
+        "operating_systems": (
+            "latitudesh_python_sdk.operatingsystems_sdk",
+            "OperatingSystemsSDK",
+        ),
+        "plans": ("latitudesh_python_sdk.plans", "Plans"),
+        "projects": ("latitudesh_python_sdk.projects_sdk", "ProjectsSDK"),
+        "regions": ("latitudesh_python_sdk.regions_sdk", "RegionsSDK"),
+        "roles": ("latitudesh_python_sdk.roles", "Roles"),
+        "servers": ("latitudesh_python_sdk.servers_sdk", "ServersSDK"),
+        "ssh_keys": ("latitudesh_python_sdk.sshkeys_sdk", "SSHKeysSDK"),
+        "storage": ("latitudesh_python_sdk.storage", "Storage"),
+        "tags": ("latitudesh_python_sdk.tags", "Tags"),
+        "teams": ("latitudesh_python_sdk.teams_sdk", "TeamsSDK"),
+        "traffic": ("latitudesh_python_sdk.traffic_sdk", "TrafficSDK"),
+        "user_data": ("latitudesh_python_sdk.userdata_sdk", "UserDataSDK"),
+        "user_profile": ("latitudesh_python_sdk.userprofile", "UserProfile"),
+        "virtual_machines": (
+            "latitudesh_python_sdk.virtualmachines",
+            "VirtualMachines",
+        ),
+        "private_networks": (
+            "latitudesh_python_sdk.privatenetworks",
+            "PrivateNetworks",
+        ),
+        "vpn_sessions": ("latitudesh_python_sdk.vpnsessions", "VpnSessions"),
+    }
 
     def __init__(
         self,
@@ -145,6 +181,9 @@ class Latitudesh(BaseSDK):
 
         hooks = SDKHooks()
 
+        # pylint: disable=protected-access
+        self.sdk_configuration.__dict__["_hooks"] = hooks
+
         current_server_url, *_ = self.sdk_configuration.get_server_details()
         server_url, self.sdk_configuration.client = hooks.sdk_init(
             current_server_url, client
@@ -152,9 +191,6 @@ class Latitudesh(BaseSDK):
         if current_server_url != server_url:
             self.sdk_configuration.server_url = server_url
 
-        # pylint: disable=protected-access
-        self.sdk_configuration.__dict__["_hooks"] = hooks
-
         weakref.finalize(
             self,
             close_clients,
@@ -165,31 +201,32 @@ class Latitudesh(BaseSDK):
             self.sdk_configuration.async_client_supplied,
         )
 
-        self._init_sdks()
-
-    def _init_sdks(self):
-        self.api_keys = APIKeys(self.sdk_configuration)
-        self.billing = Billing(self.sdk_configuration)
-        self.events = EventsSDK(self.sdk_configuration)
-        self.firewalls = FirewallsSDK(self.sdk_configuration)
-        self.ip_addresses = IPAddressesSDK(self.sdk_configuration)
-        self.teams_members = TeamsMembers(self.sdk_configuration)
-        self.operating_systems = OperatingSystemsSDK(self.sdk_configuration)
-        self.plans = Plans(self.sdk_configuration)
-        self.projects = ProjectsSDK(self.sdk_configuration)
-        self.regions = RegionsSDK(self.sdk_configuration)
-        self.roles = Roles(self.sdk_configuration)
-        self.servers = ServersSDK(self.sdk_configuration)
-        self.ssh_keys = SSHKeys(self.sdk_configuration)
-        self.storage = Storage(self.sdk_configuration)
-        self.tags = Tags(self.sdk_configuration)
-        self.teams = TeamsSDK(self.sdk_configuration)
-        self.traffic = TrafficSDK(self.sdk_configuration)
-        self.user_data = UserDataSDK(self.sdk_configuration)
-        self.user_profile = UserProfile(self.sdk_configuration)
-        self.virtual_machines = VirtualMachines(self.sdk_configuration)
-        self.private_networks = PrivateNetworks(self.sdk_configuration)
-        self.vpn_sessions = VpnSessions(self.sdk_configuration)
+    def __getattr__(self, name: str):
+        if name in self._sub_sdk_map:
+            module_path, class_name = self._sub_sdk_map[name]
+            try:
+                module = importlib.import_module(module_path)
+                klass = getattr(module, class_name)
+                instance = klass(self.sdk_configuration)
+                setattr(self, name, instance)
+                return instance
+            except ImportError as e:
+                raise AttributeError(
+                    f"Failed to import module {module_path} for attribute {name}: {e}"
+                ) from e
+            except AttributeError as e:
+                raise AttributeError(
+                    f"Failed to find class {class_name} in module {module_path} for attribute {name}: {e}"
+                ) from e
+
+        raise AttributeError(
+            f"'{type(self).__name__}' object has no attribute '{name}'"
+        )
+
+    def __dir__(self):
+        default_attrs = list(super().__dir__())
+        lazy_attrs = list(self._sub_sdk_map.keys())
+        return sorted(list(set(default_attrs + lazy_attrs)))
 
     def __enter__(self):
         return self

latitudesh_python_sdk/sdkconfiguration.py

@@ -1,6 +1,5 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
-from ._hooks import SDKHooks
 from ._version import (
     __gen_version__,
     __openapi_doc_version__,
@@ -42,9 +41,6 @@ class SDKConfiguration:
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None
 
-    def __post_init__(self):
-        self._hooks = SDKHooks()
-
     def get_server_details(self) -> Tuple[str, Dict[str, str]]:
         if self.server_url is not None and self.server_url:
             return remove_suffix(self.server_url, "/"), {}
@@ -52,6 +48,3 @@ class SDKConfiguration:
             self.server_idx = 0
 
         return SERVERS[self.server_idx], self.server_defaults[self.server_idx]
-
-    def get_hooks(self) -> SDKHooks:
-        return self._hooks