Class CompatEncoder

src/personalization_api_function/util.py:7–19

Compatibility encoder that supports Decimal type Usage: json.dumps(data, cls=CompatEncoder)

import decimal

class CompatEncoder(json.JSONEncoder):
    """ Compatibility encoder that supports Decimal type
    Usage:
    json.dumps(data, cls=CompatEncoder)
    """
    def default(self, obj):
        if isinstance(obj, decimal.Decimal):
            if obj % 1 > 0:
                return float(obj)
            else:
                return int(obj)
        else:
            return super(CompatEncoder, self).default(obj)