302 lines
9.1 KiB
Python
302 lines
9.1 KiB
Python
"""News app common serializers."""
|
|
from django.utils.translation import gettext_lazy as _
|
|
from rest_framework import serializers
|
|
from rest_framework.fields import SerializerMethodField
|
|
|
|
from account.serializers.common import UserBaseSerializer
|
|
from gallery.models import Image
|
|
from main.models import SiteSettings
|
|
from location import models as location_models
|
|
from location.serializers import CountrySimpleSerializer, AddressBaseSerializer
|
|
from news import models
|
|
from tag.serializers import TagBaseSerializer
|
|
from utils import exceptions as utils_exceptions
|
|
from utils.serializers import (TranslatedField, ProjectModelSerializer,
|
|
FavoritesCreateSerializer, ImageBaseSerializer, CarouselCreateSerializer)
|
|
|
|
|
|
class AgendaSerializer(ProjectModelSerializer):
|
|
event_datetime = serializers.DateTimeField()
|
|
address = AddressBaseSerializer()
|
|
content_translated = TranslatedField()
|
|
|
|
class Meta:
|
|
"""Meta class."""
|
|
|
|
model = models.Agenda
|
|
fields = (
|
|
'id',
|
|
'event_datetime',
|
|
'address',
|
|
'content_translated'
|
|
)
|
|
|
|
|
|
class NewsBannerSerializer(ProjectModelSerializer):
|
|
title_translated = TranslatedField()
|
|
image_url = serializers.URLField()
|
|
content_url = serializers.URLField()
|
|
|
|
class Meta:
|
|
"""Meta class."""
|
|
|
|
model = models.NewsBanner
|
|
fields = (
|
|
'id',
|
|
'title_translated',
|
|
'image_url',
|
|
'content_url'
|
|
)
|
|
|
|
|
|
class NewsTypeSerializer(serializers.ModelSerializer):
|
|
"""News type serializer."""
|
|
|
|
class Meta:
|
|
"""Meta class."""
|
|
|
|
model = models.NewsType
|
|
fields = ('id', 'name')
|
|
|
|
|
|
class NewsBaseSerializer(ProjectModelSerializer):
|
|
"""Base serializer for News model."""
|
|
|
|
title_translated = TranslatedField()
|
|
subtitle_translated = TranslatedField()
|
|
news_type = NewsTypeSerializer(read_only=True)
|
|
tags = TagBaseSerializer(read_only=True, many=True, source='visible_tags')
|
|
in_favorites = serializers.BooleanField(allow_null=True, read_only=True)
|
|
view_counter = serializers.IntegerField(read_only=True)
|
|
|
|
class Meta:
|
|
"""Meta class."""
|
|
|
|
model = models.News
|
|
fields = (
|
|
'id',
|
|
'title_translated',
|
|
'subtitle_translated',
|
|
'is_highlighted',
|
|
'news_type',
|
|
'tags',
|
|
'slug',
|
|
'view_counter',
|
|
)
|
|
|
|
|
|
class NewsSimilarListSerializer(NewsBaseSerializer):
|
|
"""List serializer for News model."""
|
|
preview_image_url = serializers.URLField()
|
|
|
|
class Meta(NewsBaseSerializer.Meta):
|
|
"""Meta class."""
|
|
fields = NewsBaseSerializer.Meta.fields + (
|
|
'preview_image_url',
|
|
)
|
|
|
|
|
|
class NewsListSerializer(NewsBaseSerializer):
|
|
"""List serializer for News model."""
|
|
|
|
image = ImageBaseSerializer(source='crop_main_image', allow_null=True)
|
|
|
|
class Meta(NewsBaseSerializer.Meta):
|
|
"""Meta class."""
|
|
|
|
fields = NewsBaseSerializer.Meta.fields + (
|
|
'image',
|
|
)
|
|
|
|
|
|
class NewsDetailSerializer(NewsBaseSerializer):
|
|
"""News detail serializer."""
|
|
|
|
description_translated = TranslatedField()
|
|
country = CountrySimpleSerializer(read_only=True)
|
|
author = UserBaseSerializer(source='created_by', read_only=True)
|
|
state_display = serializers.CharField(source='get_state_display',
|
|
read_only=True)
|
|
gallery = ImageBaseSerializer(read_only=True, source='crop_gallery', many=True)
|
|
|
|
class Meta(NewsBaseSerializer.Meta):
|
|
"""Meta class."""
|
|
|
|
fields = NewsBaseSerializer.Meta.fields + (
|
|
'description_translated',
|
|
'start',
|
|
'end',
|
|
'is_publish',
|
|
'state',
|
|
'state_display',
|
|
'author',
|
|
'country',
|
|
'gallery',
|
|
)
|
|
|
|
|
|
class NewsDetailWebSerializer(NewsDetailSerializer):
|
|
"""News detail serializer for web users.."""
|
|
|
|
same_theme = SerializerMethodField()
|
|
should_read = SerializerMethodField()
|
|
agenda = AgendaSerializer()
|
|
banner = NewsBannerSerializer()
|
|
|
|
class Meta(NewsDetailSerializer.Meta):
|
|
"""Meta class."""
|
|
|
|
fields = NewsDetailSerializer.Meta.fields + (
|
|
'same_theme',
|
|
'should_read',
|
|
'agenda',
|
|
'banner',
|
|
)
|
|
|
|
def get_same_theme(self, obj):
|
|
return NewsSimilarListSerializer(obj.same_theme(self.context['request'].user), many=True, read_only=True).data
|
|
|
|
def get_should_read(self, obj):
|
|
return NewsSimilarListSerializer(obj.should_read(self.context['request'].user), many=True, read_only=True).data
|
|
|
|
|
|
class NewsBackOfficeBaseSerializer(NewsBaseSerializer):
|
|
"""News back office base serializer."""
|
|
is_published = serializers.BooleanField(source='is_publish', read_only=True)
|
|
|
|
class Meta(NewsBaseSerializer.Meta):
|
|
"""Meta class."""
|
|
|
|
fields = NewsBaseSerializer.Meta.fields + (
|
|
'title',
|
|
'backoffice_title',
|
|
'subtitle',
|
|
'is_published',
|
|
)
|
|
extra_kwargs = {
|
|
'backoffice_title': {'allow_null': False},
|
|
}
|
|
|
|
|
|
class NewsBackOfficeDetailSerializer(NewsBackOfficeBaseSerializer,
|
|
NewsDetailSerializer):
|
|
"""News detail serializer for back-office users."""
|
|
|
|
news_type_id = serializers.PrimaryKeyRelatedField(
|
|
source='news_type', write_only=True,
|
|
queryset=models.NewsType.objects.all())
|
|
country_id = serializers.PrimaryKeyRelatedField(
|
|
source='country', write_only=True,
|
|
queryset=location_models.Country.objects.all())
|
|
site_id = serializers.PrimaryKeyRelatedField(
|
|
source='site', write_only=True,
|
|
queryset=SiteSettings.objects.all())
|
|
template_display = serializers.CharField(source='get_template_display',
|
|
read_only=True)
|
|
|
|
class Meta(NewsBackOfficeBaseSerializer.Meta, NewsDetailSerializer.Meta):
|
|
"""Meta class."""
|
|
|
|
fields = NewsBackOfficeBaseSerializer.Meta.fields + \
|
|
NewsDetailSerializer.Meta.fields + (
|
|
'description',
|
|
'news_type_id',
|
|
'country_id',
|
|
'site_id',
|
|
'template',
|
|
'template_display',
|
|
'is_international',
|
|
)
|
|
|
|
|
|
class NewsBackOfficeGallerySerializer(serializers.ModelSerializer):
|
|
"""Serializer class for model NewsGallery."""
|
|
|
|
class Meta:
|
|
"""Meta class"""
|
|
|
|
model = models.NewsGallery
|
|
fields = [
|
|
'id',
|
|
'is_main',
|
|
]
|
|
|
|
def get_request_kwargs(self):
|
|
"""Get url kwargs from request."""
|
|
return self.context.get('request').parser_context.get('kwargs')
|
|
|
|
def validate(self, attrs):
|
|
"""Override validate method."""
|
|
news_pk = self.get_request_kwargs().get('pk')
|
|
image_id = self.get_request_kwargs().get('image_id')
|
|
|
|
news_qs = models.News.objects.filter(pk=news_pk)
|
|
image_qs = Image.objects.filter(id=image_id)
|
|
|
|
if not news_qs.exists():
|
|
raise serializers.ValidationError({'detail': _('News not found')})
|
|
if not image_qs.exists():
|
|
raise serializers.ValidationError({'detail': _('Image not found')})
|
|
|
|
news = news_qs.first()
|
|
image = image_qs.first()
|
|
|
|
if image in news.gallery.all():
|
|
raise serializers.ValidationError({'detail': _('Image is already added.')})
|
|
|
|
attrs['news'] = news
|
|
attrs['image'] = image
|
|
|
|
return attrs
|
|
|
|
|
|
class NewsFavoritesCreateSerializer(FavoritesCreateSerializer):
|
|
"""Serializer to favorite object w/ model News."""
|
|
|
|
def validate(self, attrs):
|
|
"""Overridden validate method"""
|
|
# Check establishment object
|
|
news_qs = models.News.objects.filter(slug=self.slug)
|
|
|
|
# Check establishment obj by slug from lookup_kwarg
|
|
if not news_qs.exists():
|
|
raise serializers.ValidationError({'detail': _('Object not found.')})
|
|
else:
|
|
news = news_qs.first()
|
|
|
|
# Check existence in favorites
|
|
if news.favorites.filter(user=self.user).exists():
|
|
raise utils_exceptions.FavoritesError()
|
|
|
|
attrs['news'] = news
|
|
return attrs
|
|
|
|
def create(self, validated_data, *args, **kwargs):
|
|
"""Overridden create method"""
|
|
validated_data.update({
|
|
'user': self.user,
|
|
'content_object': validated_data.pop('news')
|
|
})
|
|
return super().create(validated_data)
|
|
|
|
|
|
class NewsCarouselCreateSerializer(CarouselCreateSerializer):
|
|
"""Serializer to carousel object w/ model News."""
|
|
|
|
def validate(self, attrs):
|
|
news = models.News.objects.filter(pk=self.pk).first()
|
|
if not news:
|
|
raise serializers.ValidationError({'detail': _('Object not found.')})
|
|
|
|
if news.carousels.exists():
|
|
raise utils_exceptions.CarouselError()
|
|
|
|
attrs['news'] = news
|
|
return attrs
|
|
|
|
def create(self, validated_data, *args, **kwargs):
|
|
validated_data.update({
|
|
'content_object': validated_data.pop('news')
|
|
})
|
|
return super().create(validated_data)
|