Python FlaskTM：我有一个FASK中间件，我想在其中计算请求所用的时间，并将其添加到响应的头部

问题描述

我有以下中间件：

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)