lark-billing 0.0.6__py3-none-any.whl

lark/checkout/client.py

@@ -0,0 +1,214 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from ..types.complete_subscription_checkout_response import CompleteSubscriptionCheckoutResponse
+from ..types.create_subscription_checkout_session_response import CreateSubscriptionCheckoutSessionResponse
+from .raw_client import AsyncRawCheckoutClient, RawCheckoutClient
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class CheckoutClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawCheckoutClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawCheckoutClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawCheckoutClient
+        """
+        return self._raw_client
+
+    def create_subscription_checkout_session(
+        self,
+        *,
+        subject_id: str,
+        rate_card_id: str,
+        rate_card_overrides: typing.Optional[typing.Dict[str, str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> CreateSubscriptionCheckoutSessionResponse:
+        """
+        Parameters
+        ----------
+        subject_id : str
+
+        rate_card_id : str
+
+        rate_card_overrides : typing.Optional[typing.Dict[str, str]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CreateSubscriptionCheckoutSessionResponse
+            Successful Response
+
+        Examples
+        --------
+        from lark import Lark
+
+        client = Lark(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.checkout.create_subscription_checkout_session(
+            subject_id="subject_id",
+            rate_card_id="rate_card_id",
+        )
+        """
+        _response = self._raw_client.create_subscription_checkout_session(
+            subject_id=subject_id,
+            rate_card_id=rate_card_id,
+            rate_card_overrides=rate_card_overrides,
+            request_options=request_options,
+        )
+        return _response.data
+
+    def complete_subscription_checkout(
+        self, checkout_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> CompleteSubscriptionCheckoutResponse:
+        """
+        Parameters
+        ----------
+        checkout_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CompleteSubscriptionCheckoutResponse
+            Successful Response
+
+        Examples
+        --------
+        from lark import Lark
+
+        client = Lark(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.checkout.complete_subscription_checkout(
+            checkout_id="checkout_id",
+        )
+        """
+        _response = self._raw_client.complete_subscription_checkout(checkout_id, request_options=request_options)
+        return _response.data
+
+
+class AsyncCheckoutClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawCheckoutClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawCheckoutClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawCheckoutClient
+        """
+        return self._raw_client
+
+    async def create_subscription_checkout_session(
+        self,
+        *,
+        subject_id: str,
+        rate_card_id: str,
+        rate_card_overrides: typing.Optional[typing.Dict[str, str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> CreateSubscriptionCheckoutSessionResponse:
+        """
+        Parameters
+        ----------
+        subject_id : str
+
+        rate_card_id : str
+
+        rate_card_overrides : typing.Optional[typing.Dict[str, str]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CreateSubscriptionCheckoutSessionResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from lark import AsyncLark
+
+        client = AsyncLark(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+
+
+        async def main() -> None:
+            await client.checkout.create_subscription_checkout_session(
+                subject_id="subject_id",
+                rate_card_id="rate_card_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create_subscription_checkout_session(
+            subject_id=subject_id,
+            rate_card_id=rate_card_id,
+            rate_card_overrides=rate_card_overrides,
+            request_options=request_options,
+        )
+        return _response.data
+
+    async def complete_subscription_checkout(
+        self, checkout_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> CompleteSubscriptionCheckoutResponse:
+        """
+        Parameters
+        ----------
+        checkout_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        CompleteSubscriptionCheckoutResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from lark import AsyncLark
+
+        client = AsyncLark(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+
+
+        async def main() -> None:
+            await client.checkout.complete_subscription_checkout(
+                checkout_id="checkout_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.complete_subscription_checkout(checkout_id, request_options=request_options)
+        return _response.data

lark/checkout/raw_client.py

@@ -0,0 +1,252 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.http_response import AsyncHttpResponse, HttpResponse
+from ..core.jsonable_encoder import jsonable_encoder
+from ..core.pydantic_utilities import parse_obj_as
+from ..core.request_options import RequestOptions
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..types.complete_subscription_checkout_response import CompleteSubscriptionCheckoutResponse
+from ..types.create_subscription_checkout_session_response import CreateSubscriptionCheckoutSessionResponse
+from ..types.http_validation_error import HttpValidationError
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class RawCheckoutClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def create_subscription_checkout_session(
+        self,
+        *,
+        subject_id: str,
+        rate_card_id: str,
+        rate_card_overrides: typing.Optional[typing.Dict[str, str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[CreateSubscriptionCheckoutSessionResponse]:
+        """
+        Parameters
+        ----------
+        subject_id : str
+
+        rate_card_id : str
+
+        rate_card_overrides : typing.Optional[typing.Dict[str, str]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[CreateSubscriptionCheckoutSessionResponse]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "checkout",
+            method="POST",
+            json={
+                "subject_id": subject_id,
+                "rate_card_id": rate_card_id,
+                "rate_card_overrides": rate_card_overrides,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    CreateSubscriptionCheckoutSessionResponse,
+                    parse_obj_as(
+                        type_=CreateSubscriptionCheckoutSessionResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        HttpValidationError,
+                        parse_obj_as(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def complete_subscription_checkout(
+        self, checkout_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[CompleteSubscriptionCheckoutResponse]:
+        """
+        Parameters
+        ----------
+        checkout_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[CompleteSubscriptionCheckoutResponse]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"checkout/{jsonable_encoder(checkout_id)}/complete",
+            method="POST",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    CompleteSubscriptionCheckoutResponse,
+                    parse_obj_as(
+                        type_=CompleteSubscriptionCheckoutResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        HttpValidationError,
+                        parse_obj_as(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+
+class AsyncRawCheckoutClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def create_subscription_checkout_session(
+        self,
+        *,
+        subject_id: str,
+        rate_card_id: str,
+        rate_card_overrides: typing.Optional[typing.Dict[str, str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[CreateSubscriptionCheckoutSessionResponse]:
+        """
+        Parameters
+        ----------
+        subject_id : str
+
+        rate_card_id : str
+
+        rate_card_overrides : typing.Optional[typing.Dict[str, str]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[CreateSubscriptionCheckoutSessionResponse]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "checkout",
+            method="POST",
+            json={
+                "subject_id": subject_id,
+                "rate_card_id": rate_card_id,
+                "rate_card_overrides": rate_card_overrides,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    CreateSubscriptionCheckoutSessionResponse,
+                    parse_obj_as(
+                        type_=CreateSubscriptionCheckoutSessionResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        HttpValidationError,
+                        parse_obj_as(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def complete_subscription_checkout(
+        self, checkout_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[CompleteSubscriptionCheckoutResponse]:
+        """
+        Parameters
+        ----------
+        checkout_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[CompleteSubscriptionCheckoutResponse]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"checkout/{jsonable_encoder(checkout_id)}/complete",
+            method="POST",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    CompleteSubscriptionCheckoutResponse,
+                    parse_obj_as(
+                        type_=CompleteSubscriptionCheckoutResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        HttpValidationError,
+                        parse_obj_as(
+                            type_=HttpValidationError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

lark/client.py

@@ -0,0 +1,240 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import httpx
+from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+
+if typing.TYPE_CHECKING:
+    from .checkout.client import AsyncCheckoutClient, CheckoutClient
+    from .feature_access.client import AsyncFeatureAccessClient, FeatureAccessClient
+    from .rate_cards.client import AsyncRateCardsClient, RateCardsClient
+    from .subjects.client import AsyncSubjectsClient, SubjectsClient
+    from .subscriptions.client import AsyncSubscriptionsClient, SubscriptionsClient
+    from .usage_events.client import AsyncUsageEventsClient, UsageEventsClient
+
+
+class Lark:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
+
+    Parameters
+    ----------
+    base_url : str
+        The base url to use for requests from the client.
+
+    api_key : str
+    headers : typing.Optional[typing.Dict[str, str]]
+        Additional headers to send with every request.
+
+    timeout : typing.Optional[float]
+        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
+
+    follow_redirects : typing.Optional[bool]
+        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
+
+    httpx_client : typing.Optional[httpx.Client]
+        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+
+    Examples
+    --------
+    from lark import Lark
+
+    client = Lark(
+        api_key="YOUR_API_KEY",
+        base_url="https://yourhost.com/path/to/api",
+    )
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: str,
+        api_key: str,
+        headers: typing.Optional[typing.Dict[str, str]] = None,
+        timeout: typing.Optional[float] = None,
+        follow_redirects: typing.Optional[bool] = True,
+        httpx_client: typing.Optional[httpx.Client] = None,
+    ):
+        _defaulted_timeout = (
+            timeout if timeout is not None else 60 if httpx_client is None else httpx_client.timeout.read
+        )
+        self._client_wrapper = SyncClientWrapper(
+            base_url=base_url,
+            api_key=api_key,
+            headers=headers,
+            httpx_client=httpx_client
+            if httpx_client is not None
+            else httpx.Client(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
+            if follow_redirects is not None
+            else httpx.Client(timeout=_defaulted_timeout),
+            timeout=_defaulted_timeout,
+        )
+        self._checkout: typing.Optional[CheckoutClient] = None
+        self._rate_cards: typing.Optional[RateCardsClient] = None
+        self._usage_events: typing.Optional[UsageEventsClient] = None
+        self._subscriptions: typing.Optional[SubscriptionsClient] = None
+        self._feature_access: typing.Optional[FeatureAccessClient] = None
+        self._subjects: typing.Optional[SubjectsClient] = None
+
+    @property
+    def checkout(self):
+        if self._checkout is None:
+            from .checkout.client import CheckoutClient  # noqa: E402
+
+            self._checkout = CheckoutClient(client_wrapper=self._client_wrapper)
+        return self._checkout
+
+    @property
+    def rate_cards(self):
+        if self._rate_cards is None:
+            from .rate_cards.client import RateCardsClient  # noqa: E402
+
+            self._rate_cards = RateCardsClient(client_wrapper=self._client_wrapper)
+        return self._rate_cards
+
+    @property
+    def usage_events(self):
+        if self._usage_events is None:
+            from .usage_events.client import UsageEventsClient  # noqa: E402
+
+            self._usage_events = UsageEventsClient(client_wrapper=self._client_wrapper)
+        return self._usage_events
+
+    @property
+    def subscriptions(self):
+        if self._subscriptions is None:
+            from .subscriptions.client import SubscriptionsClient  # noqa: E402
+
+            self._subscriptions = SubscriptionsClient(client_wrapper=self._client_wrapper)
+        return self._subscriptions
+
+    @property
+    def feature_access(self):
+        if self._feature_access is None:
+            from .feature_access.client import FeatureAccessClient  # noqa: E402
+
+            self._feature_access = FeatureAccessClient(client_wrapper=self._client_wrapper)
+        return self._feature_access
+
+    @property
+    def subjects(self):
+        if self._subjects is None:
+            from .subjects.client import SubjectsClient  # noqa: E402
+
+            self._subjects = SubjectsClient(client_wrapper=self._client_wrapper)
+        return self._subjects
+
+
+class AsyncLark:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
+
+    Parameters
+    ----------
+    base_url : str
+        The base url to use for requests from the client.
+
+    api_key : str
+    headers : typing.Optional[typing.Dict[str, str]]
+        Additional headers to send with every request.
+
+    timeout : typing.Optional[float]
+        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
+
+    follow_redirects : typing.Optional[bool]
+        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
+
+    httpx_client : typing.Optional[httpx.AsyncClient]
+        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+
+    Examples
+    --------
+    from lark import AsyncLark
+
+    client = AsyncLark(
+        api_key="YOUR_API_KEY",
+        base_url="https://yourhost.com/path/to/api",
+    )
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: str,
+        api_key: str,
+        headers: typing.Optional[typing.Dict[str, str]] = None,
+        timeout: typing.Optional[float] = None,
+        follow_redirects: typing.Optional[bool] = True,
+        httpx_client: typing.Optional[httpx.AsyncClient] = None,
+    ):
+        _defaulted_timeout = (
+            timeout if timeout is not None else 60 if httpx_client is None else httpx_client.timeout.read
+        )
+        self._client_wrapper = AsyncClientWrapper(
+            base_url=base_url,
+            api_key=api_key,
+            headers=headers,
+            httpx_client=httpx_client
+            if httpx_client is not None
+            else httpx.AsyncClient(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
+            if follow_redirects is not None
+            else httpx.AsyncClient(timeout=_defaulted_timeout),
+            timeout=_defaulted_timeout,
+        )
+        self._checkout: typing.Optional[AsyncCheckoutClient] = None
+        self._rate_cards: typing.Optional[AsyncRateCardsClient] = None
+        self._usage_events: typing.Optional[AsyncUsageEventsClient] = None
+        self._subscriptions: typing.Optional[AsyncSubscriptionsClient] = None
+        self._feature_access: typing.Optional[AsyncFeatureAccessClient] = None
+        self._subjects: typing.Optional[AsyncSubjectsClient] = None
+
+    @property
+    def checkout(self):
+        if self._checkout is None:
+            from .checkout.client import AsyncCheckoutClient  # noqa: E402
+
+            self._checkout = AsyncCheckoutClient(client_wrapper=self._client_wrapper)
+        return self._checkout
+
+    @property
+    def rate_cards(self):
+        if self._rate_cards is None:
+            from .rate_cards.client import AsyncRateCardsClient  # noqa: E402
+
+            self._rate_cards = AsyncRateCardsClient(client_wrapper=self._client_wrapper)
+        return self._rate_cards
+
+    @property
+    def usage_events(self):
+        if self._usage_events is None:
+            from .usage_events.client import AsyncUsageEventsClient  # noqa: E402
+
+            self._usage_events = AsyncUsageEventsClient(client_wrapper=self._client_wrapper)
+        return self._usage_events
+
+    @property
+    def subscriptions(self):
+        if self._subscriptions is None:
+            from .subscriptions.client import AsyncSubscriptionsClient  # noqa: E402
+
+            self._subscriptions = AsyncSubscriptionsClient(client_wrapper=self._client_wrapper)
+        return self._subscriptions
+
+    @property
+    def feature_access(self):
+        if self._feature_access is None:
+            from .feature_access.client import AsyncFeatureAccessClient  # noqa: E402
+
+            self._feature_access = AsyncFeatureAccessClient(client_wrapper=self._client_wrapper)
+        return self._feature_access
+
+    @property
+    def subjects(self):
+        if self._subjects is None:
+            from .subjects.client import AsyncSubjectsClient  # noqa: E402
+
+            self._subjects = AsyncSubjectsClient(client_wrapper=self._client_wrapper)
+        return self._subjects