16 lines
496 B
Python
16 lines
496 B
Python
from flask.json import JSONEncoder
|
|
from werkzeug.datastructures import FileStorage
|
|
from datetime import date
|
|
from atst.models.attachment import Attachment
|
|
|
|
|
|
class CustomJSONEncoder(JSONEncoder):
|
|
def default(self, obj):
|
|
if isinstance(obj, Attachment):
|
|
return obj.filename
|
|
elif isinstance(obj, date):
|
|
return obj.strftime("%Y-%m-%d")
|
|
elif isinstance(obj, FileStorage):
|
|
return obj.filename
|
|
return JSONEncoder.default(self, obj)
|