from django_elasticsearch_dsl import Document
from django_elasticsearch_dsl.registries import registry

from .models import Job


# TODO disabled for testing purpose
# @registry.register_document
class JobDocument(Document):
    class Index:
        # Name of the Elasticsearch index
        name = "jobs"
        # See Elasticsearch Indices API reference for available settings
        settings = {"number_of_shards": 2}
        # settings = {'number_of_shards': 1,
        #             'number_of_replicas': 0}

    class Django:
        model = Job  # The model associated with this Document

        # The fields of the model you want to be indexed in Elasticsearch
        fields = ["title", "location"]

        # Ignore auto updating of Elasticsearch when a model is saved
        # or deleted:
        # ignore_signals = True

        # Don't perform an index refresh after every update (overrides global setting):
        # auto_refresh = False

        # Paginate the django queryset used to populate the index with the specified size
        # (by default it uses the database driver's default setting)
        # queryset_pagination = 5000

    # def get_queryset(self):
    #     """Not mandatory but to improve performance we can select related in one sql request"""
    #     return super(JobDocument, self).get_queryset().select_related('study')