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

latitudesh_python_sdk/_hooks/types.py

@@ -3,10 +3,12 @@
 from abc import ABC, abstractmethod
 import httpx
 from latitudesh_python_sdk.httpclient import HttpClient
+from latitudesh_python_sdk.sdkconfiguration import SDKConfiguration
 from typing import Any, Callable, List, Optional, Tuple, Union
 
 
 class HookContext:
+    config: SDKConfiguration
     base_url: str
     operation_id: str
     oauth2_scopes: Optional[List[str]] = None
@@ -14,11 +16,13 @@ class HookContext:
 
     def __init__(
         self,
+        config: SDKConfiguration,
         base_url: str,
         operation_id: str,
         oauth2_scopes: Optional[List[str]],
         security_source: Optional[Union[Any, Callable[[], Any]]],
     ):
+        self.config = config
         self.base_url = base_url
         self.operation_id = operation_id
         self.oauth2_scopes = oauth2_scopes
@@ -28,6 +32,7 @@ class HookContext:
 class BeforeRequestContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
         super().__init__(
+            hook_ctx.config,
             hook_ctx.base_url,
             hook_ctx.operation_id,
             hook_ctx.oauth2_scopes,
@@ -38,6 +43,7 @@ class BeforeRequestContext(HookContext):
 class AfterSuccessContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
         super().__init__(
+            hook_ctx.config,
             hook_ctx.base_url,
             hook_ctx.operation_id,
             hook_ctx.oauth2_scopes,
@@ -48,6 +54,7 @@ class AfterSuccessContext(HookContext):
 class AfterErrorContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
         super().__init__(
+            hook_ctx.config,
             hook_ctx.base_url,
             hook_ctx.operation_id,
             hook_ctx.oauth2_scopes,

latitudesh_python_sdk/_version.py

@@ -3,10 +3,12 @@
 import importlib.metadata
 
 __title__: str = "latitudesh-python-sdk"
-__version__: str = "1.0.0"
-__openapi_doc_version__: str = "v3"
-__gen_version__: str = "2.578.0"
-__user_agent__: str = "speakeasy-sdk/python 1.0.0 2.578.0 v3 latitudesh-python-sdk"
+__version__: str = "2.0.0"
+__openapi_doc_version__: str = "2023-06-01"
+__gen_version__: str = "2.640.2"
+__user_agent__: str = (
+    "speakeasy-sdk/python 2.0.0 2.640.2 2023-06-01 latitudesh-python-sdk"
+)
 
 try:
     if __package__ is not None:

latitudesh_python_sdk/apikeys.py

@@ -3,9 +3,9 @@
 from .basesdk import BaseSDK
 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, Mapping, Optional, Union
 
 
 class APIKeys(BaseSDK):
@@ -62,6 +62,7 @@ class APIKeys(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-api-keys",
                 oauth2_scopes=[],
@@ -149,6 +150,7 @@ class APIKeys(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-api-keys",
                 oauth2_scopes=[],
@@ -186,9 +188,7 @@ class APIKeys(BaseSDK):
     def create(
         self,
         *,
-        request: Union[
-            models.CreateAPIKey, models.CreateAPIKeyTypedDict
-        ] = models.CreateAPIKey(),
+        data: Optional[Union[models.Data, models.DataTypedDict]] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -199,7 +199,7 @@ class APIKeys(BaseSDK):
         Create a new API Key that is tied to the current user account. The created API key is only listed ONCE upon creation. It can however be regenerated or deleted.
 
 
-        :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
@@ -215,9 +215,9 @@ class APIKeys(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        if not isinstance(request, BaseModel):
-            request = utils.unmarshal(request, models.CreateAPIKey)
-        request = cast(models.CreateAPIKey, request)
+        request = models.CreateAPIKey(
+            data=utils.get_pydantic_model(data, Optional[models.Data]),
+        )
 
         req = self._build_request(
             method="POST",
@@ -233,7 +233,7 @@ class APIKeys(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request, False, True, "json", Optional[models.CreateAPIKey]
+                request, False, False, "json", models.CreateAPIKey
             ),
             timeout_ms=timeout_ms,
         )
@@ -248,6 +248,7 @@ class APIKeys(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="post-api-key",
                 oauth2_scopes=[],
@@ -289,9 +290,7 @@ class APIKeys(BaseSDK):
     async def create_async(
         self,
         *,
-        request: Union[
-            models.CreateAPIKey, models.CreateAPIKeyTypedDict
-        ] = models.CreateAPIKey(),
+        data: Optional[Union[models.Data, models.DataTypedDict]] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -302,7 +301,7 @@ class APIKeys(BaseSDK):
         Create a new API Key that is tied to the current user account. The created API key is only listed ONCE upon creation. It can however be regenerated or deleted.
 
 
-        :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
@@ -318,9 +317,9 @@ class APIKeys(BaseSDK):
         else:
             base_url = self._get_url(base_url, url_variables)
 
-        if not isinstance(request, BaseModel):
-            request = utils.unmarshal(request, models.CreateAPIKey)
-        request = cast(models.CreateAPIKey, request)
+        request = models.CreateAPIKey(
+            data=utils.get_pydantic_model(data, Optional[models.Data]),
+        )
 
         req = self._build_request_async(
             method="POST",
@@ -336,7 +335,7 @@ class APIKeys(BaseSDK):
             http_headers=http_headers,
             security=self.sdk_configuration.security,
             get_serialized_body=lambda: utils.serialize_request_body(
-                request, False, True, "json", Optional[models.CreateAPIKey]
+                request, False, False, "json", models.CreateAPIKey
             ),
             timeout_ms=timeout_ms,
         )
@@ -351,6 +350,7 @@ class APIKeys(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="post-api-key",
                 oauth2_scopes=[],
@@ -459,6 +459,7 @@ class APIKeys(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update-api-key",
                 oauth2_scopes=[],
@@ -567,6 +568,7 @@ class APIKeys(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update-api-key",
                 oauth2_scopes=[],
@@ -665,6 +667,7 @@ class APIKeys(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete-api-key",
                 oauth2_scopes=[],
@@ -763,6 +766,7 @@ class APIKeys(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete-api-key",
                 oauth2_scopes=[],

latitudesh_python_sdk/basesdk.py

@@ -222,12 +222,12 @@ class BaseSDK:
         client = self.sdk_configuration.client
         logger = self.sdk_configuration.debug_logger
 
+        hooks = self.sdk_configuration.__dict__["_hooks"]
+
         def do():
             http_res = None
             try:
-                req = self.sdk_configuration.get_hooks().before_request(
-                    BeforeRequestContext(hook_ctx), request
-                )
+                req = hooks.before_request(BeforeRequestContext(hook_ctx), request)
                 logger.debug(
                     "Request:\nMethod: %s\nURL: %s\nHeaders: %s\nBody: %s",
                     req.method,
@@ -241,9 +241,7 @@ class BaseSDK:
 
                 http_res = client.send(req, stream=stream)
             except Exception as e:
-                _, e = self.sdk_configuration.get_hooks().after_error(
-                    AfterErrorContext(hook_ctx), None, e
-                )
+                _, e = hooks.after_error(AfterErrorContext(hook_ctx), None, e)
                 if e is not None:
                     logger.debug("Request Exception", exc_info=True)
                     raise e
@@ -261,7 +259,7 @@ class BaseSDK:
             )
 
             if utils.match_status_codes(error_status_codes, http_res.status_code):
-                result, err = self.sdk_configuration.get_hooks().after_error(
+                result, err = hooks.after_error(
                     AfterErrorContext(hook_ctx), http_res, None
                 )
                 if err is not None:
@@ -281,9 +279,7 @@ class BaseSDK:
             http_res = do()
 
         if not utils.match_status_codes(error_status_codes, http_res.status_code):
-            http_res = self.sdk_configuration.get_hooks().after_success(
-                AfterSuccessContext(hook_ctx), http_res
-            )
+            http_res = hooks.after_success(AfterSuccessContext(hook_ctx), http_res)
 
         return http_res
 
@@ -298,12 +294,12 @@ class BaseSDK:
         client = self.sdk_configuration.async_client
         logger = self.sdk_configuration.debug_logger
 
+        hooks = self.sdk_configuration.__dict__["_hooks"]
+
         async def do():
             http_res = None
             try:
-                req = self.sdk_configuration.get_hooks().before_request(
-                    BeforeRequestContext(hook_ctx), request
-                )
+                req = hooks.before_request(BeforeRequestContext(hook_ctx), request)
                 logger.debug(
                     "Request:\nMethod: %s\nURL: %s\nHeaders: %s\nBody: %s",
                     req.method,
@@ -317,9 +313,7 @@ class BaseSDK:
 
                 http_res = await client.send(req, stream=stream)
             except Exception as e:
-                _, e = self.sdk_configuration.get_hooks().after_error(
-                    AfterErrorContext(hook_ctx), None, e
-                )
+                _, e = hooks.after_error(AfterErrorContext(hook_ctx), None, e)
                 if e is not None:
                     logger.debug("Request Exception", exc_info=True)
                     raise e
@@ -337,7 +331,7 @@ class BaseSDK:
             )
 
             if utils.match_status_codes(error_status_codes, http_res.status_code):
-                result, err = self.sdk_configuration.get_hooks().after_error(
+                result, err = hooks.after_error(
                     AfterErrorContext(hook_ctx), http_res, None
                 )
                 if err is not None:
@@ -359,8 +353,6 @@ class BaseSDK:
             http_res = await do()
 
         if not utils.match_status_codes(error_status_codes, http_res.status_code):
-            http_res = self.sdk_configuration.get_hooks().after_success(
-                AfterSuccessContext(hook_ctx), http_res
-            )
+            http_res = hooks.after_success(AfterSuccessContext(hook_ctx), http_res)
 
         return http_res

latitudesh_python_sdk/billing.py

@@ -75,6 +75,7 @@ class Billing(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-billing-usage",
                 oauth2_scopes=[],
@@ -175,6 +176,7 @@ class Billing(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-billing-usage",
                 oauth2_scopes=[],

latitudesh_python_sdk/events_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 List, Mapping, Optional
+from typing import Any, Dict, List, Mapping, Optional, Union
 
 
 class EventsSDK(BaseSDK):
@@ -20,11 +21,13 @@ class EventsSDK(BaseSDK):
         filter_created_at_gte: Optional[str] = None,
         filter_created_at_lte: Optional[str] = None,
         filter_created_at: Optional[List[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.GetEventsResponseBody:
+    ) -> Optional[models.GetEventsResponse]:
         r"""List all Events
 
         Lists all events.
@@ -38,6 +41,8 @@ class EventsSDK(BaseSDK):
         :param filter_created_at_gte: The created at greater than equal date to filter by, in ISO formatting (yyyy-MM-dd'T'HH:mm:ss)
         :param filter_created_at_lte: The created at less than equal date to filter by, in ISO formatting (yyyy-MM-dd'T'HH:mm:ss)
         :param filter_created_at: The created at between date range date1, date2 (inclusive) to filter by, in ISO formatting (yyyy-MM-dd'T'HH:mm:ss)
+        :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 EventsSDK(BaseSDK):
             filter_created_at_gte=filter_created_at_gte,
             filter_created_at_lte=filter_created_at_lte,
             filter_created_at=filter_created_at,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request(
@@ -90,6 +97,7 @@ class EventsSDK(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-events",
                 oauth2_scopes=[],
@@ -102,8 +110,41 @@ class EventsSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetEventsResponse]:
+            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_author=filter_author,
+                filter_project=filter_project,
+                filter_target_name=filter_target_name,
+                filter_target_id=filter_target_id,
+                filter_action=filter_action,
+                filter_created_at_gte=filter_created_at_gte,
+                filter_created_at_lte=filter_created_at_lte,
+                filter_created_at=filter_created_at,
+                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.GetEventsResponseBody)
+            return models.GetEventsResponse(
+                result=utils.unmarshal_json(
+                    http_res.text, models.GetEventsResponseBody
+                ),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise models.APIError(
@@ -135,11 +176,13 @@ class EventsSDK(BaseSDK):
         filter_created_at_gte: Optional[str] = None,
         filter_created_at_lte: Optional[str] = None,
         filter_created_at: Optional[List[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.GetEventsResponseBody:
+    ) -> Optional[models.GetEventsResponse]:
         r"""List all Events
 
         Lists all events.
@@ -153,6 +196,8 @@ class EventsSDK(BaseSDK):
         :param filter_created_at_gte: The created at greater than equal date to filter by, in ISO formatting (yyyy-MM-dd'T'HH:mm:ss)
         :param filter_created_at_lte: The created at less than equal date to filter by, in ISO formatting (yyyy-MM-dd'T'HH:mm:ss)
         :param filter_created_at: The created at between date range date1, date2 (inclusive) to filter by, in ISO formatting (yyyy-MM-dd'T'HH:mm:ss)
+        :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
@@ -177,6 +222,8 @@ class EventsSDK(BaseSDK):
             filter_created_at_gte=filter_created_at_gte,
             filter_created_at_lte=filter_created_at_lte,
             filter_created_at=filter_created_at,
+            page_size=page_size,
+            page_number=page_number,
         )
 
         req = self._build_request_async(
@@ -205,6 +252,7 @@ class EventsSDK(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get-events",
                 oauth2_scopes=[],
@@ -217,8 +265,41 @@ class EventsSDK(BaseSDK):
             retry_config=retry_config,
         )
 
+        def next_func() -> Optional[models.GetEventsResponse]:
+            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_author=filter_author,
+                filter_project=filter_project,
+                filter_target_name=filter_target_name,
+                filter_target_id=filter_target_id,
+                filter_action=filter_action,
+                filter_created_at_gte=filter_created_at_gte,
+                filter_created_at_lte=filter_created_at_lte,
+                filter_created_at=filter_created_at,
+                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.GetEventsResponseBody)
+            return models.GetEventsResponse(
+                result=utils.unmarshal_json(
+                    http_res.text, models.GetEventsResponseBody
+                ),
+                next=next_func,
+            )
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise models.APIError(