a !f@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) GeneratorUnionListAnycsreZdZdZddfdd Zdejjdfddd d d d Zd dZ ddZ dejjdfdddd ddZ Z S)AggregationQueryz5Represents an aggregation query to the Firestore API.None)returncstt||dS)N)superr __init__)selfZ nested_query __class__f/var/www/html/python-backend/venv/lib/python3.9/site-packages/google/cloud/firestore_v1/aggregation.pyr(szAggregationQuery.__init__Nz:Union[retries.Retry, None, gapic_v1.method._MethodDefault]z float | NonezList[AggregationResult])rtimeoutrcCs|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||||\}}|jjjf||jjd|S)!Helper method for :meth:`stream`.)requestmetadata)Z _prep_stream_client_firestore_apirun_aggregation_queryZ _rpc_metadata)rrrrrkwargsrrr_get_stream_iteratorMs z%AggregationQuery._get_stream_iteratorcCs8|dur4|tjjur*|jjj}|j}|j}||SdS)rNF) rmethodDEFAULTr r! _transportr"Z_retryZ _predicate)rexcrr transportZgapic_callablerrr_retry_query_after_exception[s   z-AggregationQuery._retry_query_after_exceptionz4Union[Generator[List[AggregationResult], Any, None]]c cs||||}zt|d}WnRtjyn}z8||||rX||||}WYd}~qnWYd}~n d}~00|durzqt|}|VqdS)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$nextrZGoogleAPICallErrorr*r)rrrrZresponse_iteratorresponser(rrrrrfs( zAggregationQuery.stream) __name__ __module__ __qualname____doc__rrr%r&rr$r*r __classcell__rrrrr %s r N)r0 __future__rZgoogle.api_corerrrretriesZ*google.cloud.firestore_v1.base_aggregationrrrtypingr r r r r rrrrs