Source code for mt940.json

from __future__ import absolute_import
import json
import decimal
import datetime

from . import models

[docs]class JSONEncoder(json.JSONEncoder):
[docs] def default(self, value): # The following types should simply be cast to strings str_types = (, datetime.datetime, datetime.timedelta, datetime.tzinfo, decimal.Decimal, ) dict_types = ( models.Balance, models.Amount, ) # Handle native types that should be converted to strings if isinstance(value, str_types): return str(value) # Handling of the Transaction objects to include the actual # transactions elif isinstance(value, models.Transactions): data = data['transactions'] = value.transactions return data # If an object has a `data` attribute, return that instead of the # `__dict__` ro prevent loops elif hasattr(value, 'data'): return # Handle types that have a `__dict__` containing the data (doesn't work # for classes using `__slots__` such as `datetime`) elif isinstance(value, dict_types): return value.__dict__ else: # pragma: no cover return json.JSONEncoder.default(self, value)