summaryrefslogtreecommitdiff
path: root/animism-align/cli/commands/admin/createuser.py
blob: ef90f1ed217c31e2f856bbd06b647f401f83bd32 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import click

@click.command('createuser')
@click.pass_context
def cli(ctx):
  from getpass import getpass
  from app.utils.auth_utils import encrypt_password
  from app.sql.common import db, Session, User

  username = input("Username: ")
  session = Session()
  user_exists = session.query(User).filter(User.username == username).first()
  if user_exists:
    session.close()
    raise ValueError("User already exists")

  password = encrypt_password(getpass())
  is_admin = input("Is admin? (y/n): ") == 'y'

  user = User(
    username=username,
    password=password,
    is_admin=is_admin,
    settings={}
    )
  session.add(user)
  session.commit()
  session.close()