VSf&(dZddlmZddlZddlmZddlmZmZmZm Z m Z m Z m Z ddl mZddl mZddlmZdd lmZdd lmZdd lmZGd d eZGddeZGddeZGddeZGddeZ ddZGddeZy)aUClasses for representing aggregation queries for the Google Cloud Firestore API. A :class:`~google.cloud.firestore_v1.aggregation.AggregationQuery` can be created directly from a :class:`~google.cloud.firestore_v1.collection.Collection` and that can be a more common way to create an aggregation query than direct usage of the constructor. ) annotationsN)ABC)List CoroutineUnionTuple GeneratorAnyAsyncGenerator)gapic_v1)retry) FieldPath)RunAggregationQueryResponse)StructuredAggregationQuery)_helpersc eZdZdZdddZdZy)AggregationResulta A class representing result from Aggregation Query :type alias: str :param alias: The alias for the aggregation. :type value: int :param value: The resulting value from the aggregation. :type read_time: :param value: The resulting read_time Nc.||_||_||_yNaliasvalue read_time)selfrrrs iC:\Users\user\Documents\project_loop\venv\Lib\site-packages\google/cloud/firestore_v1/base_aggregation.py__init__zAggregationResult.__init__5s  "cVd|jd|jd|jdS)Nzrrs r__repr__zAggregationResult.__repr__:s.$TZZL KPTP^P^O__`aarr)rstrrint)__name__ __module__ __qualname____doc__rr!rrrr*s# brrc:eZdZdddZej dZy)BaseAggregationNc||_yrr)rrs rrzBaseAggregation.__init__?s  rcy)4Convert this instance to the protobuf representationNr(r s r _to_protobufzBaseAggregation._to_protobufBrrr str | None)r$r%r&rabcabstractmethodr/r(rrr*r*>s# CCrr*c(eZdZddfd ZdZxZS)CountAggregationc.tt| |yNr,)superr6r)rr __class__s rrzCountAggregation.__init__Hs .U.;rctj}|j|_tjj|_|Sr.)r AggregationrCountcountraggregation_pbs rr/zCountAggregation._to_protobufKs93??A#zz9EEKKMrrr1r$r%r&rr/ __classcell__r:s@rr6r6Gs <rr6c(eZdZddfd ZdZxZS)SumAggregationc|t|tr|j}||_tt ||yr8) isinstancer to_api_repr field_refr9rFrrrJrr:s rrzSumAggregation.__init__T4 i +!--/I" nd,5,9rctj}|j|_tjj|_|j |jj _|Sr<)rr=rSumsumrJfield field_pathr@s rr/zSumAggregation._to_protobuf[Q3??A#zz7CCGGI.2nn  +rrrJzstr | FieldPathrr2rBrDs@rrFrFS :rrFc(eZdZddfd ZdZxZS)AvgAggregationc|t|tr|j}||_tt ||yr8)rHrrIrJr9rVrrKs rrzAvgAggregation.__init__erLrctj}|j|_tjj|_|j |jj _|Sr<)rr=rAvgavgrJrPrQr@s rr/zAvgAggregation._to_protobuflrRrrrSrBrDs@rrVrVdrTrrVc>|jjjjDcgc]`}t ||jj|j xs#|jj|j |jb}}|Scc}w)Nr)resultaggregate_fieldspbkeysr integer_value double_valuer) response_pbkeyresultss r_query_response_to_resultreus%%6699>>@  $$55c:HHE!!2237DD!++  G NsA%Bc:eZdZdZdddZedZdddZdddZdddZ ddZ dd Z dd Z d dd Z ejdej"j$df dd Zejdej"j$df dd Zy)BaseAggregationQueryz5Represents an aggregation query to the Firestore API.NcP||_||_|j|_g|_yr) _nested_query_alias_parent_collection_ref _aggregations)r nested_queryrs rrzBaseAggregationQuery.__init__s') +3346rc.|jjSr)rl_clientr s rrpzBaseAggregationQuery._clients##+++rcTt|}|jj||S)z4 Adds a count over the nested query r,)r6rmappend)rrcount_aggregations rr?zBaseAggregationQuery.counts)-59 !!"34 rcVt||}|jj||S)z2 Adds a sum over the nested query r,)rFrmrr)rrJrsum_aggregations rrOzBaseAggregationQuery.sum))%@ !!/2 rcVt||}|jj||S)z3 Adds an avg over the nested query r,)rVrmrr)rrJravg_aggregations rrZzBaseAggregationQuery.avgrvrc:|jj|y)z Adds an aggregation operation to the nested query :type aggregation: :class:`google.cloud.firestore_v1.aggregation.BaseAggregation` :param aggregation: An aggregation operation, e.g. a CountAggregation N)rmrr)r aggregations radd_aggregationz$BaseAggregationQuery.add_aggregations !!+.rc:|jj|y)z Adds a list of aggregations to the nested query :type aggregations: list :param aggregations: a list of aggregation operations N)rmextend)r aggregationss radd_aggregationsz%BaseAggregationQuery.add_aggregationss !!,/rct}|jj|_|jD]-}|j}|j j |/|Sr)rrir/structured_queryrmr~rr)rr^rzrAs rr/z!BaseAggregationQuery._to_protobufs] ' )"00==?-- 3K(557N OO " "> 2 3 rc|jj\}}||jtj|d}tj ||}||fS)N)parentstructured_aggregation_query transaction)rl _parent_infor/rget_transaction_idmake_retry_timeout_kwargs)rrr timeout parent_pathexpected_prefixrequestkwargss r _prep_streamz!BaseAggregationQuery._prep_streams_ (,';';'H'H'J$ _!,0,=,=,?#66{C  33E7Crcy)aRuns the aggregation query. This sends a ``RunAggregationQuery`` RPC and returns a list of aggregation results in the stream of ``RunAggregationQueryResponse`` messages. Args: transaction (Optional[:class:`~google.cloud.firestore_v1.transaction.Transaction`]): An existing transaction that this query will run in. If a ``transaction`` is used and it already has write operations added, this method cannot be used (i.e. read-after-write is not allowed). retry (google.api_core.retry.Retry): Designation of what errors, if any, should be retried. Defaults to a system-specified policy. timeout (float): The timeout for this request. Defaults to a system-specified value. Returns: list: The aggregation query results Nr(rrr rs rgetzBaseAggregationQuery.getr0rcy)a}Runs the aggregation query. This sends a``RunAggregationQuery`` RPC and returns an iterator in the stream of ``RunAggregationQueryResponse`` messages. Args: transaction (Optional[:class:`~google.cloud.firestore_v1.transaction.Transaction`]): An existing transaction that this query will run in. If a ``transaction`` is used and it already has write operations added, this method cannot be used (i.e. read-after-write is not allowed). retry (google.api_core.retry.Retry): Designation of what errors, if any, should be retried. Defaults to a system-specified policy. timeout (float): The timeout for this request. Defaults to a system-specified value. Returns: list: The aggregation query results Nr(rs rstreamzBaseAggregationQuery.streamr0rr)rr2returnNoner1rS)rzr*rr)r~zList[BaseAggregation]rr)rr)NNN)r :Union[retries.Retry, None, gapic_v1.method._MethodDefault]r float | NonerzTuple[dict, dict])r rrrrzFList[AggregationResult] | Coroutine[Any, Any, List[AggregationResult]])r rrrrz]Generator[List[AggregationResult], Any, None] | AsyncGenerator[List[AggregationResult], None])r$r%r&r'rpropertyrpr?rOrZr{rr/rr3r4r methodDEFAULTrrr(rrrgrgs ?7 ,,/0LP $ J      OO # # $       P  :  OO # # $       8  rrg)rbrrzList[AggregationResult]) r' __future__rr3rtypingrrrrr r r google.api_corer r retries$google.cloud.firestore_v1.field_pathrgoogle.cloud.firestore_v1.typesrrgoogle.cloud.firestore_v1robjectrr*r6rFrVrergr(rrrs# PPP$,;GF.bb(CcC  _"_" ,   M 3M r