create_user.py

import asyncio

import typer

from app.main.db import async_session
from app.user.constants import UserRole
from app.user.schemas import NewUser
from app.user.services.users import UsersDbService

app = typer.Typer()


@app.command()
def create_user() -> None:
  """Create a new user."""
  first_name = typer.prompt("First name")
  last_name = typer.prompt("Last name")
  email = typer.prompt("Email")
  username = typer.prompt("Username")
  role = typer.prompt("Role", type=UserRole, default=UserRole.user)
  password = typer.prompt("Password", hide_input=True, confirmation_prompt=True)
  user = NewUser(
    first_name=first_name,
    last_name=last_name,
    email=email,
    username=username,
    password=password,
    repeat_password=password,
  )
  asyncio.run(save_user(user, role))


async def save_user(user: NewUser, role: UserRole) -> None:
  """Save info about user into the database.

  :param user: info about user to be saved
  :return: None
  """
  async with async_session() as session:
    service = UsersDbService(session)
    await service.create(user, role=role.name)


if __name__ == "__main__":
  app()