Flask中before_request与after_request使用

如果有一天公司业务需求需要给所有视图函数添加功能，可以通过装饰器实现，但视图函数太多，有没有更好的办法呢？

before_request与after_request很简单，实现中功能同Django中中间件process_request与process_response的功能相同

一、前提，装饰器的弊端

　　我们现在有一个 Flask 程序其中有3个路由和视图函数,如下:

from flask import Flask

app = Flask(__name__)  # type:Flask


@app.route("/login")
def login():
    return "Login"

@app.route("/index")
def index():
    return "Index"

@app.route("/home")
def home():
    return "Login"

app.run("0.0.0.0", 5000)

from flask import Flask

app = Flask(__name__)  # type:Flask


@app.route("/login")
def login():
    return "Login"

@app.route("/index")
def index():
    return "Index"

@app.route("/home")
def home():
    return "Login"

app.run("0.0.0.0", 5000)

　　如果登陆了,就可以访问 index 和 home 页面,如果没登录就跳转到 login 登录

　　要怎么解决呢, session 对, 用 session 除了 Login 函数之外的所有函数里面全校验 session 是否登录了

　　太麻烦了,现在咱们只有3个函数,如果成百上千个怎么整啊

　　装饰器,对没错,装饰器是一个很好的方案,但是啊,我现在还是成败上千个函数,我要在每一个函数定义的时候加上@装饰器,还是很麻烦

二、before_request与after_request

　　2.1 before_request分析：

from flask import Flask, render_template

app = Flask(__name__)


@app.route("/home")
def home():
    return "Hello"


@app.before_request
def be1():
    print("be1")
    return "出错了"
    # return None


@app.before_request
def be2():
    print("be2")
    return None


@app.after_request
def af1(res):
    print("af1")
    return res


@app.after_request
def af2(res):
    print("af2")
    return res


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

　报错信息：

　注意：对比
　# Django中，如果当请求到达请求1的时候直接不符合条件返回，即return HttpResponse("Md1中断")，程序将把请求直接发给中间件1返回，然后依次返回到请求者，不再执行视图函数

　# 在Flask中，如果当请求到达请求1的时候直接不符合条件返回，则会中最后一个@app.after_request中依次返回到请求者，不再执行视图函数，结果如下：
　返回Md2中断的页面，后台打印如下：

be1
af2
af1

　　2.2 after_request分析：

　　@app.after_request报错则会依次返回结果　　

from flask import Flask, render_template

app = Flask(__name__)


@app.route("/home")
def home():
    return "Hello"


@app.before_request
def be1():
    print("be1")
    # return "出错了"
    return None


@app.before_request
def be2():
    print("be2")
    return None


@app.after_request
def af1(res):
    print("af1")
    return res


@app.after_request
def af2(res):
    print("af2")
    # return res
    return "出错了"


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

　　报错信息：

# 返回结果
be1
be2
af2
af1

三、before_request应用

　　@app.before_request 在请求(request)进入视图函数之前执行

　　@app.before_request 也是一个装饰器,他所装饰的函数,都会在请求进入视图函数之前执行

　　request.path 是来读取当前的url地址如果是 /login 就允许直接通过 return None 你可以理解成通过放行

　　校验session中是否有user 如果没有的话,证明没有登录,所以毫不留情的 redirect("/login") 跳转登录页面

　　还有一个要提的 @app.before_first_request 它与 @app.before_request 极为相似或者说是一模一样,只不过它只会被执行一次

from flask import Flask
from flask import request
from flask import redirect
from flask import session

app = Flask(__name__)  # type:Flask
app.secret_key = "DragonFire"


@app.before_request
def is_login():
　　 #白名单
    if request.path == "/login":
        return None
　　 #验证session
    if not session.get("user"):
        return redirect("/login")
　　return None


@app.route("/login")
def login():
    return "Login"


@app.route("/index")
def index():
    return "Index"


@app.route("/home")
def home():
    return "Login"


app.run("0.0.0.0", 5000)

四、before_request应用

　　@app.after_request 在响应(response)返回客户端之前执行 , 结束视图函数之后

　　@app.after_request必须return返回上一层的参数environ，否则会报错

@app.after_request
def foot_log(environ):
    if request.path != "/login":
        print("有客人访问了",request.path)
    return environ

五、用flask的实现一个简单的页面登录

　　基于before_request与after_request的验证登录

from flask import Flask,render_template,request,redirect,session
app = Flask(__name__,template_folder='templates')
app.secret_key = "sdsfdsgdfgdfgfh"

@app.before_request
def process_request():
    if request.path=="/login":
        return None
    if not session.get("user_info"):
        return redirect("/login")
    return None
@app.after_request
def process_response(response):
    print(2222)
    return response

@app.route("/login",methods=["GET","POST"])
def login():
    if request.method=="GET":
        return render_template("login.html")
    else:
        # print(request.values)   #这个里面什么都有，相当于body
        username = request.form.get("username")
        password = request.form.get("password")
        if username=="annie" and password=="123":
            session["user_info"] = username
            # session.pop("user_info")  #删除session
            return redirect("/index")
        else:
            # return render_template("login.html",**{"msg":"用户名或密码错误"})
            return render_template("login.html",msg="用户名或者密码错误")


@app.route("/index",methods=["GET","POST"])
def index():
    # if not session.get("user_info"):
    #     return redirect("/login")
    return render_template("index.html")


if __name__ == '__main__':
    app.run(debug=True)