Python Flask：我有一个 Flask 中间件，我想在其中计算请求所花费的时间并将其添加到响应的 header

Question

我有以下中间件：

from werkzeug.wrappers import Request
from datetime import datetime 

class TimingMiddleware(object):
    def __init__(self, app):
        self.app = app

    def __call__(self, environ, start_response):
        request = Request(environ)

        startTime = datetime.now()

        response = self.app(environ, start_response)

        endTime = datetime.now()
        timeTaken = (endTime-startTime).total_seconds()

        return response

我想将此 timeTaken 值添加到响应中的 header。我尝试导入并使用 make_response 和来自烧瓶的响应，但没有成功。

Answer 1

这很有趣。这是我快速破解的内容。似乎有效：

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

    def __call__(self, env, start_response):
        start_time = datetime.now()

        def injecting_start_response(status, headers, exc_info=None):
            end_time = datetime.now()
            time_taken = (end_time - start_time).total_seconds()
            headers.append(('X-Time-Taken', str(time_taken)))
            return start_response(status, headers, exc_info)

        return self.app(env, injecting_start_response)

Python Flask：我有一个 Flask 中间件，我想在其中计算请求所花费的时间并将其添加到响应的 header

Python Flask: I have a flask middleware in which I want to calculate the time taken by the requests and add it to the header of the response

python

middleware

flask