from pprint import pprint

from django.db.models import Value, IntegerField, F

from news.models import NewsType
from transfer.models import PageTexts
from transfer.serializers.news import NewsSerializer


def transfer_news():
    news_type, _ = NewsType.objects.get_or_create(name="News")

    queryset = PageTexts.objects.filter(page__type="News").annotate(
        news_type=Value(news_type.id, output_field=IntegerField()),
        country_code=F('page__site__country_code_2'),
    )
    queryset = queryset.annotate(template=F('page__template'))

    serialized_data = NewsSerializer(data=list(queryset.values()), many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        pprint(f"News serializer errors: {serialized_data.errors}")


data_types = {
    "news": [transfer_news]
}