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

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 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)

这篇关于Python FlaskTM：我有一个FASK中间件，我想在其中计算请求所用的时间，并将其添加到响应的头部的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！