Flask(中间件与全局变量 九)

中间件 middleware

request

当Flask应用处理请求时，它会根据从WSGI服务器收到的环境创建一个Request对象.

因为工作者（取决于服务器的线程，进程或者协程）一次只能处理一个请求，所以在该请求期间请求数据可被认为是该工作者的全部数据。

Flask对此的术语为本地请求

请求钩子 before_request

在获取某个请求数据之前，可以做额外的事情。

场景：运营-数据埋点

各个接口的访问数量与频次

@app.before_request def get_num_of_interface():     # 从数据库中读取缓存数量     print("接口访问数量的缓存+1")     # 将缓存数量+1后存入数据库

场景：验证是否登陆

@app.before_request def get_num_of_interface():     sign = request.args.get('sign')     if sign != 'admin':         abort(401)

场景：操作时间大于一定时间后需要重新登陆

after_request

response参数必须有

return 的必须是一个Response对象

通常用来修改响应

如果视图出现错误，不会调用

场景：封装响应信息

@app.after_request def make_res(response):     response.headers['server'] = 'zx'     print(f'after {request.url}')     return response 响应头修改

teardown_request

通常用来释放资源

不管有没有错误都会调用

场景：关闭数据库连接

@app.teardown_request def teardown_make_res(error):     print(error)     print("tear Down")

before_first_request

第一次请求的时候触发

@app.before_first_request def first():     print("第一次请求")

## 全局变量

g：同一个请求中 共享数据

from flask import g

场景：验证用户信息

可以在同一请求之间共享g的内容，相当于实现了这个请求的一个全局变量

@app.before_request def get_num_of_interface():     sign = request.args.get('sign')     if sign != 'admin':         abort(401)     # 获取sign对于的用户     a = sign + 'md'     g.user = a @app.route('/') def hello_world():     print(g.user)     return 'Hello World!'

场景：数据库连接

def connect_to_database():     conn = pymysql.connect(host='127.0.0.1',                            user='root',                            password='123456',                            db='zx',                            charset='utf8mb4')     return conn.cursor() @app.before_request def get_db():     if 'db' not in g:         g.db = connect_to_database() @app.teardown_request def close_db(exception):     db = g.pop('db', None)     if db is not None:         db.close() @app.route('/hi') def hi():     g.db.execute('SELECT * FROM user_info;')     a = g.db.fetchall()     print(a)

session

用来记录用户会话信息

from flask import Flask, session, request app = Flask(__name__) app.config['SECRET_KEY'] = 'sfoowff' @app.route('/') def home():     if not session.get('user'):         return '你没有登陆'     return '这是首页' @app.route('/login/') def login():     username = request.args.get('username')     pwd = request.args.get('pwd')     if username and pwd:         session['user'] = username         return '登陆成功'     return '账号密码不能为空' @app.route('/logout/') def logout():     session.pop('user', None)     return '登出' if __name__ == '__main__':     app.run()

测试

http://127.0.0.1:5000/login/?username=zx&pwd=123 登陆

访问首页

首页