client.py

# This file was auto-generated by Fern from our API Definition.

from __future__ import annotations

import typing

from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
from ..core.request_options import RequestOptions
from ..types.get_practice_team_response import GetPracticeTeamResponse
from ..types.register_user_response import RegisterUserResponse
from .raw_client import AsyncRawTeamClient, RawTeamClient
from .types.register_user_request_address import RegisterUserRequestAddress
from .types.register_user_request_licenses_item import RegisterUserRequestLicensesItem
from .types.register_user_request_profile_details import RegisterUserRequestProfileDetails
from .types.register_user_request_role import RegisterUserRequestRole
from .types.register_user_request_roles_item import RegisterUserRequestRolesItem

if typing.TYPE_CHECKING:
    from .invitations.client import AsyncInvitationsClient, InvitationsClient
    from .members.client import AsyncMembersClient, MembersClient
    from .prescribers.client import AsyncPrescribersClient, PrescribersClient
# this is used as the default value for optional parameters
OMIT = typing.cast(typing.Any, ...)


class TeamClient:
    def __init__(self, *, client_wrapper: SyncClientWrapper):
        self._raw_client = RawTeamClient(client_wrapper=client_wrapper)
        self._client_wrapper = client_wrapper
        self._invitations: typing.Optional[InvitationsClient] = None
        self._members: typing.Optional[MembersClient] = None
        self._prescribers: typing.Optional[PrescribersClient] = None

    @property
    def with_raw_response(self) -> RawTeamClient:
        """
        Retrieves a raw implementation of this client that returns raw responses.

        Returns
        -------
        RawTeamClient
        """
        return self._raw_client

    def register(
        self,
        practice_id: str,
        *,
        idempotency_key: str,
        external_id: str,
        email: str,
        name: str,
        role: RegisterUserRequestRole,
        identity_attestation: bool,
        roles: typing.Optional[typing.Sequence[RegisterUserRequestRolesItem]] = OMIT,
        profile_details: typing.Optional[RegisterUserRequestProfileDetails] = OMIT,
        npi: typing.Optional[str] = OMIT,
        licenses: typing.Optional[typing.Sequence[RegisterUserRequestLicensesItem]] = OMIT,
        legal_name: typing.Optional[str] = OMIT,
        display_name: typing.Optional[str] = OMIT,
        credentials: typing.Optional[str] = OMIT,
        address: typing.Optional[RegisterUserRequestAddress] = OMIT,
        phone: typing.Optional[str] = OMIT,
        location_ids: typing.Optional[typing.Sequence[str]] = OMIT,
        request_options: typing.Optional[RequestOptions] = None,
    ) -> RegisterUserResponse:
        """
        Requires team:write and Idempotency-Key. Registers a practice member without an invitation. Test requires synthetic .test emails and Affinity Test NPIs. Live requires approved integration and practice access. Identity attestation records the integration's assertion; it does not verify login email or clinical credentials. Existing memberships and verified provider records are preserved. Use the returned user ID for orders and signing.

        Parameters
        ----------
        practice_id : str

        idempotency_key : str

        external_id : str

        email : str

        name : str

        role : RegisterUserRequestRole

        identity_attestation : bool

        roles : typing.Optional[typing.Sequence[RegisterUserRequestRolesItem]]

        profile_details : typing.Optional[RegisterUserRequestProfileDetails]

        npi : typing.Optional[str]

        licenses : typing.Optional[typing.Sequence[RegisterUserRequestLicensesItem]]

        legal_name : typing.Optional[str]

        display_name : typing.Optional[str]

        credentials : typing.Optional[str]

        address : typing.Optional[RegisterUserRequestAddress]

        phone : typing.Optional[str]

        location_ids : typing.Optional[typing.Sequence[str]]

        request_options : typing.Optional[RequestOptions]
            Request-specific configuration.

        Returns
        -------
        RegisterUserResponse
            HTTP 200

        Examples
        --------
        from affinity import Affinity

        client = Affinity(
            affinity_version="YOUR_AFFINITY_VERSION",
            api_key="YOUR_API_KEY",
        )
        client.team.register(
            practice_id="prac_01j2y8m6jcc9tt24af5pw9x1bc",
            idempotency_key="Idempotency-Key",
            external_id="externalId",
            email="email",
            name="name",
            role="administrator",
            identity_attestation=True,
        )
        """
        _response = self._raw_client.register(
            practice_id,
            idempotency_key=idempotency_key,
            external_id=external_id,
            email=email,
            name=name,
            role=role,
            identity_attestation=identity_attestation,
            roles=roles,
            profile_details=profile_details,
            npi=npi,
            licenses=licenses,
            legal_name=legal_name,
            display_name=display_name,
            credentials=credentials,
            address=address,
            phone=phone,
            location_ids=location_ids,
            request_options=request_options,
        )
        return _response.data

    def get(
        self, practice_id: str, *, request_options: typing.Optional[RequestOptions] = None
    ) -> GetPracticeTeamResponse:
        """
        Requires team:read. Returns counts of members, invitations, and prescribers. Use the paginated members, prescribers, and invitations collections for individual records. Team access and clinician credentials are shared between Test and Live.

        Parameters
        ----------
        practice_id : str

        request_options : typing.Optional[RequestOptions]
            Request-specific configuration.

        Returns
        -------
        GetPracticeTeamResponse
            HTTP 200

        Examples
        --------
        from affinity import Affinity

        client = Affinity(
            affinity_version="YOUR_AFFINITY_VERSION",
            api_key="YOUR_API_KEY",
        )
        client.team.get(
            practice_id="prac_01j2y8m6jcc9tt24af5pw9x1bc",
        )
        """
        _response = self._raw_client.get(practice_id, request_options=request_options)
        return _response.data

    @property
    def invitations(self):
        if self._invitations is None:
            from .invitations.client import InvitationsClient  # noqa: E402

            self._invitations = InvitationsClient(client_wrapper=self._client_wrapper)
        return self._invitations

    @property
    def members(self):
        if self._members is None:
            from .members.client import MembersClient  # noqa: E402

            self._members = MembersClient(client_wrapper=self._client_wrapper)
        return self._members

    @property
    def prescribers(self):
        if self._prescribers is None:
            from .prescribers.client import PrescribersClient  # noqa: E402

            self._prescribers = PrescribersClient(client_wrapper=self._client_wrapper)
        return self._prescribers


