o }l~i@sdZddlmZddlmZmZmZddlmZ ddl m Z ddl m Z erJddlmZddlmZdd lmZdd lmZdd lmZdd lmZed e dZGdddeeZdS)z .. warning:: **Preview API**: Firestore Pipelines is currently in preview and is subject to potential breaking changes in future releases. ) annotations)GenericTypeVar TYPE_CHECKING)pipeline_stages) _BasePipeline)DOCUMENT_PATH_DELIMITER)Client) AsyncClient)BaseDocumentReference) BaseQuery)BaseAggregationQuery)BaseCollectionReference PipelineType)boundc@sTeZdZdZdddZddZdd d ZdddZd ddZd!ddZ d"ddZ dS)#PipelineSourceaz A factory for creating Pipeline instances, which provide a framework for building data transformation and query pipelines for Firestore. Not meant to be instantiated directly. Instead, start by calling client.pipeline() to obtain an instance of PipelineSource. From there, you can use the provided methods to specify the data source for your pipeline. clientClient | AsyncClientcCs ||_dSN)r)selfrrp/var/www/newdalilibackend/backend/venv/lib/python3.10/site-packages/google/cloud/firestore_v1/pipeline_source.py__init__0s zPipelineSource.__init__cCs|jj|j|Sr)r _pipeline_cls_create_with_stages)r source_stagerrr_create_pipeline3szPipelineSource._create_pipelinequery@'BaseQuery' | 'BaseAggregationQuery' | 'BaseCollectionReference'returnrcCs ||S)a Create a pipeline from an existing query Queries containing a `cursor` or `limit_to_last` are not currently supported Args: query: the query to build the pipeline off of Raises: - NotImplementedError: raised if the query contains a `cursor` or `limit_to_last` Returns: a new pipeline instance representing the query )_build_pipeline)rrrrr create_from6s zPipelineSource.create_frompathstr | tuple[str]cCs$t|tr t|}|t|S)a Creates a new Pipeline that operates on a specified Firestore collection. Args: path: The path to the Firestore collection (e.g., "users"). Can either be: * A single ``/``-delimited path to a collection * A tuple of collection path segment Returns: a new pipeline instance targeting the specified collection ) isinstancetuplerjoinrstages Collection)rr"rrr collectionGs zPipelineSource.collection collection_idstrcCs|t|S)a  Creates a new Pipeline that that operates on all documents in a collection group. Args: collection_id: The ID of the collection group Returns: a new pipeline instance targeting the specified collection group )rr'CollectionGroup)rr*rrrcollection_groupVszPipelineSource.collection_groupcCs|tS)z Creates a new Pipeline that operates on all documents in the Firestore database. Returns: a new pipeline instance targeting the specified collection )rr'Database)rrrrdatabase`szPipelineSource.databasedocs'BaseDocumentReference'cGs|tjj|S)a. Creates a new Pipeline that operates on a specific set of Firestore documents. Args: docs: The DocumentReference instances representing the documents to include in the pipeline. Returns: a new pipeline instance targeting the specified documents )rr' Documentsof)rr0rrr documentshszPipelineSource.documentsN)rr)rrrr)r"r#rr)r*r+rr)rr)r0r1rr) __name__ __module__ __qualname____doc__rrr!r)r-r/r4rrrrr&s     rN)r8 __future__rtypingrrrgoogle.cloud.firestore_v1rr''google.cloud.firestore_v1.base_pipeliner"google.cloud.firestore_v1._helpersr google.cloud.firestore_v1.clientr &google.cloud.firestore_v1.async_clientr 'google.cloud.firestore_v1.base_documentr $google.cloud.firestore_v1.base_queryr *google.cloud.firestore_v1.base_aggregationr )google.cloud.firestore_v1.base_collectionrrrrrrrs