o }l~i#@sdZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddl m Z erPdd lZdd lmZdd lmZdd lmZdd lmZddlmZGddde Zd S)z .. warning:: **Preview API**: Firestore Pipelines is currently in preview and is subject to potential breaking changes in future releases ) annotations) TYPE_CHECKING)pipeline_stages) _BasePipeline)AsyncPipelineStream)PipelineSnapshot)PipelineResultN) AsyncClient)AsyncTransaction)Constant)Value)PipelineExplainOptionscsNeZdZdZdfdd Zdddid dddZdddiddddZZS) AsyncPipelinea  Pipelines allow for complex data transformations and queries involving multiple stages like filtering, projection, aggregation, and vector search. This class extends `_BasePipeline` and provides methods to execute the defined pipeline stages using an asynchronous `AsyncClient`. Usage Example: >>> from google.cloud.firestore_v1.pipeline_expressions import Field >>> >>> async def run_pipeline(): ... client = AsyncClient(...) ... pipeline = client.pipeline() ... .collection("books") ... .where(Field.of("published").gt(1980)) ... .select("title", "author") ... async for result in pipeline.stream(): ... print(result) Use `client.pipeline()` to create instances of this class. .. warning:: **Preview API**: Firestore Pipelines is currently in preview and is subject to potential breaking changes in future releases clientr stages stages.Stagecstj|g|RdS)z Initializes an asynchronous Pipeline. Args: client: The asynchronous `AsyncClient` instance to use for execution. *stages: Initial stages for the pipeline. N)super__init__)selfrr __class__o/var/www/newdalilibackend/backend/venv/lib/python3.10/site-packages/google/cloud/firestore_v1/async_pipeline.pyr@szAsyncPipeline.__init__N) transaction read_timeexplain_optionsadditional_optionsr'AsyncTransaction' | Nonerdatetime.datetime | NonerPipelineExplainOptions | Nonerdict[str, Value | Constant]return PipelineSnapshot[PipelineResult]csFddtD}tt|fi|}dd|2IdH}t||S)a( Executes this pipeline and returns results as a list 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). read_time (Optional[datetime.datetime]): If set, reads documents as they were at the given time. This must be a microsecond precision 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. For the most accurate results, use UTC timezone. explain_options (Optional[:class:`~google.cloud.firestore_v1.query_profile.PipelineExplainOptions`]): Options to enable query profiling for this query. When set, explain_metrics will be available on the returned list. additional_options (Optional[dict[str, Value | Constant]]): Additional options to pass to the query. These options will take precedence over method argument if there is a conflict (e.g. explain_options) cSi|] \}}|dkr||qSrr.0kvrrr ez)AsyncPipeline.execute..csg|z3dHW}|q6S)Nr)r&resultrrr gsz)AsyncPipeline.execute..N)localsitemsrrr)rrrrrkwargsstreamresultsrrrexecuteJs  zAsyncPipeline.execute)rrrr#AsyncPipelineStream[PipelineResult]cCs&ddtD}tt|fi|S)aF Process this pipeline as a stream, providing results through an AsyncIterable 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). read_time (Optional[datetime.datetime]): If set, reads documents as they were at the given time. This must be a microsecond precision 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. For the most accurate results, use UTC timezone. explain_options (Optional[:class:`~google.cloud.firestore_v1.query_profile.PipelineExplainOptions`]): Options to enable query profiling for this query. When set, explain_metrics will be available on the returned generator. additional_options (Optional[dict[str, Value | Constant]]): Additional options to pass to the query. These options will take precedence over method argument if there is a conflict (e.g. explain_options) cSr#r$rr%rrrr)r*z(AsyncPipeline.stream..)r-r.rr)rrrrrr/rrrr0jszAsyncPipeline.stream)rr rr) rrrrrrrr r!r") rrrrrrrr r!r3)__name__ __module__ __qualname____doc__rr2r0 __classcell__rrrrr%s #r)r7 __future__rtypingrgoogle.cloud.firestore_v1rr'google.cloud.firestore_v1.base_pipeliner)google.cloud.firestore_v1.pipeline_resultrrrdatetime&google.cloud.firestore_v1.async_clientr +google.cloud.firestore_v1.async_transactionr .google.cloud.firestore_v1.pipeline_expressionsr (google.cloud.firestore_v1.types.documentr 'google.cloud.firestore_v1.query_profiler rrrrrs