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.')