Fix create user programmatically cookbook

This commit is contained in:
François Voron
2022-01-07 10:31:52 +01:00
parent 89daead99b
commit 8b865f7609
2 changed files with 13 additions and 11 deletions

View File

@ -1,23 +1,25 @@
import contextlib
from app.db import get_user_db
from app.db import get_async_session, get_user_db
from app.models import UserCreate
from app.users import get_user_manager
from fastapi_users.manager import UserAlreadyExists
get_async_session_context = contextlib.asynccontextmanager(get_async_session)
get_user_db_context = contextlib.asynccontextmanager(get_user_db)
get_user_manager_context = contextlib.asynccontextmanager(get_user_manager)
async def create_user(email: str, password: str, is_superuser: bool = False):
try:
async with get_user_db_context() as user_db:
async with get_user_manager_context(user_db) as user_manager:
user = await user_manager.create(
UserCreate(
email=email, password=password, is_superuser=is_superuser
async with get_async_session_context() as session:
async with get_user_db_context(session) as user_db:
async with get_user_manager_context(user_db) as user_manager:
user = await user_manager.create(
UserCreate(
email=email, password=password, is_superuser=is_superuser
)
)
)
print(f"User created {user}")
print(f"User created {user}")
except UserAlreadyExists:
print(f"User {email} already exists")