VSfzdZddlmZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZmZGdd e Zy ) 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. ) annotations) exceptions)gapic_v1)retry)AggregationResultBaseAggregationQuery_query_response_to_result) GeneratorUnionListAnyceZdZdZ dfd Zdej jdf d dZdZ dZ dej jdf d dZ xZ S) AggregationQueryz5Represents an aggregation query to the Firestore API.c,tt| |y)N)superr__init__)self nested_query __class__s dC:\Users\user\Documents\project_loop\venv\Lib\site-packages\google/cloud/firestore_v1/aggregation.pyrzAggregationQuery.__init__(s .|<Nc@|j|||}t|S)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 ) transactionrtimeout)streamlist)rrrrresults rgetzAggregationQuery.get.s"8E7SF|rc|j|||\}}|jjjd||jjd|S)!Helper method for :meth:`stream`.)requestmetadata) _prep_stream_client_firestore_apirun_aggregation_query _rpc_metadata)rrrrr!kwargss r_get_stream_iteratorz%AggregationQuery._get_stream_iteratorMs`++     At||**@@ \\//   rc|e|tjjur8|jjj }|j }|j}|j|Sy)r F) rmethodDEFAULTr%r& _transportr'_retry _predicate)rexcrr transportgapic_callables r_retry_query_after_exceptionz-AggregationQuery._retry_query_after_exception[sY  /// LL77BB !*!@!@&--##C( (rc#K|j|||} t|d}|yt |}| #tj$r1}|j |||r|j|||}Yd}~_d}~wwxYww)aRuns the aggregation query. This sends a ``RunAggregationQuery`` RPC and then returns an iterator which consumes each document returned in the stream of ``RunAggregationQueryResponse`` messages. 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). Args: transaction (Optional[:class:`~google.cloud.firestore_v1.transaction.Transaction`]): An existing transaction that this query will run in. 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. Yields: :class:`~google.cloud.firestore_v1.base_aggregation.AggregationResult`: The result of aggregations of this query N)r*nextrGoogleAPICallErrorr4r )rrrrresponse_iteratorresponser1rs rrzAggregationQuery.streamfs@!55       148.x8FL#00 44S%M(,(A(A#)%  s1A> 7A>A; &A60A>5A66A;;A>)returnNone)r:Union[retries.Retry, None, gapic_v1.method._MethodDefault]r float | Noner:zList[AggregationResult])rr<rr=r:z4Union[Generator[List[AggregationResult], Any, None]]) __name__ __module__ __qualname____doc__rrr,r-rr*r4r __classcell__)rs@rrr%s?= = OO # # $     !>    OO # # $ 6 6  6 >6rrN)rA __future__rgoogle.api_corerrrretries*google.cloud.firestore_v1.base_aggregationrrr typingr r r r rr#rrrHs9 #&$, /.w+wr