python案例中Flask全局配置示例详解

时间:2022-12-07 老边人气:0

WEB服务全局配置

在目前的开发过市场当中，有很多WEB服务框架，Flask只是其中之一，但是总体上来看，所有的WEB框架都是依据HTTP协议的逻辑从请求到响应设计的。固然有很多功能是独立的，但是也有一部分功能需要全局设定，比如安全校验，比如埋点日志，那么这里就用到了全局配置。

所谓的全局配置，就是在框架全局，请求前后，响应前后，设置的全局配置，比如登录校验，这个功能并不是针对单个页面，而是针对所有页面的操作，当然可以一个页面一个页面的实现，但是更好的办法就是在请求进入视图之前，就开始校验，这样也可以降低服务器的部分压力，那么这样全局设置就是一个很好的方法。

Flask全局配置

flask 同样也设置了全局的配置，常用的有以下两种：

before_request

在请求之前执行，一般用在对请求内容的过滤的场景下。一般使用的时候需要结合flask本身的request方法。

默认采用装饰器的方式使用

from flask import Flask
from flask import request
app = Flask(__name__)
@app.before_request
def ip_filter():
    black_list = [
        "127.0.0.1"
    ]
    remote_addr = request.remote_addr #获取请求用户ip
    if remote_addr in black_list:
        return "Not Allowed",405 
@app.route("/index")
def index():
    return "index"
if __name__ == '__main__':
    app.run()

这个时候，如果采用本地ip访问，就会返回405状态，拒绝访问。

当然也可以不采用装饰器使用，其实原理上是一样的：

类似代码就全部注释了，嘿嘿嘿

from flask import Flask
from flask import request
app = Flask(__name__)
#....
app.before_request(ip_filter)
#.... 
if __name__ == '__main__':
    app.run()

after_request

在请求处理完成执行，这里可以对全局的响应进行配置，比如，全局设置json响应格式，但是需要注意的是after_request的函数必须设置参数接收response对象，这个response对象就是视图返回的的对象。

from flask import Flask
app = Flask(__name__)
@app.route("/index")
def index():
    return {"data": {"nav":["index","news_list"]},"message": "success", "error_code": 1}
@app.after_request
def after_request_fun(response):
    response_format = {"data": {},"message": "success", "error_code": 1}
    if response.content_type == "application/json": #如果返回数据的类型是json
        json_data = response.json
        response_format.update(json_data)
    return response
# app.after_request(after_request_fun) 第二种写法
if __name__ == '__main__':
    app.run()

其他一些全局方法也列在这里吧：

方法	描述
before_first_request	服务器启动第一次请求之前执行，一般初始化一些数据使用
teardown_request	比较难理解，直接和请求上下文环境挂钩. 只有在请求上下文被 pop 出请求栈的时候才会触发，需要先去看看flask请求上下文
errorhandler	发生错误时执行,需要提前定义号错误句柄，被触发的前提是 view_function 中抛出了错误
template_global	前后端不分离，jinja2全局变量，比如用户状态可以设置在这里
template_filter	之前聊过的全局jinja2过滤器

Flask自定义中间件

flask 支持通过自定义类来替换Flask本身的wsgi_app，当然，简单的修改建议保留原来的wsgi_app对象

from flask import Flask
app = Flask(__name__)
@app.route("/index")
def index():
    return {"data": {"nav":["index","news_list"]},"message": "success", "error_code": 1}
class MiddleWare:
    def __init__(self, old_wsgi_app):
        self.old_wsgi_app = old_wsgi_app
    def __call__(self, environ, start_response):
        print('类似before_request')
        ret = self.old_wsgi_app(environ, start_response)
        print('类似after_request')
        return ret
if __name__ == '__main__':
    app.wsgi_app = MiddleWare(app.wsgi_app)
    app.run()

加载全部内容