NwgeddlZddlZddlmZddlmZmZddlmZddl m Z m Z ddl m Z ddlmZmZddlmZGd d ZGd d eZGd deZGddeZGddeZy)N)reduce)connection_router connections)SQ)DEFAULT_OPERATORITERATOR_LOAD_PER_QUERY) NotHandled) AutoQueryRaw)logcNeZdZdZd6dZdZdZdZdZdZ d Z d Z d Z d Z d ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ d Z!d7d!Z"d"Z#d8d#Z$d$Z%d%Z&d&Z'd'Z(d9d(Z)d)Z*d*Z+d+Z,d,Z-d-Z.d.Z/d/Z0d0Z1d1Z2d7d2Z3d3Z4d4Z5d7d5Z6y):SearchQuerySetz Provides a way to specify search parameters and lazily load results. Supports chaining (a la QuerySet) to narrow the search. Nc||_d|_|j|||_g|_d|_d|_d|_d|_tjd|_ y)NFrhaystack) _usingquery_determine_backend _result_cache _result_count _cache_full _load_all_ignored_result_countlogging getLoggerr )selfusingrs E/var/www/horilla/myenv/lib/python3.12/site-packages/haystack/query.py__init__zSearchQuerySet.__init__sf   !  DJ! %&"$$Z0ct|j't|jj|_yi}|jr|jj|d<t j di|}|jr!|jj||_yt|j|_y)Nmodels)rr get_queryrr!rfor_readr)rhints backend_aliass rrz!SearchQuerySet._determine_backend's ;; "$T[[1;;=DJ  ::"jj//E(O)22;U;  ::))-8DJ$]3==?DJrcdt||jj}d|d<d|d<|S)z For pickling. N_iterr )len__dict__copy)robj_dicts r __getstate__zSearchQuerySet.__getstate__<s5 D ==%%' rcF||_tjd|_y)z! For unpickling. rN)r*rrr )r data_dicts r __setstate__zSearchQuerySet.__setstate__Fs" $$Z0rc<d|jd|jdS)Nz)rrrs r__repr__zSearchQuerySet.__repr__Ms9=T[[QQrc|j2|jj|_|jsd|_|j|jz SNr)rr get_countrr3s r__len__zSearchQuerySet.__len__PsO    %!%!5!5!7D %%%&"!!D$>$>>>rcl|jrt|jS|jSN)_cache_is_fulliterr _manual_iterr3s r__iter__zSearchQuerySet.__iter__[s.    **+ +  ""rct|tr|jS|j}|jj |jt j |Sr:) isinstanceEmptySearchQuerySet_clonercombinerANDrothercombineds r__and__zSearchQuerySet.__and__bsE e0 1<<> !;;=u{{BFF3rc|j}t|tr|S|jj |jt j |Sr:)rBr@rArrCrORrEs r__or__zSearchQuerySet.__or__is>;;= e0 1Ou{{BEE2rc|jjsyt|dkry |jj dy#t $rt|jdkDcYSwxYw)NFrT)rhas_runr)rindex ValueErrorr3s rr;zSearchQuerySet._cache_is_fullpsdzz!!# t9> /    $ $T * /t))*Q. . /sA!A,+A,c#^Kd}d} t|jdkDr |jjd}||kr|j||dz }||kr|j ry|j ||t zsy#t$rt|j}YmwxYww)Nr)r)rrNrOr; _fill_cacher)rcurrent_positioncurrent_cache_maxs rr=zSearchQuerySet._manual_iter~s 4%%&*@(,(:(:(@(@(F%#%66(()9:: A% #%66""$## "25L"L'"@(+D,>,>(?%@s-B-B B-,B- B*'B-)B**B-cg}|jr_i}i}|D]7}|j|jgj|j9|D]}|j |||||<|D]}|jrj |ji}|rJttt|}||j|_ ||j|_ n9|jdt|=|xjt|z c_ |S|j||S#t$r2|xjdz c_ |jjYwxYw)NrQ)r setdefaultmodelappendpk_load_model_objectsgettypenextr<_objectKeyErrorrrpopr)) rresultsto_cache models_pksloaded_objectsresultrW model_objects result_klasss rpost_process_resultsz#SearchQuerySet.post_process_resultsss >>JN" J%%fllB7>>vyyI J$ (,(@(@:e,)u%   $F~~ . 2 26<< D !#'T--@(A#BL ,VYY 7FI !)6vyy)A**>S\>:..#g,>. OOF #9 $<'$!22a72**..0 !sD558E0/E0cv t|jjj}|j |}|j |jj}|j |S#t$r:|jjd||jj |cYSwxYw)N)rz&Model '%s' not handled by the routers.) rrrget_unified_index get_index read_querysetin_bulkr r warning_default_manager)rrWpksuirNobjectss rrZz"SearchQuerySet._load_model_objectss 7TZZ../AACBLL'E)) 0A0A)BG??3' ' 7 HH  Eu M))11#6 6 7sA2A55AB87B8c |jj|d}|}||jz }|}|||jz }|jj|||jjdi|}|t |dk(r|j d||_yt |j dk(r#dg|jjz|_||}}||jj}|} |j|} | |j | | t | zd|j ||vr|}|tz }| t | z } |jj|jj|||jj }|t |dk(r|j d| |_ y y)NrFTr") r_resetr set_limits get_resultsr)rr7rhr) rstartendkwargs query_start query_endra fill_startfill_end cache_startrbs rrRzSearchQuerySet._fill_caches  =E t111   33 3I k95($**((262 ?c'la/!%!3!3FU!;D  t!! "a '"&$***>*>*@!@D $cH  zz++-H 009HMUD  {[3x=-H It))%44% 33s8},  !!# %%j(;**002?c'la&7)-););L[)ID&/rct|ttfstt|ts|dk\sMt|tr6|j|jdk\r|j |j dk\sJdt|tr3d}|j}|j t|j }n d}n d}|}|dz}t |jdks!d|j||vr#|js |j|||r|j||S|j|S#t$rY+wxYw)zE Retrieves an item or slice from the set of results. rNz#Negative indexing is not supported.TFrQ) r@sliceint TypeErrorrwstopr)rr;rR StopIteration)rkis_slicerwbounds r __getitem__zSearchQuerySet.__getitem__s;!eS\*Oq%(a1f q% AGGqL166Q; 1 1  1  a HGGEvv!AFF HEEE t!! "a ' D&&uU3 3D>4262 2t///rcp|j}|jjt|i||S)CNarrows the search by ensuring certain attributes are not included.rBr add_filterrrrryclones rexcludezSearchQuerySet.excludeMs1  D 3F 334 rcn|j}|jjt|i||S)zENarrows the search by looking for (and including) certain attributes.rrs rrzSearchQuerySet.filter_andSs.  r42623 rcr|j}|jjt|i|d|S)rT)use_orrrs rrzSearchQuerySet.filter_orYs3  r42624@ rcj|j}|D]}|jj||S)z4Alters the order in which the results should appear.)rBr add_order_by)rrrfields rorder_byzSearchQuerySet.order_by_s5  ,E KK $ $U + , rc ^|j}|jjdi||S)z!Adds highlighting to the results.r")rBr add_highlight)rryrs r highlightzSearchQuerySet.highlighths) ! !!+F+ rc|j}|D]q}|t|jjj j vrt jd|d|jj|s|S)zFAccepts an arbitrary number of Model classes to include in the search.z The model z is not registered for search.) rBrrrrjget_indexed_modelswarningswarn add_model)rr!rrWs rr!zSearchQuerySet.modelsnsr  )E"4::#4#45""$##%&  eUV KK ! !% ( ) rc\|j}|jj||S)z Allows specifying a different class to use for results. Overrides any previous usages. If ``None`` is provided, Haystack will revert back to the default ``SearchResult`` object. )rBrset_result_class)rrrs r result_classzSearchQuerySet.result_classs&  $$U+ rc^|j}|jj|||S)z%Boosts a certain aspect of the query.)rBr add_boost)rtermboostrs rrzSearchQuerySet.boosts&  dE* rc `|j}|jj|fi||S)z0Adds faceting to a query for the provided field.)rBradd_field_facet)rroptionsrs rfacetzSearchQuerySet.facets+ # ##E5W5 rc`|j}|jj||||S)z1Spatial: Adds a bounding box search to the query.)rBr add_within)rrpoint_1point_2rs rwithinzSearchQuerySet.withins(  ugw7 rc`|j}|jj||||S)z3Spatial: Adds a distance-based search to the query.)rBr add_dwithin)rrpointdistancers rdwithinzSearchQuerySet.dwithins(  uh7 rc(|j|dS)z-Adds stats to a query for the provided field.N) facet_fields) stats_facet)rrs rstatszSearchQuerySet.statssD99rc|j}g} |jt|g|jj |||S#t$r|r|j|Yv#>{??rc4|j}d|_|S)z8Efficiently populates the objects in the search results.T)rBrrrs rload_allzSearchQuerySet.load_alls  rc@|t|i}|jdi|S)z Performs a best guess constructing the search query. This method is somewhat naive but works well enough for the simple, common cases. r")r r)rr fieldnamerys r auto_queryzSearchQuerySet.auto_querys'Y|45t{{$V$$rc h|j}g}|jD]e\}}|jdD]L}|jj |j }|s/||i}|j tdi|Ng|jttj|S)z A shortcut method to perform an autocomplete search. Must be run against fields that are either ``NgramField`` or ``EdgeNgramField``.  r") rBitemssplitrcleanstriprXrrroperatorrH)rryr query_bits field_namerwordbits r autocompletezSearchQuerySet.autocompletes  !' 4 J C( 4kk'' 5(#.F%%bl6l3  4 4||F8#3#3Z@AArct|j}|jj||_||_|S)ze Allows switching which connection the ``SearchQuerySet`` uses to search in. )rBrrr)rconnection_namers rrzSearchQuerySet.usings1  jj&&7 &  rct|S)z-Returns the total number of matching results.)r)r3s rcountzSearchQuerySet.count s 4yrc |dS)z:Returns the best/top search result that matches the query.rr"r3s r best_matchzSearchQuerySet.best_matchs Awrc|j}|jj|jjd|z|j S)z=Returns the most recent search result that matches the query.z-%s)rBrclear_order_byrr)r date_fieldrs rlatestzSearchQuerySet.latestsC  ""$   !34!!rc\|j}|jj||S)z.Finds similar results to the object passed in.)rBrmore_like_this)rmodel_instancers rrzSearchQuerySet.more_like_thiss$  "">2 rc|jjr|jjS|j}|jjS)z Returns the facet counts found by the query. This will cause the query to execute and should generally be used when presenting the data. )rrMget_facet_countsrBrs r facet_countszSearchQuerySet.facet_countssE ::   ::..0 0KKME;;//1 1rc|jjr|jjS|j}|jjS)z? Returns the stats results found by the query. )rrM get_statsrBrs r stats_resultszSearchQuerySet.stats_results,sE ::   ::'') )KKME;;((* *rc\|j}|jj||S)a`Set the exact text to be used to generate spelling suggestions When making complicated queries, such as the alt parser mechanism used by Solr dismax/edismax, this provides a convenient way to set the a simple text string which will be used to generate spelling suggestions without including unnecessary syntax. )rBrset_spelling_query)rspelling_queryrs rrz!SearchQuerySet.set_spelling_query6s&  &&~6 rc|jjr|jj|S|j}|jj|S)aN Returns the spelling suggestion found by the query. To work, you must set ``INCLUDE_SPELLING`` within your connection's settings dictionary to ``True``. Otherwise, ``None`` will be returned. This will cause the query to execute and should generally be used when presenting the data. )rrMget_spelling_suggestionrB)rpreferred_queryrs rspelling_suggestionz"SearchQuerySet.spelling_suggestionBsI ::   ::55oF FKKME;;66G Grch|jt}|jj||S)z Returns a list of dictionaries, each containing the key/value pairs for the result, exactly like Django's ``ValuesQuerySet``. r)rBValuesSearchQuerySet_fieldsextend)rfieldsqss rvalueszSearchQuerySet.valuesRs, [[3[ 4 &! rc|jdd}|rt|dkDr td|jt}|j j |||_|S)a) Returns a list of field values as tuples, exactly like Django's ``QuerySet.values``. Optionally accepts a ``flat=True`` kwarg, which in the case of a single field being provided, will return a flat list of that field rather than a list of tuples. flatFrQzH'flat' is not valid when values_list is called with more than one field.r)r`r)rrBValuesListSearchQuerySetr r _flat)rrryrrs r values_listzSearchQuerySet.values_list[sbzz&%( CK!OZ [[7[ 8 &! rc| |j}|jj}||}|j|_|S)N)r) __class__rrBr)rrrrs rrBzSearchQuerySet._cloners< =NNE !!#E".. r)NNr:)rQr)7__name__ __module__ __qualname____doc__rrr-r0r4r8r>rHrKr;r=rhrZrRrrrrrrrrrr!rrrrrrrrrrrrrrrrrrrrrrrr rrrBr"rrrr s 1&@*1R ?# /:0d 7:x()V60     "     :  @ %B&" 2+ H .rrc<eZdZdZdZdZdfd ZdZdZxZ S)rAz\ A stubbed SearchQuerySet that behaves as normal but always returns no results. cyr6r"r3s rr8zEmptySearchQuerySet.__len__srcy)NTr"r3s rr;z"EmptySearchQuerySet._cache_is_fullsrc6t||}g|_|SNr)superrBrrrrrs rrBzEmptySearchQuerySet._clones U+  rcy)NFr")rrwrxs rrRzEmptySearchQuerySet._fill_cachesrciSr:r"r3s rrz EmptySearchQuerySet.facet_countss rr:) rrrrr8r;rBrRr __classcell__rs@rrArA|s!  rrAc>eZdZdZfdZdfd ZfdZdZxZS)rz A ``SearchQuerySet`` which returns a list of field values as tuples, exactly like Django's ``ValuesListQuerySet``. cRt||i|d|_g|_gd|_y)NF)id django_ct django_idscore)r!rrr _internal_fieldsrrryrs rrz!ValuesListSearchQuerySet.__init__s. $)&)  !Jrclt||}|j|_|j|_|Sr )r!rBr rr"s rrBzValuesListSearchQuerySet._clones.U+  jj  rct|j}|j|jd|i}t |||fi|SNr)setr-updater r!rRrrwrx query_fieldsryrs rrRz$ValuesListSearchQuerySet._fill_cachesF4001 DLL)L)w"5#888rc g}|jr |j}n |j}|D]-}||jDcgc]}t ||dc}/|Scc}wr:)rr rXr getattr)rrarbaccumreis rrhz-ValuesListSearchQuerySet.post_process_resultss^ ::OOEOOE DF T\\B761d+B C DCsA r:) rrrrrrBrRrhr%r&s@rrrs J 9 rrc(eZdZdZfdZdZxZS)r z A ``SearchQuerySet`` which returns a list of dictionaries, each containing the key/value pairs for the result, exactly like Django's ``ValuesQuerySet``. ct|j}|j|jd|i}t t |||fi|Sr1)r2r-r3r r!rrRr4s rrRz ValuesSearchQuerySet._fill_cachesI4001 DLL)L)-t@VvVVrc g}|D]7}|j|jDcic]}|t||dc}9|Scc}wr:)rXr r7)rrarbrer9s rrhz)ValuesSearchQuerySet.post_process_resultssN QF OO$,,OQQ4 88O P QPsA )rrrrrRrhr%r&s@rr r s W rr c:eZdZdZfdZdZdZdfd ZxZS)RelatedSearchQuerySetzO A variant of the SearchQuerySet that can handle `load_all_queryset`s. c@t||i|i|_g|_yr:)r!r_load_all_querysetsrr.s rrzRelatedSearchQuerySet.__init__s% $)&)#% rc8||jvr|j|j|S t|jjj }|j |}|j}|j|S#t$ricYSwxYwr:) r@rmrrrrjrkload_all_querysetr )rrWrprqrNrs rrZz)RelatedSearchQuerySet._load_model_objectss D,, ,++E2::3? ?  !2!23EEG U+,,.zz#&   sAB BBcD|j}||j|<|S)a( Allows for specifying a custom ``QuerySet`` that changes how ``load_all`` will fetch records for the provided model. This is useful for post-processing the results from the query, enabling things like adding ``select_related`` or filtering certain data. )rBr@)rrWquerysetrs rrBz'RelatedSearchQuerySet.load_all_querysets$ +3!!%( rcJt||}|j|_|Sr )r!rBr@r"s rrBzRelatedSearchQuerySet._clones&U+$($<$<! rr:) rrrrrrZrBrBr%r&s@rr>r>s! $ rr>)rr functoolsrrrrhaystack.backendsrhaystack.constantsrrhaystack.exceptionsr haystack.inputsr r haystack.utilsr rrrArr r>r"rrrLs_3 H**)l l ^.2'~'T3,+N+r