Source code for arbor_imago.schemas

from pydantic import BaseModel
from typing import Type, Literal, TypeVar

from arbor_imago.models.tables import UserAccessToken, ApiKey, OTP
from arbor_imago.models.models import SignUp




class FromAttributes(BaseModel):


    class Config:
        from_attributes = True




AuthCredential = Type[UserAccessToken] | Type[ApiKey] | Type[OTP] | Type[SignUp]
AuthCredentialType = Literal['access_token', 'api_key', 'otp', 'sign_up']
AuthCredentialInstance = UserAccessToken | ApiKey | OTP | SignUp

AUTH_CREDENTIAL_TYPES: set[AuthCredentialType] = {
    'access_token',
    'api_key',
    'otp',
    'sign_up',
}


AuthCredentialJwt = Type[UserAccessToken] | Type[ApiKey] | Type[SignUp]
AuthCredentialJwtType = Literal['access_token', 'api_key', 'sign_up']
AuthCredentialJwtInstance = UserAccessToken | ApiKey | SignUp

AuthCredentialTable = Type[UserAccessToken] | Type[ApiKey] | Type[OTP]
AuthCredentialTableType = Literal['access_token', 'api_key', 'otp']
AuthCredentialTableInstance = UserAccessToken | ApiKey | OTP

AuthCredentialJwtAndTable = Type[UserAccessToken] | Type[ApiKey]
AuthCredentialJwtAndTableType = Literal['access_token', 'api_key']
AuthCredentialJwtAndTableInstance = UserAccessToken | ApiKey

AuthCredentialJwtAndNotTable = Type[SignUp]
AuthCredentialJwtAndNotTableType = Literal['sign_up']
AuthCredentialJwtAndNotTableInstance = SignUp

PrimaryAuthCredential = Type[UserAccessToken] | Type[ApiKey]
PrimaryAuthCredentialInstance = UserAccessToken | ApiKey


TAuthCredentialTableInstance = TypeVar(
    'TAuthCredentialTableInstance', bound=AuthCredentialTableInstance)


TAuthCredentialInstance_co = TypeVar(
    'TAuthCredentialInstance_co', bound=AuthCredentialInstance, covariant=True)