From 193d128d7f004cfc8e44e040571ed48b56242a88 Mon Sep 17 00:00:00 2001 From: Montana Date: Thu, 7 Feb 2019 11:31:52 -0500 Subject: [PATCH] Update JSONEncoder to accept FileStorage objects --- atst/utils/json.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/atst/utils/json.py b/atst/utils/json.py index 4ce7bd8d..8e2a3217 100644 --- a/atst/utils/json.py +++ b/atst/utils/json.py @@ -1,4 +1,5 @@ from flask.json import JSONEncoder +from werkzeug.datastructures import FileStorage from datetime import date from atst.models.attachment import Attachment @@ -7,6 +8,8 @@ class CustomJSONEncoder(JSONEncoder): def default(self, obj): if isinstance(obj, Attachment): return obj.filename - if isinstance(obj, date): + elif isinstance(obj, date): return obj.strftime("%Y-%m-%d") + elif isinstance(obj, FileStorage): + return obj.filename return JSONEncoder.default(self, obj)