o hI@s&ddlZddlZddlZddlmZddlmZddlmZm Z ddl m Z ddl m Z m Z mZmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!ddl"m#Z#dd l$m%Z%m&Z&m'Z'Gd d d e(Z)Gd d d ej*e)dZ+Gddde+Z,efddZ-Gddde+Z.dS)N)ImproperlyConfigured) force_str)connection_router connections) Indexable) DEFAULT_ALIAS DJANGO_CT DJANGO_IDID) BooleanField CharField DateField DateTimeField DecimalFieldEdgeNgramFieldFacetCharFieldFacetDateFieldFacetDateTimeFieldFacetIntegerField FloatField IntegerField LocationFieldMultiValueField NgramField SearchFieldSearchFieldError)SearchIndexManager)get_facet_field_nameget_identifier get_model_ctcseZdZfddZZS)DeclarativeMetaclassc si|d<z!dd|D}||D]}t|dd}|r#|d|qWn ty.Ynwi}|D]\}} t| drP| j|vrHg|| j<|| j|q5i} |D]5\}} t| t r||} | || | |<t| ds| j r||vrt |} | j |d} | | | | | <qW|d| d|vrz t|dj|d<Wnttfytt|d<Ynwtt|||||S)NfieldscSsg|] }t|tr|qS) issubclass SearchIndex).0br"r"B/var/www/html/env/lib/python3.10/site-packages/haystack/indexes.py (sz0DeclarativeMetaclass.__new__.. facet_for)r)objectsMeta)reversegetattrupdate NameErroritemshasattrr)append isinstancerset_instance_namefacetedr facet_classr index_labelKeyErrorAttributeErrorrsuperr __new__)clsnamebasesattrsparentspr! facet_fields field_nameobj built_fieldsfieldshadow_facet_nameshadow_facet_field __class__r"r'r;#sT          zDeclarativeMetaclass.__new__)__name__ __module__ __qualname__r; __classcell__r"r"rIr'r "sr c@seZdZdZddZddZd*ddZd*d d Zd+d d Zd dZ d,ddZ ddZ ddZ ddZ d*ddZd*ddZd*ddZd*ddZd*d d!Zd*d"d#Zd$d%Zd&d'Zd(d)ZdS)-r$a Base class for building indexes. An example might look like this:: import datetime from haystack import indexes from myapp.models import Note class NoteIndex(indexes.SearchIndex, indexes.Indexable): text = indexes.CharField(document=True, use_template=True) author = indexes.CharField(model_attr='user') pub_date = indexes.DateTimeField(model_attr='pub_date') def get_model(self): return Note def index_queryset(self, using=None): return self.get_model().objects.filter(pub_date__lte=datetime.datetime.now()) cCsdd|_g}i|_|jD]\}}||j|j<|jdur!||q t|dks0td|j j dS)NTKThe index '%s' must have one (and only one) SearchField with document=True.) prepared_data field_mapr!r0index_fieldnamedocumentr2lenrrJrK)selfcontent_fieldsrCrFr"r"r'__init__us    zSearchIndex.__init__cCs td|)z Should return the ``Model`` class (not an instance) that the rest of the ``SearchIndex`` should use. This method is required & you must override it to return the correct class. z9You must provide a 'get_model' method for the '%r' index.)NotImplementedErrorrVr"r"r' get_modelszSearchIndex.get_modelNcC|jS)z Get the default QuerySet to index when doing a full update. Subclasses can override this method to avoid indexing certain objects. r[_default_managerallrVusingr"r"r'index_querysetszSearchIndex.index_querysetcCs |j|dS)z Get the default QuerySet for read actions. Subclasses can override this method to work with other managers. Useful when working with default managers that filter some objects. ra)rbr`r"r"r' read_querysets zSearchIndex.read_querysetc Csi}|}|}d|j}|r|r||d|<nt||r/|r*||d|<nt|d}t|dr@td|}n|j|d}t|dsQtd ||j d i| |j j j S) a7 Get the default QuerySet to index when doing an index update. Subclasses can override this method to take into account related model modification times. The default is to use ``SearchIndex.index_queryset`` and filter based on ``SearchIndex.get_updated_field`` z>No updated date field found for '%s' - not restricting by age.z%s__gtez%s__lteN get_querysetzd'SearchIndex.get_queryset' was deprecated in Haystack v2. Please rename the method 'index_queryset'.rcfilterzGThe '%r' class must return a 'QuerySet' in the 'index_queryset' method.r")r[get_updated_fieldrKwarningswarnr1rerbrrforder_by_metapkr=) rVra start_dateend_dateextra_lookup_kwargsmodel updated_fieldupdate_field_msgindex_qsr"r"r'build_querysets8       zSearchIndex.build_querysetcCs|tt|tt|tt|ji|_|j D]#\}}| ||j|j <t |d|r:t|d||}||j|j <q|jS)z? Fetches and adds/alters data before indexing. z prepare_%s)r rrrr[r rrlrQr!r0preparerSr1r-)rVrDrCrFvaluer"r"r'rus   zSearchIndex.prepareTcCs|||_|jD]I\}}|s |jdkr t|dd|jvr q t|ddrB|j|jj}|j|durB||jvrB|j||j|j<|jdurT|j|jdurT|j|j=q |jS)Nstringr)T) rurQr!r0 field_typer-r)rSnull)rVrDwith_string_facetrCrFsource_field_namer"r"r' full_prepares&      zSearchIndex.full_preparecCs,|jD]\}}|jdur|jSqdS)zCReturns the field that supplies the primary document to be indexed.TN)r!r0rTrS)rV_rFr"r"r'get_content_field s   zSearchIndex.get_content_fieldcCs,i}|jD] \}}|jr|j||<q|S)z+Returns a dict of fields with weight values)r!r0boost)rVweightsrCrFr"r"r'get_field_weightss  zSearchIndex.get_field_weightscCstdt||S)NzKSearchIndex._get_backend is deprecated; use SearchIndex.get_backend instead)rhriDeprecationWarning get_backendr`r"r"r' _get_backends  zSearchIndex._get_backendcCs>|durz tj|dd}Wn tyYdSwt|S)N)indexr)r for_write IndexErrorrrr`r"r"r'rs  zSearchIndex.get_backendcCs.||}|dur|||j|ddSdS)z Updates the entire index. If ``using`` is provided, it specifies which connection should be used. Default relies on the routers to decide which backend should be used. Nrc)rr.rbrVrabackendr"r"r'r.(s zSearchIndex.updatecKs>|j|fi|r||}|dur|||gdSdSdS)a Update the index for a single object. Attached to the class's post-save hook. If ``using`` is provided, it specifies which connection should be used. Default relies on the routers to decide which backend should be used. N) should_updaterr.rVinstancerakwargsrr"r"r' update_object6s  zSearchIndex.update_objectcKs,||}|dur|j|fi|dSdS)a  Remove an object from the index. Attached to the class's post-delete hook. If ``using`` is provided, it specifies which connection should be used. Default relies on the routers to decide which backend should be used. N)rremoverr"r"r' remove_objectFs zSearchIndex.remove_objectcCs,||}|dur|j|gddSdS)z Clears the entire index. If ``using`` is provided, it specifies which connection should be used. Default relies on the routers to decide which backend should be used. N)models)rclearr[rr"r"r'rTs zSearchIndex.clearcCs|j|d|j|ddS)z Completely clear the index for this model and rebuild it. If ``using`` is provided, it specifies which connection should be used. Default relies on the routers to decide which backend should be used. rcN)rr.r`r"r"r'reindexas zSearchIndex.reindexcCdS)ax Get the field name that represents the updated date for the model. If specified, this is used by the reindex command to filter out results from the QuerySet, enabling you to reindex only recent records. This method should either return None (reindex everything always) or a string of the Model's DateField/DateTimeField name. Nr"rZr"r"r'rgls zSearchIndex.get_updated_fieldcKr)aN Determine if an object should be updated in the index. It's useful to override this when an object may save frequently and cause excessive reindexing. You should check conditions on the instance and return False if it is not to be indexed. By default, returns True (always reindex). Tr")rVrrr"r"r'rws zSearchIndex.should_updatecCr\)ac Provides the ability to override how objects get loaded in conjunction with ``SearchQuerySet.load_all``. This is useful for post-processing the results from the query, enabling things like adding ``select_related`` or filtering certain data. By default, returns ``all()`` on the model's default manager. r]rZr"r"r'load_all_querysets zSearchIndex.load_all_querysetN)NNN)T)rKrLrM__doc__rXr[rbrdrtrur|r~rrrr.rrrrrgrrr"r"r"r'r$^s*  3        r$) metaclassc@seZdZedddZdS)BasicSearchIndexTrT use_templateN)rKrLrMr textr"r"r"r'rsrcCsh|}|dvr t}|S|dvrt}|S|dvr t}|S|dvr*t}|S|dvr2t}|S)zd Returns the Haystack field type that would likely be associated with each Django type. )r r)r NullBooleanField)CommaSeparatedIntegerField)rr)rPositiveIntegerFieldPositiveSmallIntegerFieldSmallIntegerField)get_internal_typerr rrr)fdefaultresultr"r"r'index_field_from_django_fields      rcs^eZdZdZedddZeeeddfZ dfdd Z d d Z d d Z d dZ dddZZS)ModelSearchIndexa) Introspects the model assigned to it and generates a `SearchIndex` based on the fields of that model. In addition, it adds a `text` field that is the `document=True` field and has `use_template=True` option set, just like the `BasicSearchIndex`. Usage of this class might result in inferior `SearchIndex` objects, which can directly affect your search results. Use this to establish basic functionality and move to custom `SearchIndex` objects for better control. At this time, it does not handle related fields. TrcontentrNcstd|_d|_g}|pi|_t|dd|_|jrs$   L <4