29 lines
795 B
Python
29 lines
795 B
Python
import click
|
|
from flask.cli import with_appcontext
|
|
|
|
from .models import User, db
|
|
|
|
|
|
@click.command()
|
|
@click.argument('identity')
|
|
@with_appcontext
|
|
def delete_user(identity):
|
|
print(identity)
|
|
user = None
|
|
if str.isdigit(identity):
|
|
user = User.query.filter(id=identity).first()
|
|
else:
|
|
user = User.query.filter(
|
|
(User.account_address == identity) |
|
|
(User.email_address == identity)
|
|
).first()
|
|
|
|
if user:
|
|
db.session.delete(user)
|
|
db.session.commit()
|
|
click.echo(f'Succesfully deleted {user.display_name} (uid {user.id})')
|
|
else:
|
|
raise click.BadParameter('Invalid user identity. Must be a userid, '\
|
|
'account address, or email address of an '\
|
|
'existing user.')
|