中间件
一、剖析:
在前面讲session部分提到过:请求一进来,Flask会自动调用应用程序对象【Flask(__name__)】的__call__
方法,这个方法负责处理请求并返回响应(其实如下图:其内部就是wsgi_app方法
)。它是WSGI规范所要求的。
在wsgi_app
方法内部,Flask会根据路由规则和视图函数来确定如何处理请求,并生成相应的响应。最终,wsgi_app
方法会将响应返回给Web服务器,供其发送给客户端(前面详细讲过,后面还会再细细剖析)。
所以,这里就有一个坑可以让我们操作,即我们可以通过覆写wsgi_app
方法,实现自己的中间件逻辑,例如身份验证、日志记录等。
而且这个坑牛逼之处在于:我们可以借由它实现在最开始的开始和最后的最后做操作!!!多加理解,底下会详细讲解。
结合实战讲解—通过覆写wsgi_app
函数实现一个简单的中间件:
from flask import Flask
app = Flask(__name__)
class MyMiddleware:
def __init__(self, old_wsgi_app):
# 服务端启动时,自动执行
self.old_wsgi_app = old_wsgi_app
def __call__(self, environ, start_response):
# 每次有用户请求到来时执行
# 在请求到达视图函数之前执行的代码
print("Before request")
# 调用原始的wsgi_app函数处理请求
response = self.old_wsgi_app(environ, start_response)
# 在响应发送给客户端之前执行的代码
print("After request")
return response
@app.route('/')
def home():
return "Hello, GuHanZhe~"
# 设置自定义中间件实例为应用程序对象的wsgi_app属性
app.wsgi_app = MyMiddleware(app.wsgi_app)
if __name__ == '__main__':
app.run()
在上述示例中,我定义了一个名为MyMiddleware
的类,它接受一个Flask应用程序对象的wsgi_app方法
作为参数。该类实现了__call__
方法,这是一个wsgi应用程序必须具备的方法。在__call__
方法中,大家可以编写自己的中间件逻辑。
在__call__
方法中,大家可以首先执行在请求到达视图函数之前需要执行的代码,然后调用原始的wsgi_app
方法处理请求,并将响应保存在response
变量中。最后,在响应发送给客户端之前,大家可以执行一些在响应阶段需要执行的代码。
通过创建自定义中间件实例,并将其设置为Flask应用程序对象的wsgi_app
属性,就可以使用自定义的中间件了。
请注意,自定义中间件类必须实现__call__
方法,并且接受environ
和start_response
参数,这是遵守WSGI规范。
而且看Flask的wsgi_app函数
源码,也是如此:
需要注意的是:在print("Before request")
部分做操作时,只有原生的请求相关的数据environ,所以就只能对原生的environ做操作! request和session是都没有的!!!
二、应用点:
通过覆写wsgi_app
函数实现的自定义中间件可以有如下几种在开发时的应用点:
-
可以对请求进行预处理:通过覆写
wsgi_app
函数,可以在请求到达视图函数之前执行一些代码逻辑,例如身份验证、参数解析、请求日志记录等。这样可以方便地对请求进行预处理,并根据需要做出相应的处理。 -
可以对响应进行后处理:同样地,在覆写
wsgi_app
函数时,还可以在响应发送给客户端之前执行一些代码逻辑,例如响应的加工、错误处理、响应日志记录等。这样可以方便地对响应进行后处理,以满足特定的需求。 -
可以实现自定义中间件功能:通过覆写
wsgi_app
函数,可以实现自定义的中间件功能。中间件是一种可重用的组件,可以用于添加额外的逻辑或修改请求/响应的行为。可以根据具体需求编写自己的中间件,并将其插入Flask应用程序对象的处理流程中。 -
可以实现多个中间件的串联:Flask允许使用多个中间件,并且这些中间件可以按照特定的顺序串联起来。通过覆写
wsgi_app
函数,可以轻松地将多个中间件组合起来,形成一个中间件链条。每个中间件都可以独立地处理请求和响应,并将处理结果传递给下一个中间件。 -
可以修改请求和响应:通过覆写
wsgi_app
函数,可以自由地修改请求和响应对象。这包括添加、删除或修改请求头部信息,修改请求体内容,修改响应状态码,添加响应头部信息等。这样可以实现更加灵活和定制化的请求/响应处理。
总而言之,覆写wsgi_app
函数实现中间件提供了对请求和响应进行预处理和后处理的能力,同时也允许编写自定义的中间件功能。这样可以增强Flask应用程序的功能和灵活性,满足特定的需求。
就比如要做IP黑名单,就可以在before里直接写逻辑(environ里有请求IP信息),这样就可以在最开始的开始直接限制!
拓展:
(1)在Python中,当一个对象后面能加括号,那么这个对象可能是什么?
-
函数:一个函数是可调用的对象。通过在函数名后加上括号,可以执行该函数并传递相应的参数。
-
方法:方法是属于类的函数。通过在实例或类名后加上括号,可以调用该方法并传递相应的参数。
-
类:类本身也是可调用的对象。通过在类名后加上括号,可以创建类的实例。
-
对象:某个类的实例对象也可以是可调用的对象。通过在对象名后加上括号,可以调用该对象所属类中定义的特殊方法,例如
__call__()
方法。
(2)在Python中,函数和方法分别是什么?
首先,要认识到在Python中,函数(function)和方法(method)是两种不同的概念。这也是为啥我给的问题是“函数和方法分别是什么?”
-
函数(function)是一段封装了特定功能的可重用代码块。它接收输入参数,执行特定的操作,并返回结果。 函数可以在任何地方定义和使用,不依赖于任何类或对象。它们通常用于模块化代码、提高代码的复用性和可维护性。
例如,下面是一个简单的函数示例:
def add(a, b):
return a + b
result = add(2, 3)
print(result)
-
**方法(method)是属于某个类的函数。**它定义在类的内部,并且可以访问类的属性和其他方法。**方法通过对类的实例进行调用来执行相应的操作。**每个方法的第一个参数通常都是
self
,它表示方法所属的实例对象。例如,下面是一个简单的类和方法示例:
class Circle:
def __init__(self, radius):
self.radius = radius
def area(self):
return 3.14 * self.radius * self.radius
circle = Circle(5)
circle_area = circle.area()
print(circle_area) # 输出:78.5
在上面的示例中,area()
是 Circle
类的一个方法,它可以通过 circle.area()
的方式进行调用。
需要注意的是,对于一个类里面的函数,它究竟真是函数还是方法,取决于谁调用它!!!
举个例子:
上图中Test.index执行的话,此时index作为函数,意思是执行类里面的函数;
而下面是通过类的实例调用,所以此时是方法。
或者这样也能证明:
from types import MethodType, FunctionType
class Test(object):
def index(self):
pass
# print(Test.index)
print(isinstance(Test.index, FunctionType))
test = Test()
# print(test.index)
print(isinstance(test.index, MethodType))
总结:函数是独立的可调用代码块,而方法是属于类的函数,需要通过类的实例进行调用。