latitudesh-python-sdk 1.1.0__py3-none-any.whl → 2.0.1__py3-none-any.whl

latitudesh_python_sdk/roles.py

@@ -6,6 +6,7 @@ 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 latitudesh_python_sdk.utils.unmarshal_json_response import unmarshal_json_response
 from typing import Any, Dict, List, Mapping, Optional, Union
 
 
@@ -73,6 +74,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=[],
@@ -107,28 +109,17 @@ class Roles(BaseSDK):
 
         if utils.match_response(http_res, "200", "application/vnd.api+json"):
             return models.GetRolesResponse(
-                result=utils.unmarshal_json(http_res.text, models.GetRolesResponseBody),
+                result=unmarshal_json_response(models.GetRolesResponseBody, http_res),
                 next=next_func,
             )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
-            raise models.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
+            raise models.APIError("API error occurred", http_res, http_res_text)
         if utils.match_response(http_res, "5XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
-            raise models.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
+            raise models.APIError("API error occurred", http_res, http_res_text)
 
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = utils.stream_to_text(http_res)
-        raise models.APIError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+        raise models.APIError("Unexpected response received", http_res)
 
     async def list_async(
         self,
@@ -193,6 +184,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=[],
@@ -227,28 +219,17 @@ class Roles(BaseSDK):
 
         if utils.match_response(http_res, "200", "application/vnd.api+json"):
             return models.GetRolesResponse(
-                result=utils.unmarshal_json(http_res.text, models.GetRolesResponseBody),
+                result=unmarshal_json_response(models.GetRolesResponseBody, http_res),
                 next=next_func,
             )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
-            raise models.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
+            raise models.APIError("API error occurred", http_res, http_res_text)
         if utils.match_response(http_res, "5XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
-            raise models.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
+            raise models.APIError("API error occurred", http_res, http_res_text)
 
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = await utils.stream_to_text_async(http_res)
-        raise models.APIError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+        raise models.APIError("Unexpected response received", http_res)
 
     def get(
         self,
@@ -307,6 +288,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=[],
@@ -321,29 +303,18 @@ class Roles(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/vnd.api+json"):
-            return utils.unmarshal_json(http_res.text, models.Role)
+            return unmarshal_json_response(models.Role, http_res)
         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)
+            response_data = unmarshal_json_response(models.ErrorObjectData, http_res)
+            raise models.ErrorObject(response_data, http_res)
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
-            raise models.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
+            raise models.APIError("API error occurred", http_res, http_res_text)
         if utils.match_response(http_res, "5XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
-            raise models.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
+            raise models.APIError("API error occurred", http_res, http_res_text)
 
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = utils.stream_to_text(http_res)
-        raise models.APIError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+        raise models.APIError("Unexpected response received", http_res)
 
     async def get_async(
         self,
@@ -402,6 +373,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=[],
@@ -416,26 +388,15 @@ class Roles(BaseSDK):
 
         response_data: Any = None
         if utils.match_response(http_res, "200", "application/vnd.api+json"):
-            return utils.unmarshal_json(http_res.text, models.Role)
+            return unmarshal_json_response(models.Role, http_res)
         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)
+            response_data = unmarshal_json_response(models.ErrorObjectData, http_res)
+            raise models.ErrorObject(response_data, http_res)
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
-            raise models.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
+            raise models.APIError("API error occurred", http_res, http_res_text)
         if utils.match_response(http_res, "5XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
-            raise models.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
+            raise models.APIError("API error occurred", http_res, http_res_text)
 
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = await utils.stream_to_text_async(http_res)
-        raise models.APIError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+        raise models.APIError("Unexpected response received", http_res)

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