Move sqlalchemy_dumps to atst.utils.json

This commit is contained in:
richard-dds
2019-10-07 15:22:21 -04:00
parent 343e989de8
commit 78c4949776
2 changed files with 14 additions and 12 deletions

View File

@@ -1,6 +1,9 @@
from flask.json import JSONEncoder
import json
from werkzeug.datastructures import FileStorage
from datetime import date
from enum import Enum
from atst.models.attachment import Attachment
@@ -13,3 +16,13 @@ class CustomJSONEncoder(JSONEncoder):
elif isinstance(obj, FileStorage):
return obj.filename
return JSONEncoder.default(self, obj)
def sqlalchemy_dumps(dct):
def _default(obj):
if isinstance(obj, Enum):
return obj.name
else:
raise TypeError()
return json.dumps(dct, default=_default)