python学习笔记(flask-script)

1.什么是Flask-Script?

Flask-Script用来生成shell命令；为在Flask里编写额外的脚本提供了支持。

• 这包括运行一个开发服务器，一个定制的Python命令行，用于执行
  初始化数据库、定时任务和其他属于web应用之外的命令行任务的脚本。
• Flask-Script和Flask本身的工作方式类似。只需要定义和添加能从命令行中
  被Manager实例调用的命令即可。

2. 为什么使用Flask-Script?

Flask的开发Web服务器支持很多启动设置选项，但只能在脚本中作为参数传给app.run()函数。
这种方式很不方便，传递设置选项的理想方式是使用命令行参数。
Flask-Scrip就是这么一个Flask扩展，为Flask程序添加一个命令行解析器。
Flask-Script自带了一组常用选项，而且还支持自定义命令。

3. 如何安装Flask-Script?

pip install flask_script

4. 如何配置Flask-Script?

创建一个Python模块运行你的命令脚本。可以任意起名，例如manage.py。
无需把所有的命令都放在同一个文件里，例如，在一个大型项目中，可以把相关联的命令放在不同的文件里。

from flask_script import Manager

app = Flask(__name__)
# Manager类将追踪所有的在命令行中调用的命令和处理过程的调用运行情况;
# configure your app
manager = Manager(app)

if __name__ == "__main__":
    # 将启动Manger实例接收命令行中的命令。
    manager.run()

• 实现功能
  

5. 添加自定义命令的3中方式:

• 网站参考: https://flask-script.readthedocs.io/en/latest/
• 定义Command的子类;

from flask_script import Manager, prompt_bool, Command

from app import db
from app.models import User, Role

database_manager  = Manager(usage="数据库的操作详情")



# 1. 方法一： 类的继承
class AddUser(Command):
    """添加用户"""
    def run(self):
        u = User(username="fentiao90009", password="fentiao", email="[email protected]")
        db.session.add(u)
        db.session.commit()
        return "正在添加用户%s........" % (u.username)
# 添加类到命令中， 让manger进行管理
database_manager.add_command('adduser', AddUser)

• 使用command装饰器

# 2. 通过装饰器@database_manager.command
@database_manager.command
def deleteUser():
    """删除用户"""
    u = User.query.filter_by(username='fentiao90009').first()
    if u:
        db.session.delete(u)
        db.session.commit()
        return "删除用户%s成功....." % (u.username)
    else:
        return "删除用户失败:"

• 使用option装饰器

# 3). option装饰器， 可以指定参数
@database_manager.option('-u', '--username', help="指定用户名")
@database_manager.option('-p', '--password', help="指定密码")
def add_user(username, password):
    """添加用户， 指定用户名和密码"""
    if username and password:
        u = User(username=username, password=password)
        db.session.add(u)
        db.session.commit()
        return  "添加用户%s成功" %(u.username)
    else:
        return  "请指定用户名和密码"

6. 命令行拓展开发

from flask_script import Manager, prompt_bool, Command

from app import db
from app.models import User, Role

database_manager  = Manager(usage="数据库的操作详情")


@database_manager.command
def dropdb():
    """删除数据库"""
    if prompt_bool("是否删除数据库"):
        db.drop_all()


@database_manager.command
def createdb():
    """删除数据库"""
    if prompt_bool("是否创建数据库"):
        db.create_all()


@database_manager.command
def recreate():
    """重建数据库"""
    if prompt_bool("是否重数据库"):
        dropdb()
        createdb()


@database_manager.command
def init_data():
    role = Role(name="管理员")
    db.session.add(role)


    for user in range(100):
        u = User(username="westos%s" %(user), password="hello", email="westos%[email protected]" %(user))
        db.session.add(u)


    db.session.commit()
    print("初始化完成")