in tsqa/endpoint.py [0:0]
def __init__(self, app, port=0):
threading.Thread.__init__(self)
# dict to store request data in
self._tracked_requests = {}
self.daemon = True
self.port = port
self.ready = threading.Event()
self.app = app
self.app.debug = True
@self.app.before_request
def save_request():
'''
If the tracking header is set, save the request
'''
if flask.request.headers.get(self.TRACKING_HEADER):
self._tracked_requests[flask.request.headers[self.TRACKING_HEADER]] = {'request': request.copy()}
@self.app.after_request
def save_response(response):
'''
If the tracking header is set, save the response
'''
if flask.request.headers.get(self.TRACKING_HEADER):
self._tracked_requests[flask.request.headers[self.TRACKING_HEADER]]['response'] = response
return response