o }l~i'@sdZddlmZddlmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZddlmZmZdd lmZeradd lmZdd lmZmZdd lmmmmZ dd l!Z!Gd ddeZ"d S)aqClasses for representing Async aggregation queries for the Google Cloud Firestore API. A :class:`~google.cloud.firestore_v1.async_aggregation.AsyncAggregationQuery` can be created directly from a :class:`~google.cloud.firestore_v1.async_collection.AsyncCollection` and that can be a more common way to create an aggregation query than direct usage of the constructor. ) annotations) TYPE_CHECKINGAnyAsyncGeneratorListOptionalUnion)gapic_v1) retry_async) transaction)AsyncStreamGenerator)BaseAggregationQuery_query_response_to_result)QueryResultsList)AggregationResult)ExplainMetricsExplainOptionsNcsxeZdZdZdfdd ZdejjdfddddddZdejjdddfdddZ dejjdfdddd ddZ Z S)!AsyncAggregationQueryz5Represents an aggregation query to the Firestore API.returnNonecstt||dSN)superr__init__)self nested_query __class__r/var/www/newdalilibackend/backend/venv/lib/python3.10/site-packages/google/cloud/firestore_v1/async_aggregation.pyr.szAsyncAggregationQuery.__init__N)explain_options read_timeretry'Union[retries.AsyncRetry, None, object]timeout float | NonerOptional[ExplainOptions]r Optional[datetime.datetime])QueryResultsList[List[AggregationResult]]c szd}|j|||||d}z!dd|2IdH}|durd}n|IdH}W|IdHn|IdHwt|||S)aFRuns 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. explain_options (Optional[:class:`~google.cloud.firestore_v1.query_profile.ExplainOptions`]): Options to enable query profiling for this query. When set, explain_metrics will be available on the returned generator. read_time (Optional[datetime.datetime]): If set, reads documents as they were at the given time. This must be a timestamp within the past one hour, or if Point-in-Time Recovery is enabled, can additionally be a whole minute timestamp within the past 7 days. If no timezone is specified in the :class:`datetime.datetime` object, it is assumed to be UTC. Returns: QueryResultsList[List[AggregationResult]]: The aggregation query results. Nr r!r#rr csg|z3dHW}|q6Srr).0 aggregationrrr asz-AsyncAggregationQuery.get..)streamget_explain_metricsacloser) rr r!r#rr explain_metrics stream_resultresultrrrget4s"#  zAsyncAggregationQuery.getr !Optional[transaction.Transaction]"retries.AsyncRetry | object | NoneOptional[float]NAsyncGenerator[List[AggregationResult] | query_profile_pb.ExplainMetrics, Any]c Csx||||||\}}|jjjd||jjd|IdH}|2z3dHW} t| } | r/| V| jr8| j} | Vq 6dS)aInternal method for stream(). Runs the aggregation query. This sends a ``RunAggregationQuery`` RPC and then returns a generator 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 the query will run in. retry (Optional[google.api_core.retry.Retry]): Designation of what errors, if any, should be retried. Defaults to a system-specified policy. timeout (Optional[float]): The timeout for this request. Defaults to a system-specified value. explain_options (Optional[:class:`~google.cloud.firestore_v1.query_profile.ExplainOptions`]): Options to enable query profiling for this query. When set, explain_metrics will be available on the returned generator. read_time (Optional[datetime.datetime]): If set, reads documents as they were at the given time. This must be a timestamp within the past one hour, or if Point-in-Time Recovery is enabled, can additionally be a whole minute timestamp within the past 7 days. If no timezone is specified in the :class:`datetime.datetime` object, it is assumed to be UTC. Yields: List[AggregationResult] | query_profile_pb.ExplainMetrics: The result of aggregations of this query. Query results will be yielded as `List[AggregationResult]`. When the result contains returned explain metrics, yield `query_profile_pb.ExplainMetrics` individually. )requestmetadataNr) _prep_stream_client_firestore_apirun_aggregation_query _rpc_metadatarr/) rr r!r#rr r7kwargsresponse_iteratorresponser1metricsrrr _make_streamls.*  z"AsyncAggregationQuery._make_stream-AsyncStreamGenerator[List[AggregationResult]]cCs|j|||||d}t||S)aRuns the aggregation query. This sends a ``RunAggregationQuery`` RPC and then returns a generator 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 the query will run in. retry (Optional[google.api_core.retry.Retry]): Designation of what errors, if any, should be retried. Defaults to a system-specified policy. timeout (Optional[float]): The timeout for this request. Defaults to a system-specified value. explain_options (Optional[:class:`~google.cloud.firestore_v1.query_profile.ExplainOptions`]): Options to enable query profiling for this query. When set, explain_metrics will be available on the returned generator. read_time (Optional[datetime.datetime]): If set, reads documents as they were at the given time. This must be a timestamp within the past one hour, or if Point-in-Time Recovery is enabled, can additionally be a whole minute timestamp within the past 7 days. If no timezone is specified in the :class:`datetime.datetime` object, it is assumed to be UTC. Returns: `AsyncStreamGenerator[List[AggregationResult]]`: A generator of the query results. r()rBr )rr r!r#rr inner_generatorrrrr,s( zAsyncAggregationQuery.stream)rr) r!r"r#r$rr%r r&rr') r r3r!r4r#r5rr%r r&rr6) r r3r!r4r#r5rr%r r&rrC) __name__ __module__ __qualname____doc__rr methodDEFAULTr2rBr, __classcell__rrrrr+s.: Cr)#rH __future__rtypingrrrrrrgoogle.api_corer r retriesgoogle.cloud.firestore_v1r 0google.cloud.firestore_v1.async_stream_generatorr *google.cloud.firestore_v1.base_aggregationr r'google.cloud.firestore_v1.query_resultsrr'google.cloud.firestore_v1.query_profilerr-google.cloud.firestore_v1.types.query_profilecloud firestore_v1types query_profilequery_profile_pbdatetimerrrrrs