Adding fixture system

This commit is contained in:
2025-02-10 22:23:47 +01:00
parent 39c4ab9102
commit ed6be838fe
5 changed files with 192 additions and 10 deletions

View File

@@ -2,11 +2,11 @@ import uuid
from sqlmodel import select
from fastapi import Depends
from fastapi_users import BaseUserManager, FastAPIUsers, UUIDIDMixin, models, exceptions, schemas
from fastapi_users import BaseUserManager, FastAPIUsers, UUIDIDMixin
from fastapi_users.authentication import BearerTransport, AuthenticationBackend
from fastapi_users.authentication.strategy.db import AccessTokenDatabase, DatabaseStrategy
from user.models import User, get_user_db, AccessToken, get_access_token_db, UserRead, UserUpdate, UserCreate
from user.models import User, get_user_db, AccessToken, get_access_token_db
from db import get_session
SECRET = "SECRET"
@@ -17,7 +17,6 @@ bearer_transport = BearerTransport(tokenUrl="auth/login")
class UserManager(UUIDIDMixin, BaseUserManager[User, uuid.UUID]):
pass
async def get_user_manager(user_db=Depends(get_user_db)) -> UserManager:
yield UserManager(user_db)
@@ -37,20 +36,14 @@ fastapi_users = FastAPIUsers[User, uuid.UUID](
[auth_backend],
)
get_current_user = fastapi_users.current_user(active=True)
get_current_superuser = fastapi_users.current_user(active=True, superuser=True)
#user_router = fastapi_users.get_users_router(UserRead, UserUpdate)
#user_router.include_router(fastapi_users.get_reset_password_router())
reset_password_router = fastapi_users.get_reset_password_router()
auth_router = fastapi_users.get_auth_router(auth_backend)
def create_admin_account():
session = get_session().__next__()
def create_admin_account(session=get_session().__next__()):
admin_email = 'root@root.fr'
statement = select(User).where(User.email == admin_email).limit(1)
admin_user = session.exec(statement).first()