o Fh@stdZddlmZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZmZGdd d e Zd S) 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) GeneratorUnionListAnycs^eZdZdZdfdd Zdejjdfdd d ZddZ ddZ dejjdfdddZ Z S)AggregationQueryz5Represents an aggregation query to the Firestore API.returnNonecstt||dS)N)superr __init__)self nested_query __class__^/var/www/html/loop/nvenv/lib/python3.10/site-packages/google/cloud/firestore_v1/aggregation.pyr(szAggregationQuery.__init__Nr:Union[retries.Retry, None, gapic_v1.method._MethodDefault]timeout float | NoneList[AggregationResult]cCs|j|||d}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 ) transactionrr)streamlist)rrrrresultrrrget.szAggregationQuery.getcCs0||||\}}|jjjd||jjd|S)!Helper method for :meth:`stream`.)requestmetadataNr) _prep_stream_client_firestore_apirun_aggregation_query _rpc_metadata)rrrrr"kwargsrrr_get_stream_iteratorMs z%AggregationQuery._get_stream_iteratorcCs8|dur|tjjur|jjj}|j}|j}||SdS)r!NF) rmethodDEFAULTr%r& _transportr'_retry _predicate)rexcrr transportgapic_callablerrr_retry_query_after_exception[s   z-AggregationQuery._retry_query_after_exception4Union[Generator[List[AggregationResult], Any, None]]c cs||||} zt|d}Wn"tjy2}z||||r-||||}WYd}~qd}~ww|dur9dSt|}|Vq )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 TN)r*nextrGoogleAPICallErrorr3r)rrrrresponse_iteratorresponser0rrrrrfs2   zAggregationQuery.stream)rr)rrrrrr)rrrrrr4) __name__ __module__ __qualname____doc__rrr+r,r r*r3r __classcell__rrrrr %s  r N)r< __future__rgoogle.api_corerrrretries*google.cloud.firestore_v1.base_aggregationrrrtypingr r r r r rrrrs