from django.contrib.postgres.fields import ArrayField
from django.core.serializers.json import Deserializer
from django.core.serializers.json import Serializer as Serializer_
from django.utils.encoding import is_protected_type

# NOTE: Module must contain both Serializer and Deserializer

__all__ = (
    'Deserializer',
    'Serializer',
)


class Serializer(Serializer_):
    """
    Custom extension of Django's JSON serializer to support ArrayFields (see
    https://code.djangoproject.com/ticket/33974).
    """
    def _value_from_field(self, obj, field):
        value = field.value_from_object(obj)

        # Handle ArrayFields (including subclasses) of protected types
        if isinstance(field, ArrayField):
            if not value or is_protected_type(value[0]):
                return value

        return value if is_protected_type(value) else field.value_to_string(obj)