class AsyncTeamClient:
    def __init__(self, *, client_wrapper: AsyncClientWrapper):
        self._raw_client = AsyncRawTeamClient(client_wrapper=client_wrapper)
        self._client_wrapper = client_wrapper
        self._invitations: typing.Optional[AsyncInvitationsClient] = None
        self._members: typing.Optional[AsyncMembersClient] = None
        self._prescribers: typing.Optional[AsyncPrescribersClient] = None

    @property
    def with_raw_response(self) -> AsyncRawTeamClient:
        """
        Retrieves a raw implementation of this client that returns raw responses.

        Returns
        -------
        AsyncRawTeamClient
        """
        return self._raw_client

    async def register(
        self,
        practice_id: str,
        *,
        idempotency_key: str,
        external_id: str,
        email: str,
        name: str,
        role: RegisterUserRequestRole,
        identity_attestation: bool,
        roles: typing.Optional[typing.Sequence[RegisterUserRequestRolesItem]] = OMIT,
        profile_details: typing.Optional[RegisterUserRequestProfileDetails] = OMIT,
        npi: typing.Optional[str] = OMIT,
        licenses: typing.Optional[typing.Sequence[RegisterUserRequestLicensesItem]] = OMIT,
        legal_name: typing.Optional[str] = OMIT,
        display_name: typing.Optional[str] = OMIT,
        credentials: typing.Optional[str] = OMIT,
        address: typing.Optional[RegisterUserRequestAddress] = OMIT,
        phone: typing.Optional[str] = OMIT,
        location_ids: typing.Optional[typing.Sequence[str]] = OMIT,
        request_options: typing.Optional[RequestOptions] = None,
    ) -> RegisterUserResponse:
        """
        Requires team:write and Idempotency-Key. Registers a practice member without an invitation. Test requires synthetic .test emails and Affinity Test NPIs. Live requires approved integration and practice access. Identity attestation records the integration's assertion; it does not verify login email or clinical credentials. Existing memberships and verified provider records are preserved. Use the returned user ID for orders and signing.

        Parameters
        ----------
        practice_id : str

        idempotency_key : str

        external_id : str

        email : str

        name : str

        role : RegisterUserRequestRole

        identity_attestation : bool

        roles : typing.Optional[typing.Sequence[RegisterUserRequestRolesItem]]

        profile_details : typing.Optional[RegisterUserRequestProfileDetails]

        npi : typing.Optional[str]

        licenses : typing.Optional[typing.Sequence[RegisterUserRequestLicensesItem]]

        legal_name : typing.Optional[str]

        display_name : typing.Optional[str]

        credentials : typing.Optional[str]

        address : typing.Optional[RegisterUserRequestAddress]

        phone : typing.Optional[str]

        location_ids : typing.Optional[typing.Sequence[str]]

        request_options : typing.Optional[RequestOptions]
            Request-specific configuration.

        Returns
        -------
        RegisterUserResponse
            HTTP 200

        Examples
        --------
        import asyncio

        from affinity import AsyncAffinity

        client = AsyncAffinity(
            affinity_version="YOUR_AFFINITY_VERSION",
            api_key="YOUR_API_KEY",
        )


        async def main() -> None:
            await client.team.register(
                practice_id="prac_01j2y8m6jcc9tt24af5pw9x1bc",
                idempotency_key="Idempotency-Key",
                external_id="externalId",
                email="email",
                name="name",
                role="administrator",
                identity_attestation=True,
            )


        asyncio.run(main())
        """
        _response = await self._raw_client.register(
            practice_id,
            idempotency_key=idempotency_key,
            external_id=external_id,
            email=email,
            name=name,
            role=role,
            identity_attestation=identity_attestation,
            roles=roles,
            profile_details=profile_details,
            npi=npi,
            licenses=licenses,
            legal_name=legal_name,
            display_name=display_name,
            credentials=credentials,
            address=address,
            phone=phone,
            location_ids=location_ids,
            request_options=request_options,
        )
        return _response.data

    async def get(
        self, practice_id: str, *, request_options: typing.Optional[RequestOptions] = None
    ) -> GetPracticeTeamResponse:
        """
        Requires team:read. Returns counts of members, invitations, and prescribers. Use the paginated members, prescribers, and invitations collections for individual records. Team access and clinician credentials are shared between Test and Live.

        Parameters
        ----------
        practice_id : str

        request_options : typing.Optional[RequestOptions]
            Request-specific configuration.

        Returns
        -------
        GetPracticeTeamResponse
            HTTP 200

        Examples
        --------
        import asyncio

        from affinity import AsyncAffinity

        client = AsyncAffinity(
            affinity_version="YOUR_AFFINITY_VERSION",
            api_key="YOUR_API_KEY",
        )


        async def main() -> None:
            await client.team.get(
                practice_id="prac_01j2y8m6jcc9tt24af5pw9x1bc",
            )


        asyncio.run(main())
        """
        _response = await self._raw_client.get(practice_id, request_options=request_options)
        return _response.data

    @property
    def invitations(self):
        if self._invitations is None:
            from .invitations.client import AsyncInvitationsClient  # noqa: E402

            self._invitations = AsyncInvitationsClient(client_wrapper=self._client_wrapper)
        return self._invitations

    @property
    def members(self):
        if self._members is None:
            from .members.client import AsyncMembersClient  # noqa: E402

            self._members = AsyncMembersClient(client_wrapper=self._client_wrapper)
        return self._members

    @property
    def prescribers(self):
        if self._prescribers is None:
            from .prescribers.client import AsyncPrescribersClient  # noqa: E402

            self._prescribers = AsyncPrescribersClient(client_wrapper=self._client_wrapper)
        return self._prescribers