U Œàôgø,ã@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(eZdZdeeddœdd„Zdd„ZdS) ÚLimitrN)ÚoffsetÚcountÚreturncCs||_||_dS©N)rr)Úselfrr©r úE/tmp/pip-unpacked-wheel-f3sx1i9r/redis/commands/search/aggregation.pyÚ__init__szLimit.__init__cCs$|jrdt|jƒt|jƒgSgSdS)NZLIMIT)rÚstrr©r r r r Ú build_args szLimit.build_args)rr©Ú__name__Ú __module__Ú __qualname__Úintr rr r r r rsrc@sNeZdZdZdZeeddœdd„Zeddœdd„Ze eed œd 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 szReducer.__init__)ÚaliasrcCs.|tkr$|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 fieldéN)Ú FIELDNAMErÚ ValueErrorr)r rr r r r s z Reducer.alias©rcCs|jSr)rrr r r r7sz Reducer.args) rrrÚ__doc__ÚNAMErr r rÚpropertyrr r r r rs rc@s$eZdZdZdZeddœdd„ZdS)Ú SortDirectionz@ This special class is used to indicate sort direction. N)ÚfieldrcCs ||_dSr)r$)r r$r r r r CszSortDirection.__init__)rrrr Ú DIRSTRINGr r r r r r r#<sr#c@seZdZdZdZdS)ÚAsczK Indicate that the given field should be sorted in ascending order ÚASCN©rrrr r%r r r r r&Gsr&c@seZdZdZdZdS)ÚDesczL Indicate that the given field should be sorted in descending order ZDESCNr(r r r r r)Osr)c@seZdZdZd+eddœdd„Zeeddœdd „Zeeee ee fdd œd d „Z dd œdd„Z e e ddœdd„Z eeddœdd„Zeeeefddœdd„Zdd œdd„Zdd œdd„Zeddœdd„Zdd œdd „Zd,e edd#œd$d%„Zeed œd&d'„Ze dd(œd)d*„ZdS)-ÚAggregateRequestzH Aggregation request which can be passed to `Client.aggregate`. Ú*N)ÚqueryrcCsF||_g|_g|_d|_d|_d|_d|_g|_d|_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. FrNZTFIDF) Ú_queryÚ_aggregateplanÚ _loadfieldsÚ_loadallZ_maxÚ _with_schemaÚ _verbatimÚ_cursorÚ_dialectÚ _add_scoresÚ_scorer)r r,r r r r \s zAggregateRequest.__init__)ÚfieldsrcGs|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)r/Úextendr0)r r7r r r Úloadus zAggregateRequest.load)r7ÚreducersrcGs”t|tƒr|gn|}t|tƒr$|gn|}dtt|ƒƒf|•}|D]B}|d|jtt|jƒƒg7}| |j¡|jdk r@|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. ZGROUPBYÚREDUCENÚAS) Ú isinstancer rÚlenr!rr8rr.)r r7r:ÚretZreducerr r r Úgroup_by…s   zAggregateRequest.group_byrcKs>| ¡D]0\}}d|g}|dk r,|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)")` ZAPPLYNr<)Úitemsr.r8)r ZkwexprrÚexprr?r r r Úapply s  zAggregateRequest.apply)rÚnumrcCst||ƒ}|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. )rr.r8r)r rrDZ_limitr r r Úlimit²s* zAggregateRequest.limitcOs–t|ttfƒr|g}g}|D]*}t|tƒr<||j|jg7}q||g7}qdtt|ƒƒg}| |¡| dd¡}|dkr†|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) ``` ZSORTBYÚmaxrÚMAX) r=r r#r$r%r>r8Úgetr.)r r7ÚkwargsZ fields_argsÚfr?rFr r r Úsort_byàs     zAggregateRequest.sort_by)Ú expressionsrcCs.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. ZFILTER)r=r r.r8)r rLZ expressionr r r Úfilters zAggregateRequest.filtercCs d|_|S)z| If set, the `schema` property will contain a list of `[field, type]` entries in the result object. T)r1rr r r Ú with_schema!szAggregateRequest.with_schemacCs d|_|S)zM If set, includes the score as an ordinary field of the row. T)r5rr r r Ú add_scores)szAggregateRequest.add_scores)ÚscorerrcCs ||_|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`) )r6)r rPr r r rP0szAggregateRequest.scorercCs d|_|S)NT)r2rr r r Úverbatim;szAggregateRequest.verbatimrç)rÚmax_idlercCs<dg}|r|dt|ƒg7}|r2|dt|dƒg7}||_|S)NZ WITHCURSORÚCOUNTÚMAXIDLEiè)r r3)r rrSrr r r Úcursor?szAggregateRequest.cursorcCsÐ|jg}|jr| d¡|jr(| d¡|jr>| d|jg¡|jrN| d¡|jr^||j7}|jrz| d¡| d¡n0|j rª| d¡| t t |j ƒƒ¡| |j ¡|j rÀ| d|j g¡| |j ¡|S)NZ WITHSCHEMAZVERBATIMZSCORERZ ADDSCORESZLOADr+ZDIALECT)r-r1Úappendr2r6r8r5r3r0r/r r>r4r.)r r?r r r rHs,         zAggregateRequest.build_args)ÚdialectrcCs ||_|S)zƒ Add a dialect field to the aggregate command. - **dialect** - dialect version to execute the query under )r4)r rXr r r rXkszAggregateRequest.dialect)r+)rrR)rrrr r r rr9rrr@rCrrErKrMrNrOrPrQÚfloatrVrrXr r r r r*Ws$þ ./  #r*c@s$eZdZeddœdd„Zdd„ZdS)ÚCursorN)ÚcidrcCs||_d|_d|_dS)Nr)r[rSr©r r[r r r r vszCursor.__init__cCs@t|jƒg}|jr$|dt|jƒg7}|jr<|dt|jƒg7}|S)NrUrT)r r[rSrrr r r r{s  zCursor.build_argsrr r r r rZusrZc@s.eZdZeddœdd„Zeefdœdd„ZdS)ÚAggregateResultN)rVrcCs||_||_||_dSr)ÚrowsrVÚschema)r r^rVr_r r r r …szAggregateResult.__init__rc Cs@|jr|jjnd}d|jj›dt|ƒd›dt|jƒ›d|›d S)Néÿÿÿÿú)rVr[Ú __class__rÚidr>r^r\r r r Ú__repr__Šs,ÿzAggregateResult.__repr__)rrrrZr r rfr r r r r]„sr]N) ÚtypingrrÚobjectrrrr#r&r)r*rZr]r r r r Ús *