41 lines
975 B
Python
41 lines
975 B
Python
import uuid
|
|
|
|
from sqlmodel import select
|
|
from fastapi import Depends
|
|
from fastapi_users import schemas
|
|
from fastapi_users_db_sqlmodel import SQLModelBaseUserDB, SQLModelUserDatabase
|
|
from fastapi_users_db_sqlmodel.access_token import SQLModelBaseAccessToken, SQLModelAccessTokenDatabase
|
|
|
|
from db import get_session, SessionDep
|
|
|
|
class User(SQLModelBaseUserDB, table=True):
|
|
pass
|
|
|
|
class UserDatabase(SQLModelUserDatabase):
|
|
def list(self):
|
|
return select(self.user_model)
|
|
|
|
async def get_user_db(session: SessionDep):
|
|
yield UserDatabase(session, User)
|
|
|
|
class AccessToken(SQLModelBaseAccessToken, table=True):
|
|
pass
|
|
|
|
class AccessTokenDatabase(SQLModelAccessTokenDatabase):
|
|
pass
|
|
|
|
async def get_access_token_db(session = Depends(get_session)):
|
|
yield AccessTokenDatabase(session, AccessToken)
|
|
|
|
|
|
class UserRead(schemas.BaseUser[uuid.UUID]):
|
|
pass
|
|
|
|
|
|
class UserCreate(schemas.BaseUserCreate):
|
|
pass
|
|
|
|
|
|
class UserUpdate(schemas.BaseUserUpdate):
|
|
pass
|