mirror of
https://github.com/fastapi-users/fastapi-users.git
synced 2025-10-28 03:26:13 +08:00
46 lines
944 B
Python
46 lines
944 B
Python
from typing import Generic, Protocol, TypeVar
|
|
|
|
ID = TypeVar("ID")
|
|
|
|
|
|
class UserProtocol(Protocol[ID]):
|
|
"""User protocol that ORM model should follow."""
|
|
|
|
id: ID
|
|
email: str
|
|
hashed_password: str
|
|
is_active: bool
|
|
is_superuser: bool
|
|
is_verified: bool
|
|
|
|
|
|
class OAuthAccountProtocol(Protocol[ID]):
|
|
"""OAuth account protocol that ORM model should follow."""
|
|
|
|
id: ID
|
|
oauth_name: str
|
|
access_token: str
|
|
expires_at: int | None
|
|
refresh_token: str | None
|
|
account_id: str
|
|
account_email: str
|
|
|
|
|
|
UP = TypeVar("UP", bound=UserProtocol)
|
|
OAP = TypeVar("OAP", bound=OAuthAccountProtocol)
|
|
|
|
|
|
class UserOAuthProtocol(UserProtocol[ID], Generic[ID, OAP]):
|
|
"""User protocol including a list of OAuth accounts."""
|
|
|
|
id: ID
|
|
email: str
|
|
hashed_password: str
|
|
is_active: bool
|
|
is_superuser: bool
|
|
is_verified: bool
|
|
oauth_accounts: list[OAP]
|
|
|
|
|
|
UOAP = TypeVar("UOAP", bound=UserOAuthProtocol)
|