autoassembler 0.3.1__py3-none-any.whl

autoassembler/auth/raw_client.py

@@ -0,0 +1,413 @@
+# 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.parse_error import ParsingError
+from ..core.pydantic_utilities import parse_obj_as
+from ..core.request_options import RequestOptions
+from ..errors.bad_request_error import BadRequestError
+from ..errors.internal_server_error import InternalServerError
+from ..errors.unauthorized_error import UnauthorizedError
+from ..types.access_token_response import AccessTokenResponse
+from ..types.device_authorization_response import DeviceAuthorizationResponse
+from ..types.o_auth_error_response import OAuthErrorResponse
+from ..types.user_info_response import UserInfoResponse
+from .types.create_access_token_request_grant_type import CreateAccessTokenRequestGrantType
+from pydantic import ValidationError
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class RawAuthClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def create_device_authorization(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[DeviceAuthorizationResponse]:
+        """
+        Starts the OAuth 2.0 device authorization grant (RFC 8628). Show the user `verification_uri_complete`, then poll `POST /oauth/token` with `device_code` every `interval` seconds until it returns a token or a terminal error.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[DeviceAuthorizationResponse]
+            Successful initiation of device authorization
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "oauth/device/code",
+            method="POST",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    DeviceAuthorizationResponse,
+                    parse_obj_as(
+                        type_=DeviceAuthorizationResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def create_access_token(
+        self,
+        *,
+        grant_type: CreateAccessTokenRequestGrantType,
+        device_code: typing.Optional[str] = OMIT,
+        client_id: typing.Optional[str] = OMIT,
+        client_secret: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[AccessTokenResponse]:
+        """
+        Follows OAuth 2.0 token endpoint (RFC 6749 §5). Device flow §4.3.3; client_credentials grant §4.4. Send the resulting token as `Authorization: Bearer <token>`.
+
+        Parameters
+        ----------
+        grant_type : CreateAccessTokenRequestGrantType
+            `client_credentials` uses an API key for machine-to-machine access. `urn:ietf:params:oauth:grant-type:device_code` completes the device authorization flow.
+
+        device_code : typing.Optional[str]
+            Code returned by `POST /oauth/device/code`. Required for the device authorization flow.
+
+        client_id : typing.Optional[str]
+            API key ID. Required for machine-to-machine access.
+
+        client_secret : typing.Optional[str]
+            API key secret shown once when the key was created. Required for machine-to-machine access.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[AccessTokenResponse]
+            Access token granted. Send it as `Authorization: Bearer <token>`.
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "oauth/token",
+            method="POST",
+            data={
+                "grant_type": grant_type,
+                "device_code": device_code,
+                "client_id": client_id,
+                "client_secret": client_secret,
+            },
+            headers={
+                "content-type": "application/x-www-form-urlencoded",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    AccessTokenResponse,
+                    parse_obj_as(
+                        type_=AccessTokenResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        OAuthErrorResponse,
+                        parse_obj_as(
+                            type_=OAuthErrorResponse,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def get_current_user(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[UserInfoResponse]:
+        """
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[UserInfoResponse]
+            Successful retrieval of user info
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v4/user",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    UserInfoResponse,
+                    parse_obj_as(
+                        type_=UserInfoResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+
+class AsyncRawAuthClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def create_device_authorization(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[DeviceAuthorizationResponse]:
+        """
+        Starts the OAuth 2.0 device authorization grant (RFC 8628). Show the user `verification_uri_complete`, then poll `POST /oauth/token` with `device_code` every `interval` seconds until it returns a token or a terminal error.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[DeviceAuthorizationResponse]
+            Successful initiation of device authorization
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "oauth/device/code",
+            method="POST",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    DeviceAuthorizationResponse,
+                    parse_obj_as(
+                        type_=DeviceAuthorizationResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def create_access_token(
+        self,
+        *,
+        grant_type: CreateAccessTokenRequestGrantType,
+        device_code: typing.Optional[str] = OMIT,
+        client_id: typing.Optional[str] = OMIT,
+        client_secret: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[AccessTokenResponse]:
+        """
+        Follows OAuth 2.0 token endpoint (RFC 6749 §5). Device flow §4.3.3; client_credentials grant §4.4. Send the resulting token as `Authorization: Bearer <token>`.
+
+        Parameters
+        ----------
+        grant_type : CreateAccessTokenRequestGrantType
+            `client_credentials` uses an API key for machine-to-machine access. `urn:ietf:params:oauth:grant-type:device_code` completes the device authorization flow.
+
+        device_code : typing.Optional[str]
+            Code returned by `POST /oauth/device/code`. Required for the device authorization flow.
+
+        client_id : typing.Optional[str]
+            API key ID. Required for machine-to-machine access.
+
+        client_secret : typing.Optional[str]
+            API key secret shown once when the key was created. Required for machine-to-machine access.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[AccessTokenResponse]
+            Access token granted. Send it as `Authorization: Bearer <token>`.
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "oauth/token",
+            method="POST",
+            data={
+                "grant_type": grant_type,
+                "device_code": device_code,
+                "client_id": client_id,
+                "client_secret": client_secret,
+            },
+            headers={
+                "content-type": "application/x-www-form-urlencoded",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    AccessTokenResponse,
+                    parse_obj_as(
+                        type_=AccessTokenResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        OAuthErrorResponse,
+                        parse_obj_as(
+                            type_=OAuthErrorResponse,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def get_current_user(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[UserInfoResponse]:
+        """
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[UserInfoResponse]
+            Successful retrieval of user info
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v4/user",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    UserInfoResponse,
+                    parse_obj_as(
+                        type_=UserInfoResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

autoassembler/auth/types/__init__.py

@@ -0,0 +1,36 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .create_access_token_request_grant_type import CreateAccessTokenRequestGrantType
+_dynamic_imports: typing.Dict[str, str] = {
+    "CreateAccessTokenRequestGrantType": ".create_access_token_request_grant_type"
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = ["CreateAccessTokenRequestGrantType"]

autoassembler/auth/types/create_access_token_request_grant_type.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+CreateAccessTokenRequestGrantType = typing.Union[
+    typing.Literal["client_credentials", "urn:ietf:params:oauth:grant-type:device_code"], typing.Any
+]