flask-script模块的具体使用

Flask-Script是一个与Flask集成的命令行解析器，使得开发者可以轻松地通过命令行来对Flask应用程序进行管理。

安装Flask-Script

在使用Flask-Script之前，需要将其安装到Flask应用程序中。在终端中输入以下命令来安装Flask-Script：

pip install flask-script

使用Flask-Script

在安装Flask-Script之后，需要在Flask应用程序中引入它，并创建一个Manager实例作为应用程序的管理器。

from flask_script import Manager

app = Flask(__name__)
manager = Manager(app)

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

在上面的代码中，我们创建了一个Manager实例，并将Flask应用程序作为参数传递给了Manager构造函数。最后通过if __name__ == '__main__':判断脚本是否作为应用程序的入口运行。

添加命令

接下来，我们就可以定义需要的命令，使用manager.command装饰器来将它们添加到管理器实例中。

@manager.command
def hello():
    print("Hello, Flask-Script")

@manager.command
def migrate():
    """
    数据库迁移
    """
    db.create_all()
    print("数据库迁移成功")

在上述代码中，我们定义了两个命令hello和migrate。hello命令打印一个简单的信息，migrate命令通过调用数据库迁移的相关代码实现。

使用命令

当我们定义好了命令之后，就可以在命令行中运行这些命令了。可以通过以下命令来列出所有可用的命令：

python manage.py --help

运行hello命令的命令如下：

python manage.py hello

运行migrate命令的命令如下：

python manage.py migrate

在运行命令时还可以传递参数。下面的代码展示了如何使用参数调用migrate命令：

@manager.option('-r', '--role', dest='role_name', default=None)
def migrate(role_name):
    """
    数据库迁移
    """
    if role_name:
        db.create_all(bind=[role_name])
        print("迁移{}成功".format(role_name))
    else:
        db.create_all()
        print("迁移所有表成功")

上述代码中，我们新增了一个-r参数，并在命令中使用-r参数指定要迁移的表名。