gault-millau/apps/transfer/serializers/news.py

from rest_framework import serializers

from gallery.models import Image
from location.models import Country
from news.models import News, NewsGallery
from tag.models import Tag
from transfer.models import PageMetadata
from utils.legacy_parser import parse_legacy_news_content
from utils.slug_generator import generate_unique_slug
from account.models import User


class NewsSerializer(serializers.Serializer):
    # old_id = page__id id -done
    # news_type = 'News' создали или получили в трансфере -done
    # title = {"en-GB":"some text"} из locale и title -done
    # backoffice_title = page__root_title -done
    # subtitle = {"en-GB":"some text"} из locale и summary -done
    # description = {"en-GB":"some text"} из locale и body -done
    # locale_to_description_is_active = {"en-GB": true, "fr-FR": false} из locale и true -done
    # publication_date = DateField из page published_at -done ??? проверить
    # publication_time = DateField из page published_at -done ??? проверить
    # slugs = {"en-GB":"some slug"} из locale и slug -done
    # state = page__state -done
    # template = page__template -done
    # country = по page__site__country_code_2 -done
    # tags = по page__tags__id -progress -!!!
    # gallery = в методе make_gallery из page__attachment_suffix_url -done
    # created_by = page__account_id -done
    # modified_by = page__account_id -done
    # created = page created_at -done

    locale = serializers.CharField()
    page__id = serializers.IntegerField()
    news_type_id = serializers.IntegerField()
    page__created_at = serializers.DateTimeField(format='%m-%d-%Y %H:%M:%S')
    page__account_id = serializers.IntegerField(allow_null=True)
    page__state = serializers.CharField()
    page__template = serializers.CharField()
    page__site__country_code_2 = serializers.CharField(allow_null=True)
    slug = serializers.CharField()
    body = serializers.CharField(allow_null=True)
    title = serializers.CharField()
    page__root_title = serializers.CharField()
    summary = serializers.CharField(allow_null=True, allow_blank=True)
    page__attachment_suffix_url = serializers.CharField()
    page__published_at = serializers.DateTimeField(format='%m-%d-%Y %H:%M:%S', allow_null=True)

    tags = serializers.CharField(allow_null=True)
    tag_cat_tag_id = serializers.IntegerField()
    tag_cat_category_id = serializers.IntegerField()

    def create(self, data):
        account = self.get_account(data)
        payload = {
            'old_id': data['page__id'],
            'news_type_id': data['news_type_id'],
            'created': data['page__created_at'],
            'created_by': account,
            'modified_by': account,
            'state': self.get_state(data),
            'template': self.get_template(data),
            'country': self.get_country(data),
            'slug': {data['locale']: data['slug']},
            'description': self.get_description(data),
            'title': {data['locale']: data['title']},
            'backoffice_title': data['page__root_title'],
            'subtitle': self.get_subtitle(data),
            'locale_to_description_is_active': {data['locale']: True},
            'publication_date': self.get_publication_date(data),
            'publication_time': self.get_publication_time(data),
        }

        obj, _ = News.objects.update_or_create(
            old_id=data['old_id'],
            defaults=payload,
        )

        tags = self.get_tags(data)
        for tag in tags:
            obj.tags.add(tag)
        obj.save()

        self.make_gallery(data, obj)
        return obj

    @staticmethod
    def get_publication_date(data):
        published_at = data.get('page__published_at')
        if published_at:
            return published_at.date()
        return None

    @staticmethod
    def get_publication_time(data):
        published_at = data.get('page__published_at')
        if published_at:
            return published_at.time()
        return None

    @staticmethod
    def get_account(data):
        return User.objects.filter(old_id=data['page__account_id']).first()

    @staticmethod
    def get_state(data):
        states = {
            'new': News.WAITING,
            'published': News.PUBLISHED,
            'hidden': News.HIDDEN,
            'published_exclusive': News.PUBLISHED_EXCLUSIVE,
            'scheduled_exclusively': News.WAITING,
        }
        return states.get(data['page__state'], News.WAITING)

    @staticmethod
    def get_template(data):
        templates = {
            'main': News.MAIN,
            'main.pdf.erb': News.MAIN_PDF_ERB,
            'newspaper': News.NEWSPAPER,
        }
        return templates.get(data['page__template'], News.MAIN)

    @staticmethod
    def get_country(data):
        return Country.objects.filter(code__iexact=data['page__site__country_code_2']).first()

    @staticmethod
    def get_description(data):
        if data['body']:
            content = parse_legacy_news_content(data['body'])
            return {data['locale']: content}
        return None

    @staticmethod
    def get_subtitle(data):
        if data.get('summary'):
            return {data['locale']: data['summary']}
        return None

    @staticmethod
    def get_tags(data):
        results = []
        if not data['tags']:
            return results

        meta_ids = (int(_id) for _id in data['tags'].split(','))
        tags = PageMetadata.objects.filter(
            id__in=meta_ids,
            value__isnull=False,
        )
        for old_tag in tags:
            tag, _ = Tag.objects.get_or_create(
                category_id=data['tag_cat_id'],
                label={data['locale']: old_tag.value},
            )
            results.append(tag)
        return results

    @staticmethod
    def make_gallery(data, obj):
        if not data['page__attachment_suffix_url'] or data['page__attachment_suffix_url'] == 'default/missing.png':
            return

        img, _ = Image.objects.get_or_create(
            image=data['page__attachment_suffix_url'],
            title=data['page__root_title'],
            created=data['page__created_at']
        )

        gal, _ = NewsGallery.objects.get_or_create(
            news=obj,
            image=img,
            is_main=True,
        )