Adding fixture system
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user