From f4ee7ddd9e4a80e578963783f06f30170be7a2a8 Mon Sep 17 00:00:00 2001 From: Ethan Dalool Date: Mon, 18 Sep 2017 14:10:25 -0700 Subject: [PATCH] Include werkzeug Response type in typecheck. --- frontends/etiquette_flask/etiquette_flask/sessions.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/frontends/etiquette_flask/etiquette_flask/sessions.py b/frontends/etiquette_flask/etiquette_flask/sessions.py index c97e7f6..f924ff3 100644 --- a/frontends/etiquette_flask/etiquette_flask/sessions.py +++ b/frontends/etiquette_flask/etiquette_flask/sessions.py @@ -2,6 +2,7 @@ import flask from flask import request import functools import uuid +import werkzeug.wrappers from etiquette import helpers @@ -43,8 +44,7 @@ class SessionManager: request.cookies['etiquette_session'] = token response = function(*args, **kwargs) - - if not isinstance(response, flask.Response): + if not isinstance(response, (flask.Response, werkzeug.wrappers.Response)): response = flask.Response(response) # Send the token back to the client