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
登陆
访问首页
首页
本文参与 腾讯云自媒体分享计划，分享自微信公众号。
原始发表：2020-07-04，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读