Python，Flask：如何为所有响应设置响应标头

Question

我想将我所有的 http 标头响应设置为这样的：

response.headers["X-Frame-Options"] = "SAMEORIGIN"

我检查了this question，但它只更改了一个特定控制器的标题。我想在类似于以下逻辑的“before_request”函数中更改我的所有标题。我该怎么做？

@app.before_request
def before_request():
    # response.headers["X-Frame-Options"] = "SAMEORIGIN"

Answer 1

我们可以使用 WSGI 中间件优雅地为 Python Flask 应用程序中的所有响应设置响应头

这种使用中间件在 Flask 应用程序上下文中设置响应标头的方式是线程安全的，可用于设置自定义和动态属性，读取请求标头如果我们从任何帮助程序类设置自定义/动态响应标头，这尤其有用.

文件：middleware.py

import flask
from flask import request, g

class SimpleMiddleWare(object):
    """
    Simple WSGI middleware
    """

    def __init__(self, app):
        self.app = app
        self._header_name = "any_request_header"

    def __call__(self, environ, start_response):
        """
        middleware to capture request header from incoming http request
        """
        request_id_header = environ.get(self._header_name)  # reading all request headers
        environ[self._header_name] = request_id_header  

        def new_start_response(status, response_headers, exc_info=None):
            """
            set custom response headers
            """
            # set the above captured request header as response header
            response_headers.append((self._header_name, request_id_header))
            # example to access flask.g values set in any class thats part of the Flask app & then set that as response header
            values = g.get(my_response_header, {})
            if values.get('x-custom-header'):
                response_headers.append(('x-custom-header', values.get('x-custom-header')))
            return start_response(status, response_headers, exc_info)

        return self.app(environ, new_start_response)

从主类调用中间件

文件：main.py

from flask import Flask
import asyncio
from gevent.pywsgi import WSGIServer
from middleware import SimpleMiddleWare

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

Answer 2

@app.after_request() hook 不适合我的用例。

我的用例如下：我有一个谷歌云功能，我想为所有响应设置 CORS 标头。可能有多个响应，因为我必须验证输入并在有问题时返回，我必须处理数据并可能在出现故障时提前返回等。所以我创建了一个辅助函数，如下所示：

# Helper function to return a response with status code and CORS headers
def prepare_response(res_object, status_code):
    response = flask.jsonify(res_object)
    response.headers.set('Access-Control-Allow-Origin', '*')
    response.headers.set('Access-Control-Allow-Methods', 'GET, POST')
    return response, status_code

因此，当我想返回响应（始终带有 CORS 标头）时，我现在可以调用此函数，并且我不会复制启用 CORS 所需的 response.headers 设置。

Answer 3

在@app.after_request() hook 中设置标头，此时您有一个响应对象来设置标头：

@app.after_request
def apply_caching(response):
    response.headers["X-Frame-Options"] = "SAMEORIGIN"
    return response

flask.request context 在此钩子运行时仍然可用，因此此时您仍然可以根据请求改变响应。