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

我有以下中间件：

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响应中标头的值。 我尝试从烧瓶导入和使用 make_response 和 Response 但没有运气。

这是一个有趣的事情。这是我快速整理出来的。似乎有效：

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)