Flask组件

组件踩坑记录 : 先注册组件在使用配置(...)

flask-script

Flask Script扩展提供向Flask插入外部脚本的功能，包括运行一个开发用的服务器，一个定制的Python shell，设置数据库的脚本，cronjobs，及其他运行在web应用之外的命令行任务；使得脚本和系统分开；

Flask Script和Flask本身的工作方式类似，只需定义和添加从命令行中被Manager实例调用的命令；

安装

pip install flask-script

创建并运行命令

首先，创建一个Python模板运行命令脚本，可起名为manager.py；
在该文件中，必须有一个Manager实例，Manager类追踪所有在命令行中调用的命令和处理过程的调用运行情况；
Manager只有一个参数——Flask实例，也可以是一个函数或其他的返回Flask实例；
调用manager.run()启动Manager实例接收命令行中的命令；

from flask_script import Manager
from debug import app  

manager = Manager(app)  

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

然后就可以在项目目录下执行命令来启动项目了

python manage.py runserver

创建更多命令

from flask_script import Manager
from flask import Flask

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

@manager.command
def custom(arg):
    """
    自定义命令
    python manage.py custom 123
    :param arg:
    :return:
    """
    print(arg)

@manager.option('-n', '--name', dest='name')
@manager.option('-u', '--url', dest='url')
def cmd(name, url):
    """
    自定义命令
    执行： python manage.py  cmd -n wupeiqi -u http://www.oldboyedu.com
    执行： python manage.py  cmd --name wupeiqi --url http://www.oldboyedu.com
    :param name:
    :param url:
    :return:
    """
    print(name, url)

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

flask-sqlalchemy

将flask与sqlalchemy封装到了一起,并且是基于线程隔离的方式

使用

实例化SQLAlchemy对象,将Flask对象注册进SQLAlchemy对象中

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

# 包含了SQLAlchemy相关的所有操作
db = SQLAlchemy()

app = Flask(__name__)
app.config.from_object('settings.DevelopmentConfig')

from .views.account import ac
# 导入蓝图时应注意蓝图中可能导入了db对象,所以要在db对象生成后导入
app.register_blueprint(ac)

db.init_app(app)

配置文件设置

class BaseConfig(object):
    SQLALCHEMY_DATABASE_URI = "mysql+pymysql://root:123456@127.0.0.1:3306/s8day130db?charset=utf8"
    SQLALCHEMY_POOL_SIZE = 5
    SQLALCHEMY_POOL_TIMEOUT = 30
    SQLALCHEMY_POOL_RECYCLE = -1

模型类:

from sqlalchemy import Column, Integer, String
from s8day130_pro import db

class Users(db.Model):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True,autoincrement=True)
    name = Column(String(32),nullable=False,unique=True)

离线脚本生成表/删除表

# 导入db
# 导入app
with app.app_context():
    db.drop_all()
    # db.create_all()

 踩坑记录1 --  无法生成表, 解决方法 : 把所有的模型导入

在视图函数中使用SQLAlchemy操作数据库

from flask import blueprints
ac = blueprints.Blueprint('ac',__name__)

@ac.route('/login',methods=['GET','POST'])
def login():
    data = db.session.query(models.Users).all()
    print(data)
    db.session.remove()
    return 'Login'

flask-migrate

帮助flask做数据库迁移,他是基于flask-script和flask-migrate来做的

安装

pip install flask-migrate

使用

# 导入db,app
from flask_script import Manager
from flask_migrate import Migrate, MigrateCommand

manager = Manager(app)
migrate = Migrate(app, db)

"""
# 数据库迁移命名
    python manage.py db init
    python manage.py db migrate
    python manage.py db upgrade
"""
manager.add_command('db', MigrateCommand)

自定义一个登陆验证的组件

from flask import request,session,redirect

class Auth(object):

    def __init__(self,app=None):
        self.app = app
        if app:
             # 实例化时完成注册
            self.init_app(app)

    def init_app(self,app):
        # 也可手动完成注册过程
        app.auth_manager = self # 将自身添加进app中

        self.app = app
        app.before_request(self.check_login) # 添加一个before_request
        app.context_processor(self.context_processor) # 向模板中添加新的变量

    def check_login(self):
        """
        检查用户是否已经登录
        :return:
        """
        if request.path == '/login':
            return

        user = session.get('user')
        if not user:
            return redirect('/login')

    def context_processor(self):
        user = session.get('user')
        return dict(current_user=user)

    def login(self,data):
        """
        将用户登录信息，放入session
        :param data:
        :return:
        """
        session['user'] = data

    def logout(self):
        """
        将用户登录信息，放入session
        :param data:
        :return:
        """
        del session['user']