Corvus/server/corvus/utility.py

from datetime import date
from typing import Any

import rfc3339
from flask.json import JSONEncoder


class CustomJSONEncoder(JSONEncoder):
    def default(self, obj: Any) -> Any:
        try:
            if isinstance(obj, date):
                return rfc3339.format(obj)
            iterable = iter(obj)
        except TypeError:
            pass
        else:
            return list(iterable)
        return JSONEncoder.default(self, obj)