添加 HTTP 方法重写支持

某些 HTTP 代理不支持任意的 HTTP 方法,或者不支持较新的方法(如 PATCH)。 在这种情况下,可以通过另一种 HTTP 方法来“代理”实际想要使用的方法,尽管这在协议层面是不规范的做法。

其工作方式是:客户端发送一个 HTTP POST 请求,并设置 X-HTTP-Method-Override 请求头。 在请求传递给 Flask 之前,服务器会用该请求头中的值替换原始的方法。

可以通过编写一个 HTTP 中间件来实现这一功能:

class HTTPMethodOverrideMiddleware(object):
    allowed_methods = frozenset([
        'GET',
        'HEAD',
        'POST',
        'DELETE',
        'PUT',
        'PATCH',
        'OPTIONS'
    ])
    bodyless_methods = frozenset(['GET', 'HEAD', 'OPTIONS', 'DELETE'])

    def __init__(self, app):
        self.app = app

    def __call__(self, environ, start_response):
        method = environ.get('HTTP_X_HTTP_METHOD_OVERRIDE', '').upper()
        if method in self.allowed_methods:
            environ['REQUEST_METHOD'] = method
        if method in self.bodyless_methods:
            environ['CONTENT_LENGTH'] = '0'
        return self.app(environ, start_response)

要在 Flask 中使用该功能,只需使用该中间件包装 app 对象:

from flask import Flask

app = Flask(__name__)
app.wsgi_app = HTTPMethodOverrideMiddleware(app.wsgi_app)