o }l~i):@sdZddlmZddlZddlZddlmZddlmZmZm Z m Z m Z m Z m Z mZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZerddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)Z)Gddde*Z+GdddeZ,Gddde,Z-Gddde,Z.Gddde,Z/d%d!d"Z0Gd#d$d$eZ1dS)&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. ) annotationsN)ABC) TYPE_CHECKINGAny CoroutineListOptionalTupleUnionIterable)gapic_v1)retry)_helpers) FieldPath)StructuredAggregationQuery)AggregateFunction)Count)AliasedExpression)Field) transaction)AsyncStreamGenerator)ExplainOptions)QueryResultsList)StreamGenerator)PipelineSourcec@s,eZdZdZd dddZd d Zd d ZdS)AggregationResulta A class representing result from Aggregation Query :type alias: str :param alias: The alias for the aggregation. :type value: int :param value: The resulting value from the aggregation. :type read_time: :param value: The resulting read_time NaliasstrvaluefloatcCs||_||_||_dSNrr read_time)selfrrr"r$q/var/www/newdalilibackend/backend/venv/lib/python3.10/site-packages/google/cloud/firestore_v1/base_aggregation.py__init__Cs zAggregationResult.__init__cCsd|jd|jd|jdS)Nzr!r#r$r$r%__repr__HszAggregationResult.__repr__cCs |j|jiSr )rrr(r$r$r%_to_dictKs zAggregationResult._to_dictr )rrrr)__name__ __module__ __qualname____doc__r&r)r*r$r$r$r%r8s   rc@s>eZdZddddZejddZejdd d ZddZdS)BaseAggregationNr str | NonecCs ||_dSr rr#rr$r$r%r&Ps zBaseAggregation.__init__cCdS)4Convert this instance to the protobuf representationNr$r(r$r$r% _to_protobufSzBaseAggregation._to_protobuf autoindexer Iterable[int]return$AliasedExpression[AggregateFunction]cCr3)a6 Convert this instance to a pipeline expression for use with pipeline.aggregate() Args: autoindexer: If an alias isn't supplied, one should be created with the format "field_n" The autoindexer is an iterable that provides the `n` value to use for each expression Nr$r#r7r$r$r%_to_pipeline_exprWr6z!BaseAggregation._to_pipeline_exprcCs|jdur|jSdt|S)zG Helper to build the alias for the pipeline expression Nfield_)rnextr;r$r$r%_pipeline_aliascs zBaseAggregation._pipeline_aliasr rr0)r7r8r9r:) r+r,r-r&abcabstractmethodr5r<r?r$r$r$r%r/Os    r/cs2eZdZd d fdd ZddZdd d ZZS)CountAggregationNrr0cstt|j|ddSNr1)superrCr&r2 __class__r$r%r&nszCountAggregation.__init__cCs&t}|jr |j|_tj|_|Sr4)r Aggregationrrcountr#aggregation_pbr$r$r%r5qs  zCountAggregation._to_protobufr7r8cCst||Sr )ras_r?r;r$r$r%r<ysz"CountAggregation._to_pipeline_exprr r@r7r8r+r,r-r&r5r< __classcell__r$r$rFr%rCmsrCc2eZdZddfdd Zdd Zdd d ZZS)SumAggregationN field_refstr | FieldPathrr0c2t|tr |n|}||_tt|j|ddSrD) isinstancer to_api_reprrSrErRr&r#rSr field_strrFr$r%r&~zSumAggregation.__init__cC2t}|jr |j|_tj|_|j|jj_|SrH)rrIrSumsumrSfield field_pathrKr$r$r%r5   zSumAggregation._to_protobufr7r8cCt|j||Sr )rofrSr]rMr?r;r$r$r%r<z SumAggregation._to_pipeline_exprr rSrTrr0rNrOr$r$rFr%rR} rRcrQ)AvgAggregationNrSrTrr0crUrD)rVrrWrSrErfr&rXrFr$r%r&rZzAvgAggregation.__init__cCr[rH)rrIrAvgavgrSr^r_rKr$r$r%r5r`zAvgAggregation._to_protobufr7r8cCrar )rrbrSaveragerMr?r;r$r$r%r<rcz AvgAggregation._to_pipeline_exprr rdrNrOr$r$rFr%rfrerfr9List[AggregationResult]cs fddjjjD}|S)Ncs4g|]}t|jj|jpjj|jjdqS)r!)rresultaggregate_fields integer_value double_valuer").0key response_pbr$r% s z-_query_response_to_result..)rkrlpbkeys)rrresultsr$rqr%_query_response_to_results   rwc@seZdZdZd8d9ddZed d Zd8d:d d Zd8d;ddZd8d;ddZ dddZ     d?d@d'd(Z ejdejjdfddd)dAd+d,Zejdejjdfddd)dBd2d3ZdCd6d7ZdS)DBaseAggregationQueryz5Represents an aggregation query to the Firestore API.Nrr0r9NonecCs||_||_|j|_g|_dSr ) _nested_query_alias_parent_collection_ref _aggregations)r# nested_queryrr$r$r%r&s zBaseAggregationQuery.__init__cCs|jjSr )r}_clientr(r$r$r%rszBaseAggregationQuery._clientcCst|d}|j||S)z4 Adds a count over the nested query r1)rCr~append)r#rcount_aggregationr$r$r%rJs  zBaseAggregationQuery.countrSrTcCt||d}|j||S)z2 Adds a sum over the nested query r1)rRr~r)r#rSrsum_aggregationr$r$r%r]  zBaseAggregationQuery.sumcCr)z3 Adds an avg over the nested query r1)rfr~r)r#rSravg_aggregationr$r$r%rhrzBaseAggregationQuery.avg aggregationr/cC|j|dS)z Adds an aggregation operation to the nested query :type aggregation: :class:`google.cloud.firestore_v1.aggregation.BaseAggregation` :param aggregation: An aggregation operation, e.g. a CountAggregation N)r~r)r#rr$r$r%add_aggregationz$BaseAggregationQuery.add_aggregation aggregationsList[BaseAggregation]cCr)z Adds a list of aggregations to the nested query :type aggregations: list :param aggregations: a list of aggregation operations N)r~extend)r#rr$r$r%add_aggregationsrz%BaseAggregationQuery.add_aggregationsrcCs6t}|j|_|jD] }|}|j|q |Sr )rrzr5structured_queryr~rr)r#rtrrLr$r$r%r5s   z!BaseAggregationQuery._to_protobufr 6Union[retries.Retry, retries.AsyncRetry, None, object]timeout float | Noneexplain_optionsOptional[ExplainOptions]r"Optional[datetime.datetime]Tuple[dict, dict]c CsX|j\}}||t|d}|r||d<|dur"||d<t||} || fS)N)parentstructured_aggregation_queryrrr")r} _parent_infor5rget_transaction_idr*make_retry_timeout_kwargs) r#rr rrr" parent_pathexpected_prefixrequestkwargsr$r$r% _prep_streams  z!BaseAggregationQuery._prep_stream)rr"XQueryResultsList[AggregationResult] | Coroutine[Any, Any, List[List[AggregationResult]]]cCr3)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. 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]] | Coroutine[Any, Any, List[List[AggregationResult]]]): The aggregation query results. Nr$r#rr rrr"r$r$r%getr6zBaseAggregationQuery.getr!Optional[transaction.Transaction]2retries.Retry | retries.AsyncRetry | object | NoneOptional[float]XStreamGenerator[List[AggregationResult]] | AsyncStreamGenerator[List[AggregationResult]]cCr3)aRuns the aggregation query. This sends a``RunAggregationQuery`` RPC and returns a generator 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. retry (Optional[google.api_core.retry.Retry]): Designation of what errors, if any, should be retried. Defaults to a system-specified policy. timeout (Optinal[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: StreamGenerator[List[AggregationResult]] | AsyncStreamGenerator[List[AggregationResult]]: A generator of the query results. Nr$rr$r$r%stream=r6zBaseAggregationQuery.streamsource'PipelineSource'cs2tjddfdd|jD}|j|j|S)a Convert this query into a Pipeline Queries containing a `cursor` or `limit_to_last` are not currently supported Args: source: the PipelineSource to build the pipeline off of Raises: - NotImplementedError: raised if the query contains a `cursor` or `limit_to_last` Returns: a Pipeline representing the query )startcsg|]}|qSr$)r<)roar7r$r%rswsz8BaseAggregationQuery._build_pipeline..) itertoolsrJr~rz_build_pipeline aggregate)r#rexprsr$rr%rhs z$BaseAggregationQuery._build_pipeliner )rr0r9ryr@rd)rr/r9ry)rrr9ry)r9r)NNNNN) r rrrrrr"rr9r) r rrrrrr"rr9r) rrr rrrrrr"rr9r)rr)r+r,r-r.r&propertyrrJr]rhrrr5rrArBr methodDEFAULTrrrr$r$r$r%rxsD         - *rx)r9rj)2r. __future__rrArrtypingrrrrrr r r google.api_corer r retriesgoogle.cloud.firestore_v1r$google.cloud.firestore_v1.field_pathrgoogle.cloud.firestore_v1.typesr.google.cloud.firestore_v1.pipeline_expressionsrrrrr0google.cloud.firestore_v1.async_stream_generatorr'google.cloud.firestore_v1.query_profiler'google.cloud.firestore_v1.query_resultsr*google.cloud.firestore_v1.stream_generatorr)google.cloud.firestore_v1.pipeline_sourcerdatetimeobjectrr/rCrRrfrwrxr$r$r$r%s<  (