Initial commit
This commit is contained in:
40
api/app/user/models.py
Normal file
40
api/app/user/models.py
Normal file
@@ -0,0 +1,40 @@
|
||||
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
|
||||
Reference in New Issue
Block a user