def admin(username, password):
db.create_all()
user = User.query.first()
if user is not None:
click.echo('更新用户信息')
user.username = username
user.set_password(password)
else:
click.echo('创建用户信息')
user = User(username=username, name='admin')
user.set_password(password)
db.session.add(user)
db.session.commit()
click.echo('管理员创建完成')
def initdb(drop):
if drop:
db.drop_all()
db.create_all()
click.echo("初始化数据库完成")