Ѯh!dZddlmZddlmZmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZddlmZmZdd lmZerdd lmZdd lmZmZdd lmcmcmcmZ Gd deZ!y )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)ExplainMetricsExplainOptionsNceZdZdZ dfd Zdej jdfdd d dZdej jddf d dZ dej jdfdd d dZ xZ S) AsyncAggregationQueryz5Represents an aggregation query to the Firestore API.c,tt| |y)N)superr__init__)self nested_query __class__s }/home/www/academy-backend.kofcorporation.com/venv/lib/python3.12/site-packages/google/cloud/firestore_v1/async_aggregation.pyrzAsyncAggregationQuery.__init__-s #T3LAN)explain_optionsc:Kd}|j||||} |2cgc3d{}| 76ncc}w}}|d}n|jd{7}|jd{7n#|jd{7wxYwt|||Sw)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. 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. Returns: QueryResultsList[List[AggregationResult]]: The aggregation query results. Nr retrytimeoutr)streamget_explain_metricsacloser) rr r!r"rexplain_metrics stream_result aggregationresults rgetzAsyncAggregationQuery.get3s@26 #+ $  );HIIKkIIIFI&"&(5(I(I(K"K"K&&( ( (-&&( ( (IIsaBA01/- /1/1 A0AA0B)A,*B0B BB  BcFK|j||||\}}|jjjd||jjd|d{}|23d{}t |} | r| |j s(|j } | :7@776yw)a Internal 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. 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. )requestmetadataN) _prep_stream_client_firestore_apirun_aggregation_query _rpc_metadatarr&) rr r!r"rr,kwargsresponse_iteratorresponser)metricss r _make_streamz"AsyncAggregationQuery._make_streamgsL++      #T$,,"="="S"S# \\//# #   0  (.x8F ''"22   /s<AB!BB!!B%B&B)B! B!BB!cD|j||||}t||S)a 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. Returns: `AsyncStreamGenerator[List[AggregationResult]]`: A generator of the query results. r )r8r )rr r!r"rinner_generators rr#zAsyncAggregationQuery.streams5H++#+ ,  $O_EEr)returnNone)r!z'Union[retries.AsyncRetry, None, object]r"z float | NonerOptional[ExplainOptions]r;z)QueryResultsList[List[AggregationResult]]) r !Optional[transaction.Transaction]r!"retries.AsyncRetry | object | Noner"Optional[float]rr=r;zNAsyncGenerator[List[AggregationResult] | query_profile_pb.ExplainMetrics, Any]) r r>r!r?r"r@rr=r;z-AsyncStreamGenerator[List[AggregationResult]]) __name__ __module__ __qualname____doc__rr methodDEFAULTr*r8r# __classcell__)rs@rrr*s?B B9A9P9P $ 2J 59 2J72J 2J 2 2J 32Jl:>44rWsL #LL$21QELVLLcF0cFr