o zqf*ã@sŽddlmZmZeƒZGdd„dƒZGdd„dƒZGdd„dƒZGdd „d eƒZGd d „d eƒZ Gd d „d ƒZ Gdd„dƒZ Gdd„dƒZ dS)é)ÚListÚUnionc@s,eZdZd dededdfdd„Zdd „ZdS) ÚLimitrÚoffsetÚcountÚreturnNcCs||_||_dS©N)rr)Úselfrr©r ú`/root/parts/websockify/install/lib/python3.10/site-packages/redis/commands/search/aggregation.pyÚ__init__s zLimit.__init__cCs |jrdt|jƒt|jƒgSgS)NÚLIMIT)rÚstrr©r r r r Ú build_args szLimit.build_args)rr©Ú__name__Ú __module__Ú __qualname__Úintr rr r r r rs rc@sReZdZdZdZdeeddfdd„Zdeddfdd „Ze deefd d „ƒZ dS) ÚReducerzr Base reducer object for all reducers. See the `redisearch.reducers` module for the actual reducers. NÚargsrcGs||_d|_d|_dSr)Ú_argsÚ_fieldÚ_alias©r rr r r r ó zReducer.__init__ÚaliascCs.|tur|js tdƒ‚|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 fieldéN)Ú FIELDNAMErÚ ValueErrorr)r rr r r r s z Reducer.aliascCs|jSr)rrr r r r7sz Reducer.args) rrrÚ__doc__ÚNAMErrr rÚpropertyrr r r r rsrc@s&eZdZdZdZdeddfdd„ZdS)Ú SortDirectionz@ This special class is used to indicate sort direction. NÚfieldrcCs ||_dSr)r%)r r%r r r r Cs zSortDirection.__init__)rrrr!Ú DIRSTRINGrr r r r r r$<sr$c@óeZdZdZdZdS)ÚAsczK Indicate that the given field should be sorted in ascending order ÚASCN©rrrr!r&r r r r r(Gór(c@r')ÚDesczL Indicate that the given field should be sorted in descending order ÚDESCNr*r r r r r,Or+r,c@seZdZdZd(deddfdd„Zdeeddfd d „Zdeed ee ee fddfd d „Z d)dd„Z de de ddfdd„Z deeddfdd„Zdeeeefddfdd„Zd)dd„Zd)dd„Zd*de d eddfd!d"„Zdeefd#d$„Zd%e ddfd&d'„ZdS)+ÚAggregateRequestzH Aggregation request which can be passed to `Client.aggregate`. Ú*ÚqueryrNcCs:||_g|_g|_d|_d|_d|_d|_g|_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. FrN) Ú_queryÚ_aggregateplanÚ _loadfieldsÚ_loadallÚ_maxÚ _with_schemaÚ _verbatimÚ_cursorÚ_dialect)r r0r r r r \s  zAggregateRequest.__init__ÚfieldscGs|r |j |¡|Sd|_|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)r3Úextendr4)r r:r r r Úloadss  ÿzAggregateRequest.loadÚreducerscGs”t|tƒr|gn|}t|tƒr|gn|}dtt|ƒƒg|¢}|D]!}|d|jtt|jƒƒg7}| |j¡|jdurA|d|jg7}q |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. ÚGROUPBYÚREDUCENÚAS) Ú isinstancerrÚlenr"rr;rr2)r r:r=ÚretÚreducerr r r Úgroup_byƒs  € zAggregateRequest.group_bycKs>| ¡D]\}}d|g}|dur|d|g7}|j |¡q|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)")` ÚAPPLYNr@)Úitemsr2r;)r ÚkwexprrÚexprrCr r r Úapplyžs   zAggregateRequest.applyrÚnumcCst||ƒ}|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. )rr2r;r)r rrKÚ_limitr r r Úlimit°s *zAggregateRequest.limitcOs–t|ttfƒr |g}g}|D]}t|tƒr||j|jg7}q||g7}qdtt|ƒƒg}| |¡| dd¡}|dkrC|dt|ƒg7}|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) ``` ÚSORTBYÚmaxrÚMAX) rArr$r%r&rBr;Úgetr2)r r:ÚkwargsÚ fields_argsÚfrCrOr r r Úsort_byÞs     zAggregateRequest.sort_byÚ expressionscCs.t|tƒr|g}|D] }|j d|g¡q |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)rArr2r;)r rVÚ expressionr r r Úfilter s zAggregateRequest.filtercCó d|_|S)z| If set, the `schema` property will contain a list of `[field, type]` entries in the result object. T)r6rr r r Ú with_schemaszAggregateRequest.with_schemacCrZ)NT)r7rr r r Úverbatim'szAggregateRequest.verbatimrçrÚmax_idlecCs<dg}|r |dt|ƒg7}|r|dt|dƒg7}||_|S)NÚ WITHCURSORÚCOUNTÚMAXIDLEiè)rr8)r rr^rr r r Úcursor+szAggregateRequest.cursorcCsª|jg}|jr | d¡|jr| d¡|jr||j7}|jr*| d¡| d¡n|jrB| d¡| tt|jƒƒ¡|  |j¡|j rM|  d|j g¡|  |j ¡|S)NÚ WITHSCHEMAÚVERBATIMÚLOADr/ÚDIALECT) r1r6Úappendr7r8r4r3rrBr;r9r2)r rCr r r r4s$        zAggregateRequest.build_argsÚdialectcCs ||_|S)zƒ Add a dialect field to the aggregate command. - **dialect** - dialect version to execute the query under )r9)r rhr r r rhPszAggregateRequest.dialect)r/)rr.)rr])rrrr!rr rr<rrrErJrrMrUrYr[r\Úfloatrbrrhr r r r r.Ws(ÿÿ þ ./   r.c@s&eZdZdeddfdd„Zdd„ZdS)ÚCursorÚcidrNcCs||_d|_d|_dS)Nr)rkr^r©r rkr r r r [rzCursor.__init__cCs@t|jƒg}|jr|dt|jƒg7}|jr|dt|jƒg7}|S)Nrar`)rrkr^rrr r r r`s zCursor.build_argsrr r r r rjZs rjc@s0eZdZdeddfdd„Zdeeffdd„ZdS)ÚAggregateResultrbrNcCs||_||_||_dSr)ÚrowsrbÚschema)r rnrbror r r r jrzAggregateResult.__init__c Cs@|jr|jjnd}d|jj›dt|ƒd›dt|jƒ›d|›d S)Néÿÿÿÿú)rbrkÚ __class__rÚidrBrnrlr r r Ú__repr__osÿÿÿzAggregateResult.__repr__)rrrrjr rrvr r r r rmisrmN) ÚtypingrrÚobjectrrrr$r(r,r.rjrmr r r r Ús *