pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update JSONEncoder to accept FileStorage objects

Browse Source
This commit is contained in:
Montana 2019-02-07 11:31:52 -05:00
parent a6eab76ab7
commit 193d128d7f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
atst/utils/json.py
View File

@ -1,4 +1,5 @@
from flask.json import JSONEncoder from flask.json import JSONEncoder
from werkzeug.datastructures import FileStorage
from datetime import date from datetime import date
from atst.models.attachment import Attachment from atst.models.attachment import Attachment
@ -7,6 +8,8 @@ class CustomJSONEncoder(JSONEncoder):
def default(self, obj): def default(self, obj):
if isinstance(obj, Attachment): if isinstance(obj, Attachment):
return obj.filename return obj.filename
if isinstance(obj, date): elif isinstance(obj, date):
return obj.strftime("%Y-%m-%d") return obj.strftime("%Y-%m-%d")
elif isinstance(obj, FileStorage):
return obj.filename
return JSONEncoder.default(self, obj) return JSONEncoder.default(self, obj)