本文介绍了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中间件，我想在其中计算请求所用的时间，并将其添加到响应的头部的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！
本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除！