flask_script

flask_script

flask_script是一个生成shell命令的模块，为在Flask里面编写额外的脚本提供了支持。包括运行一个开发的服务器，一个自定义的命令行，用于初始化数据库，定时任务和其他属于web引用之外的命令行脚本。

flask_script 对数据库的简单操作

from flask_script import Shell,Manager
from models import app,db,User,Role
from flask_migrate import Migrate,MigrateCommand

#创建一个用来管理命令的对象，Manager用来跟踪所有命令，
manager = Manager(app)
migrate = Migrate(app,db)

# 添加一条数据库管理命令
manager.add_command('db',MigrateCommand)

# 可以显示所有用户
@manager.command
def showUser():
    users = User.query.all()
    print(users[:5])

@manager.command
def showRole():
    roles = Role.query.all()
    print(roles[:5])


@manager.option('-n','--name',help='添加角色')
def addRole(name):
    try:
        role = Role(name=name)
        db.session.add(role)
        db.session.commit()
    except Exception:
        print('创建角色失败')
    else:
        print('创建角色成功')

if __name__=="__main__":
    manager.run()

结果：