from rest_framework import serializers

from advertisement.models import Advertisement


class AdvertisementSerializer(serializers.Serializer):
    id = serializers.IntegerField()
    href = serializers.CharField()
    attachment_suffix_url = serializers.CharField(allow_null=True)

    def validate(self, data):
        data.update({
            'old_id': data.pop('id'),
            'url': data.pop('href'),
            'image_url': data.pop('attachment_suffix_url'),
            'width':  300,
            'height': 250,
        })
        return data

    def create(self, validated_data):
        return Advertisement.objects.create(**validated_data)