16 lines
493 B
Python
16 lines
493 B
Python
import click
|
|
from flask.cli import with_appcontext
|
|
from models import db, User
|
|
|
|
@click.command('create-admin')
|
|
@click.argument('username')
|
|
@click.argument('email')
|
|
@click.argument('password')
|
|
@with_appcontext
|
|
def create_admin(username, email, password):
|
|
"""Create an admin user."""
|
|
user = User(username=username, email=email, is_admin=True)
|
|
user.set_password(password)
|
|
db.session.add(user)
|
|
db.session.commit()
|
|
click.echo(f'Admin user {username} created successfully!') |