20 lines
499 B
Python
20 lines
499 B
Python
|
from config import settings
|
||
|
from models import Base
|
||
|
from models.mixins import IdIntPkMixin
|
||
|
|
||
|
from fastapi_users_db_sqlalchemy import (
|
||
|
SQLAlchemyBaseUserTable,
|
||
|
SQLAlchemyUserDatabase,
|
||
|
)
|
||
|
from typing import TYPE_CHECKING
|
||
|
|
||
|
if TYPE_CHECKING:
|
||
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
||
|
|
||
|
class User(Base, IdIntPkMixin, SQLAlchemyBaseUserTable[settings.types.UserIdType]):
|
||
|
|
||
|
@classmethod
|
||
|
def get_db(cls, session: "AsyncSession"):
|
||
|
return SQLAlchemyUserDatabase(session, cls)
|