from rest_framework import serializers
from rest_framework.fields import empty

from core.classes.badge import Badge



class BadgeField(serializers.DictField):

    source = ''

    label = ''


    def __init__(self, *, read_only=True, write_only=False,
                 required=None, default=empty, initial=empty, source=None,
                 label=None, help_text=None, style=None,
                 error_messages=None, validators=None, allow_null=False):

        super().__init__(read_only=read_only, write_only=write_only,
                 required=required, default=default, initial=initial, source=source,
                 label=label, help_text=help_text, style=style,
                 error_messages=error_messages, validators=validators, allow_null=allow_null)


    def to_representation(self, badge: Badge):

        return badge.to_json


    def to_internal_value(self, data):
        return Badge(data.icon,data.colour, data.url)