+h,-ddlmZmZmZGddZGddZGddeZGdd eZGd d Zy ) )ListOptionalUnionc eZdZdZdeddfdZdefdZd/dZd/dZd0d ed e eddfd Z d e ede fd Z d1d e e de e de e de eddf dZ d2d e e ede e eddfdZdeddfdZde ddfdZdeddfdZd/dZdeddfdZde efdZde efdZde d e ddfd!Zd/d"Zd/d#Zd/d$Zd/d%Zd/d&Zd e eddfd'Zd3d(Zd4d ed)e ddfd*Z!d+eddfd,Z"d-e ddfd.Z#y)5QueryaB Query is used to build complex queries that have more parameters than just the query string. The query string is set in the constructor, and other options have setter functions. The setter functions return the query object, so they can be chained, i.e. `Query("foo").verbatim().filter(...)` etc. query_stringreturnNcH||_d|_d|_d|_d|_d|_d|_d|_d|_d|_ t|_ d|_ d|_ d|_d|_d|_g|_g|_g|_d|_d|_d|_y)z Create a new query object. The query string is set in the constructor, and other options have setter functions. r FN) _query_string_offset_num _no_content _no_stopwords_fields _verbatim_with_payloads _with_scores_scorerlist_filters_ids_slop_timeout _in_order_sortby_return_fields_summarize_fields_highlight_fields _language _expander_dialect)selfrs N/var/www/html/venv/lib/python3.12/site-packages/redis/commands/search/query.py__init__zQuery.__init__s#/  !&#(,0 $$)"'&* "f )-  )- $.2 $&')')(,(,'+ c|jS)z+Return the query string of this query only.)r r$s r%rzQuery.query_string,s!!!r'c||_|S)zTLimit the results to a specific set of pre-known document ids of any length.)r)r$idss r% limit_idszQuery.limit_ids0s  r'c0|xj|z c_|S)zAdd fields to return fields.)rr$fieldss r% return_fieldszQuery.return_fields6s v% r'fieldas_fieldcn|jj|||xjd|fz c_|S)zJAdd field to return fields (Optional: add 'AS' name to the field).AS)rappend)r$r1r2s r% return_fieldzQuery.return_field;s9 ""5)     D(#3 3  r'r/cF|sgSt|tr|gSt|SN) isinstancestrrr.s r%_mk_field_listzQuery._mk_field_listCs$I%fc2xDV Dr' context_len num_fragssepcdg}|j|}|r|dtt|g|zz }||dt|gz }||dt|gz }||d|gz }||_|S)a Return an abridged format of the field, containing only the segments of the field which contain the matching term(s). If `fields` is specified, then only the mentioned fields are summarized; otherwise all results are summarized. Server side defaults are used for each option (except `fields`) if not specified - **fields** List of fields to summarize. All fields are summarized if not specified - **context_len** Amount of context to include with each fragment - **num_frags** Number of fragments per document - **sep** Separator string to separate fragments SUMMARIZEFIELDSLENFRAGS SEPARATOR)r;r:lenr)r$r/r<r=r>argss r% summarizezQuery.summarizeHs.}$$V,  Xs3v;/069 9D  " UC ,- -D  Wc)n- -D ? [#& &D!% r'tagscdg}|j|}|r|dtt|g|zz }|r|dgt|zz }||_|S)a Apply specified markup to matched term(s) within the returned field(s). - **fields** If specified then only those mentioned fields are highlighted, otherwise all fields are highlighted - **tags** A list of two strings to surround the match. HIGHLIGHTrATAGS)r;r:rErr )r$r/rHrFs r% highlightzQuery.highlightns`}$$V,  Xs3v;/069 9D  VHtDz) )D!% r'languagec||_|S)z Analyze the query as being in the specified language. :param language: The language (e.g. `chinese` or `english`) )r!)r$rMs r%rMzQuery.language " r'slopc||_|S)zpAllow a maximum of N intervening non matched terms between phrase terms (0 means exact phrase). )r)r$rPs r%rPz Query.slops  r'timeoutc||_|S)z-overrides the timeout parameter of the module)r)r$rRs r%rRz Query.timeouts  r'cd|_|S)z Match only documents where the query terms appear in the same order in the document. i.e. for the query "hello world", we do not match "world hello" T)rr)s r%in_orderzQuery.in_orders  r'scorerc||_|S)z Use a different scoring function to evaluate document relevance. Default is `TFIDF`. :param scorer: The scoring function to use (e.g. `TFIDF.DOCNORM` or `BM25`) )r)r$rVs r%rVz Query.scorers  r'c|jg}||jz }||j|jzz }|d|j|j gz }|S)z:Format the redis arguments for this query and return them.LIMIT)r _get_args_tagsrr rr)r$rFs r%get_argszQuery.get_argss\""# ##%% &&)?)??? $,, 22 r'cg}|jr|jd|jrD|jd|jt|j||jz }|jr|jd|j r|jd|j r;|j D],}t|ts td||jz }.|jr|jd|jr|d|jgz }|jr|jd|jrD|jd |jt|j||jz }|jd k\r|d |jgz }|j |d |j gz }|j"r|jd |j$rD|jd|jt|j$||j$z }|j&rOt|j&t(s td|jd||j&jz }|j*r|d|j*gz }|j,r|d|j,gz }|j.r|d|j.gz }|S)N NOCONTENTINFIELDSVERBATIM NOSTOPWORDSz Did not receive a Filter object. WITHPAYLOADSSCORER WITHSCORESINKEYSrSLOPTIMEOUTINORDERRETURNzDid not receive a SortByField.SORTBYLANGUAGEEXPANDERDIALECT)rr5rrErrrr9FilterAttributeErrorrFrrrrrrrrr SortbyFieldr!r"r#)r$rFflts r%rZzQuery._get_args_tagssQ    KK $ << KK # KKDLL) * DLL D >> KK #    KK & ==}} !!#v.()KLL  !    KK ' << Xt||, ,D    KK % 99 KK ! KKDII ' DII D ::? VTZZ( (D == $ Y . .D >> KK "    KK ! KKD//0 1 D'' 'D <<dllK8$%EFF KK ! DLL%% %D >> Z0 0D >> Z0 0D == Y . .D r'offsetnumc"||_||_|S)z Set the paging for the query (defaults to 0..10). - **offset**: Paging offset for the results. Defaults to 0 - **num**: How many results do we want )rr)r$rqrrs r%pagingz Query.pagings   r'cd|_|S)zWSet the query to be verbatim, i.e. use no query expansion or stemming. T)rr)s r%verbatimzQuery.verbatims r'cd|_|S)z>Set the query to only return ids and not the document content.T)rr)s r% no_contentzQuery.no_contents r'cd|_|S)z Prevent the query from being filtered for stopwords. Only useful in very big queries that you are certain contain no stopwords. T)rr)s r% no_stopwordszQuery.no_stopwordss " r'cd|_|S)z+Ask the engine to return document payloads.T)rr)s r% with_payloadszQuery.with_payloadss" r'cd|_|S)z0Ask the engine to return document search scores.T)rr)s r% with_scoreszQuery.with_scores s  r'c||_|S)z Limit the search to specific TEXT fields only. - **fields**: A list of strings, case sensitive field names from the defined schema. )rr.s r% limit_fieldszQuery.limit_fieldss  r'c<|jj||S)z Add a numeric or geo filter to the query. **Currently only one of each filter is supported by the engine** - **flt**: A NumericFilter or GeoFilter object, used on a corresponding field )rr5)r$rps r% add_filterzQuery.add_filters S! r'ascc(t|||_|S)z Add a sortby field to the query. - **field** - the name of the field to sort by - **asc** - when `True`, sorting will be done in asceding order )rorr$r1rs r%sort_byz Query.sort_by&s#5#.  r'expanderc||_|S)zg Add a expander field to the query. - **expander** - the name of the expander )r")r$rs r%rzQuery.expander0rOr'dialectc||_|S)zw Add a dialect field to the query. - **dialect** - dialect version to execute the query under )r#)r$rs r%rz Query.dialect9s    r')r rr8)NNNN)NN)rprmr rT)$__name__ __module__ __qualname____doc__r:r&rr,r0rr6rr;intrGrLrMrPfloatrRrUrVr[rZrtrvrxrzr|r~rrboolrrrr'r%rrs,S,T,<"c"  ##'ET#YE4E"&%)#'! $$c]$C= $ c] $  $NOStCy)8@c8K (u  S W $s)1S 1f S s w    DI' Stwswr'rc*eZdZdededeeddfdZy)rmkeywordr1rFr Nc.||gt|z|_yr8)rrF)r$rr1rFs r%r&zFilter.__init__Dse$tDz1 r')rrrr:rr&rr'r%rmrmCs'22C2S 2d2r'rmc NeZdZdZdZ d dedeeefdeeefdededd f d Z y ) NumericFilterz+infz-infr1minvalmaxval minExclusive maxExclusiver NcX|s|nd||s|nd|g}tj|d|g|y)N(FILTERrmr&)r$r1rrrrrFs r%r&zNumericFilter.__init__Ls<'FaxL&FaxL  h55r')FF) rrrINFNEG_INFr:rrrr&rr'r%rrHsc CG#" 6 6c3h 6c3h 6  6  6  6r'rc @eZdZdZdZdZdZefdedededed ed d f d Z y ) GeoFiltermkmftmir1lonlatradiusunitr Nc :tj|d|||||y)N GEOFILTERr)r$r1rrrrs r%r&zGeoFilter.__init__bs k5#sFDIr') rrrMETERS KILOMETERSFEETMILESr:rr&rr'r%rr\sY FJ D ENXJJ$J+0J:?JGJJ Jr'rceZdZddeddfdZy)ror1r Nc,||r dg|_ydg|_y)NASCDESC)rFrs r%r&zSortbyField.__init__isSE5 f5 r'r)rrrr:r&rr'r%rorohs6c66r'roN) typingrrrrrmrrrorr'r%rsF((||~ 22 6F6( J J66r'