117 lines
3.6 KiB
Python
117 lines
3.6 KiB
Python
"""Search indexes serializers."""
|
|
from rest_framework import serializers
|
|
from elasticsearch_dsl import AttrDict
|
|
from django_elasticsearch_dsl_drf.serializers import DocumentSerializer
|
|
from news.serializers import NewsTypeSerializer
|
|
from search_indexes.documents import EstablishmentDocument, NewsDocument
|
|
from search_indexes.utils import get_translated_value
|
|
|
|
|
|
class TagsDocumentSerializer(serializers.Serializer):
|
|
"""Tags serializer for ES Document."""
|
|
|
|
id = serializers.IntegerField()
|
|
label_translated = serializers.SerializerMethodField()
|
|
|
|
def get_label_translated(self, obj):
|
|
if isinstance(obj, dict):
|
|
return get_translated_value(obj.get('label'))
|
|
return get_translated_value(obj.label)
|
|
|
|
|
|
class AddressDocumentSerializer(serializers.Serializer):
|
|
"""Address serializer for ES Document."""
|
|
|
|
id = serializers.IntegerField()
|
|
street_name_1 = serializers.CharField()
|
|
street_name_2 = serializers.CharField()
|
|
number = serializers.IntegerField()
|
|
postal_code = serializers.CharField()
|
|
latitude = serializers.FloatField(allow_null=True, source='coordinates.lat')
|
|
longitude = serializers.FloatField(allow_null=True, source='coordinates.lon')
|
|
geo_lon = serializers.FloatField(allow_null=True, source='coordinates.lon')
|
|
geo_lat = serializers.FloatField(allow_null=True, source='coordinates.lat')
|
|
|
|
# todo: refator
|
|
def to_representation(self, instance):
|
|
if instance != AttrDict(d={}) or \
|
|
(isinstance(instance, dict) and len(instance) != 0):
|
|
return super().to_representation(instance)
|
|
return None
|
|
|
|
|
|
class ScheduleDocumentSerializer(serializers.Serializer):
|
|
"""Schedule serializer for ES Document"""
|
|
|
|
id = serializers.IntegerField()
|
|
weekday = serializers.IntegerField()
|
|
weekday_display = serializers.CharField()
|
|
closed_at = serializers.CharField()
|
|
|
|
|
|
class NewsDocumentSerializer(DocumentSerializer):
|
|
"""News document serializer."""
|
|
|
|
title_translated = serializers.SerializerMethodField(allow_null=True)
|
|
subtitle_translated = serializers.SerializerMethodField(allow_null=True)
|
|
news_type = NewsTypeSerializer()
|
|
tags = TagsDocumentSerializer(many=True)
|
|
|
|
class Meta:
|
|
"""Meta class."""
|
|
|
|
document = NewsDocument
|
|
fields = (
|
|
'id',
|
|
'title_translated',
|
|
'subtitle_translated',
|
|
'is_highlighted',
|
|
'image_url',
|
|
'preview_image_url',
|
|
'news_type',
|
|
'tags',
|
|
'slug',
|
|
)
|
|
|
|
@staticmethod
|
|
def get_title_translated(obj):
|
|
return get_translated_value(obj.title)
|
|
|
|
@staticmethod
|
|
def get_subtitle_translated(obj):
|
|
return get_translated_value(obj.subtitle)
|
|
|
|
|
|
class EstablishmentDocumentSerializer(DocumentSerializer):
|
|
"""Establishment document serializer."""
|
|
|
|
address = AddressDocumentSerializer(allow_null=True)
|
|
tags = TagsDocumentSerializer(many=True)
|
|
schedule = ScheduleDocumentSerializer(many=True, allow_null=True)
|
|
|
|
class Meta:
|
|
"""Meta class."""
|
|
|
|
document = EstablishmentDocument
|
|
fields = (
|
|
'id',
|
|
'name',
|
|
'transliterated_name',
|
|
'index_name',
|
|
'price_level',
|
|
'toque_number',
|
|
'public_mark',
|
|
'slug',
|
|
'preview_image',
|
|
'address',
|
|
'tags',
|
|
'schedule',
|
|
'works_noon',
|
|
'works_evening',
|
|
'works_at_weekday',
|
|
# 'works_now',
|
|
# 'collections',
|
|
# 'establishment_type',
|
|
# 'establishment_subtypes',
|
|
)
|