keJi:-ddlmZmZmZmZddlmZeZGddZ GddZ GddZ Gd d e Z Gd d e Z Gd dZGddZGddZdS))ListOptionalTupleUnion)DEFAULT_DIALECTc*eZdZddededdfdZdZdS) LimitroffsetcountreturnNc"||_||_dSN)r r )selfr r s g/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/redis/commands/search/aggregation.py__init__zLimit.__init__ s  cf|jr)dt|jt|jgSgS)NLIMIT)r strr rs r build_argszLimit.build_args s0 : S--s4:? ?Ir)rr__name__ __module__ __qualname__intrrrrr r sQss4rr cbeZdZdZdZdeddfdZdeddfdZede edffd Z dS) Reducerzr Base reducer object for all reducers. See the `redisearch.reducers` module for the actual reducers. Nargsr c0||_d|_d|_dSr)_args_field_aliasrr s rrzReducer.__init__s&* %) %) raliascp|tur%|jstd|jdd}||_|S)a Set the alias for this reducer. ### Parameters - **alias**: The value of the alias for this reducer. If this is the special value `aggregation.FIELDNAME` then this reducer will be aliased using the same name as the field upon which it operates. Note that using `FIELDNAME` is only possible on reducers which operate on a single field value. This method returns the `Reducer` object making it suitable for chaining. z(Cannot use FIELDNAME alias with no fieldN) FIELDNAMEr# ValueErrorr$)rr&s rr&z Reducer.alias"sE I  ; ( !KLLL ABB  r.c|jSr)r"rs rr z Reducer.args:s zr) rrr__doc__NAMErrr&propertyrr rrrrrs D*c*d**** 390eCHoXrrc>eZdZUdZdZeeed<deddfdZdS) SortDirectionz@ This special class is used to indicate sort direction. N DIRSTRINGfieldr c||_dSr)r2)rr2s rrzSortDirection.__init__Fs  r) rrrr,r1rr__annotations__rrrrr0r0?sX $Ix}###cdrr0ceZdZdZdZdS)AsczK Indicate that the given field should be sorted in ascending order ASCNrrrr,r1rrrr6r6JsIIIrr6ceZdZdZdZdS)DesczL Indicate that the given field should be sorted in descending order DESCNr8rrrr:r:RsIIIrr:c"eZdZdZddeddfdZdeddfdZdeeeefd e ddfd Z d d Z d e d e ddfdZ deddfdZdeeeefddfdZd dZd dZdeddfdZd dZd!de deddfdZdeefdZde ddfdZdS)"AggregateRequestzH Aggregation request which can be passed to `Client.aggregate`. *queryr Nc||_g|_g|_d|_d|_d|_d|_g|_t|_ d|_ d|_ dS)a} Create an aggregation request. This request may then be passed to `client.aggregate()`. In order for the request to be usable, it must contain at least one group. - **query** Query string for filtering records. All member methods (except `build_args()`) return the object itself, making them useful for chaining. FrTFIDFN) _query_aggregateplan _loadfields_loadall_max _with_schema _verbatim_cursorr_dialect _add_scores_scorer)rr?s rrzAggregateRequest.__init___sY! )+&(#  "'$"$ , !&# rfieldscN|r|j|nd|_|S)aC Indicate the fields to be returned in the response. These fields are returned in addition to any others implicitly specified. ### Parameters - **fields**: If fields not specified, all the fields will be loaded. Otherwise, fields should be given in the format of `@field`. T)rDextendrE)rrMs rloadzAggregateRequest.loadxs2  !   # #F + + + + DM rreducersc jt|tr|gn|}dtt|g|}|D][}|d|jtt|jgz }||j|j |d|jgz }\|j||S)a| Specify by which fields to group the aggregation. ### Parameters - **fields**: Fields to group by. This can either be a single string, or a list of strings. both cases, the field should be specified as `@field`. - **reducers**: One or more reducers. Reducers may be found in the `aggregation` module. GROUPBYREDUCENAS) isinstancerlenr-r rOr$rC)rrMrQretreducers rgroup_byzAggregateRequest.group_bys(44@&&#c&kk**4V4 . .G HglCGL0A0A,B,BC CC JJw| $ $ $~)gn-- ""3''' rc |D],\}}d|g}||d|gz }|j|-|S)aQ Specify one or more projection expressions to add to each result ### Parameters - **kwexpr**: One or more key-value pairs for a projection. The key is the alias for the projection, and the value is the projection expression itself, for example `apply(square_root="sqrt(@foo)")` APPLYNrU)itemsrCrO)rkwexprr&exprrXs rapplyzAggregateRequest.applys["<<>> , ,KE4D/C e}$   & &s + + + + rr numc~t||}|j||S)a Sets the limit for the most recent group or query. If no group has been defined yet (via `group_by()`) then this sets the limit for the initial pool of results from the query. Otherwise, this limits the number of items operated on from the previous group. Setting a limit on the initial search results may be useful when attempting to execute an aggregation on a sample of a large data set. ### Parameters - **offset**: Result offset from which to begin paging - **num**: Number of results to return Example of sorting the initial results: ``` AggregateRequest("@sale_amount:[10000, inf]") .limit(0, 10) .group_by("@state", r.count()) ``` Will only group by the states found in the first 10 results of the query `@sale_amount:[10000, inf]`. On the other hand, ``` AggregateRequest("@sale_amount:[10000, inf]") .limit(0, 1000) .group_by("@state", r.count() .limit(0, 10) ``` Will group all the results matching the query, but only return the first 10 groups. If you only wish to return a *top-N* style query, consider using `sort_by()` instead. )r rCrOr)rr ra_limits rlimitzAggregateRequest.limits<Tvs## ""6#4#4#6#6777 rcvg}|D]6}t|ttfr||j|jgz }0||gz }7dt t |g}|||dd}|dkr|dt |gz }|j ||S)a, Indicate how the results should be sorted. This can also be used for *top-N* style queries ### Parameters - **fields**: The fields by which to sort. This can be either a single field or a list of fields. If you wish to specify order, you can use the `Asc` or `Desc` wrapper classes. - **max**: Maximum number of results to return. This can be used instead of `LIMIT` and is also faster. Example of sorting by `foo` ascending and `bar` descending: ``` sort_by(Asc("@foo"), Desc("@bar")) ``` Return the top 10 customers: ``` AggregateRequest() .group_by("@customer", r.sum("@paid").alias(FIELDNAME)) .sort_by(Desc("@paid"), max=10) ``` SORTBYmaxrMAX) rVr6r:r2r1rrWrOgetrC)rrMkwargs fields_argsfrXrgs rsort_byzAggregateRequest.sort_bys:  # #A!c4[)) #55 s" S--../ ;jj"" 77 E3s88$ $C ""3''' r expressionscxt|tr|g}|D]}|jd|g|S)z Specify filter for post-query results using predicates relating to values in the result set. ### Parameters - **fields**: Fields to group by. This can either be a single string, or a list of strings. FILTER)rVrrCrO)rrn expressions rfilterzAggregateRequest.filtersP k3 ' ' (&-K% ? ?J   & &*'= > > > > rcd|_|S)z| If set, the `schema` property will contain a list of `[field, type]` entries in the result object. T)rGrs r with_schemazAggregateRequest.with_schema!s ! rcd|_|S)zM If set, includes the score as an ordinary field of the row. T)rKrs r add_scoreszAggregateRequest.add_scores)s  rscorerc||_|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`) )rL)rrws rrwzAggregateRequest.scorer0s  rcd|_|S)NT)rHrs rverbatimzAggregateRequest.verbatim;s rrr max_idlecxdg}|r|dt|gz }|r|dt|dzgz }||_|S)N WITHCURSORCOUNTMAXIDLEi)rrI)rr r|r s rcursorzAggregateRequest.cursor?sU~  * Wc%jj) )D  6 YHtO 4 45 5D  rc|jg}|jr|d|jr|d|jr|d|jg|jr|d|jr ||jz }|jr+|d|dnj|j rc|d|tt|j ||j |j r)|dt|j g||j |S)N WITHSCHEMAVERBATIMSCORER ADDSCORESLOADr>DIALECT)rBrGappendrHrLrOrKrIrErDrrWrJrC)rrXs rrzAggregateRequest.build_argsHsc{m   % JJ| $ $ $ > # JJz " " " < 1 JJ$,/ 0 0 0   $ JJ{ # # # < 4< C = ) JJv    JJsOOOO   ) JJv    JJs3t/0011 2 2 2 JJt' ( ( ( = 8 JJ 3t}#5#56 7 7 7 4&''' rdialectc||_|S)z Add a dialect field to the aggregate command. - **dialect** - dialect version to execute the query under )rJ)rrs rrzAggregateRequest.dialectks    r)r>)r r=)rr{)rrrr,rrrPrrrrZr`rrdrmrrrtrvrwrzfloatrrrrrrr=r=Zs$$c$D$$$$2C$6 CcN+8? 4$,C,c,.@,,,,\+s+1C++++Z%T#Y"7)rr __class__ridrWrrs r__repr__zAggregateResult.__repr__sm!%4dkoo" 4' 4 4r$xx < 4 4 NN 4 4-0 4 4 4 r)rrrrrrrrrrrrsQV  #      rrN)typingrrrrredis.commands.search.dialectrobjectr)r rr0r6r:r=rrrrrrs////////////999999 FHH         ((((((((V-=XXXXXXXXv                   r