flask框架之中间件

前言

这几年一直在it行业里摸爬滚打,一路走来,不少总结了一些python行业里的高频面试,看到大部分初入行的新鲜血液,还在为各样的面试题答案或收录有各种困难问题

于是乎,我自己开发了一款面试宝典,希望能帮到大家,也希望有更多的Python新人真正加入从事到这个行业里,让python火不只是停留在广告上。

微信小程序搜索:Python面试宝典

或可关注原创个人博客:https://lienze.tech

也可关注微信公众号,不定时发送各类有趣猎奇的技术文章:Python编程学习

中间件

在flask中,在执行app.run()之后,如果有请求来,会执行app.__call__方法

通过中间件的方法可以在请求来时之前与来临之后,即执行app.__call__前后执行其他代码,进行一些生命周期的控制

  • 根据werkzeug我们可以知道 每次请求必然经历了Flask.app
  • 所以我们要查看Flask的源码发现run方法实际的调用是
    • app.run()->run_simple()->self()__call__->self.wsgi_app(environ, start_response)
  • 找到了Flask的__call__方法后发现执行了return self.wsgi_app(environ, start_response)

flask里面所有的内容调度都是基于这个self.wsgi_app(environ, start_response),这就是就是flask的入口,也就是self是app,也就是app.wsgi_app(environ, start_response)为程序的入口。

如何实现中间件呢? 原理上就是重写app.wsgi_app,然后在里面添加上一些自己想要实现的功能。

首先分析 app.wsgi_app需要加括号执行 所以我们把app.wsgi_app做成一个对象,并且这个对象需要加括号运行,也就是会触发这个对象的类的__call__()方法

class MiddleWare:
    def __init__(self, wsgi_app):
        self.wsgi_app = wsgi_app

    def __call__(self, *args, **kwargs):
        print("....")
        return self.wsgi_app(*args, **kwargs)
app.wsgi_app = MiddleWare(app.wsgi_app)

猜你喜欢

转载自blog.csdn.net/HeroicLee/article/details/120991408
今日推荐