35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
"""Establishment app documents."""
|
|
from django.conf import settings
|
|
from django_elasticsearch_dsl import Document, Index, fields
|
|
from search_indexes.utils import OBJECT_FIELD_PROPERTIES
|
|
from establishment import models
|
|
|
|
|
|
EstablishmentIndex = Index(settings.ELASTICSEARCH_INDEX_NAMES.get(__name__,
|
|
'establishment'))
|
|
EstablishmentIndex.settings(number_of_shards=1, number_of_replicas=1)
|
|
|
|
|
|
@EstablishmentIndex.doc_type
|
|
class EstablishmentDocument(Document):
|
|
"""Establishment document."""
|
|
|
|
name = fields.ObjectField(properties=OBJECT_FIELD_PROPERTIES)
|
|
description = fields.ObjectField(properties=OBJECT_FIELD_PROPERTIES)
|
|
|
|
class Django:
|
|
|
|
model = models.Establishment
|
|
fields = (
|
|
'id',
|
|
'public_mark',
|
|
'toque_number',
|
|
'price_level',
|
|
)
|
|
|
|
def prepare_name(self, instance):
|
|
return instance.name
|
|
|
|
def prepare_description(self, instance):
|
|
return instance.description
|