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)
|