29 lines
689 B
Python
29 lines
689 B
Python
import crud
|
|
import schemas
|
|
from core.config import settings
|
|
|
|
# 创建一个超级用户、、
|
|
from db import connect_to_mongo, get_database
|
|
|
|
connect_to_mongo()
|
|
db = get_database()
|
|
|
|
|
|
async def create_superuser():
|
|
user = await crud.user.get_by_user(db=db, name=settings.FIRST_NAME)
|
|
if not user:
|
|
user_in = schemas.UserCreate(
|
|
name=settings.FIRST_NAME,
|
|
email=settings.FIRST_EMAIL,
|
|
password=settings.FIRST_SUPERUSER_PASSWORD,
|
|
is_superuser=True,
|
|
)
|
|
await crud.user.create(db, user_in)
|
|
await crud.user.create_index(db)
|
|
|
|
|
|
import asyncio
|
|
|
|
loop = asyncio.get_event_loop()
|
|
loop.run_until_complete(create_superuser())
|