%9j4dZddlmZddlmZmZmZmZmZm Z m Z ddl m Z ddl m ZddlmZmZmZmZddlmZddlmZmZdd lmZdd lmZerdd lZdd lmZdd l m!Z!ddl"m#Z#GddeejHZ%y )zHClasses for representing collections for the Google Cloud Firestore API.) annotations) TYPE_CHECKINGAnyCallable GeneratorOptionalTupleUnion)gapic_v1)retry) aggregationdocument transaction vector_query)query)BaseCollectionReference_item_to_document_ref)QueryResultsList)WatchN)DocumentSnapshot)ExplainOptions)StreamGeneratorceZdZdZdfd ZddZddZddZdejjdf ddZ dejjdfdd dd Z dd Z dejjdfddd  dd Zdejjdfddd  dd ZddZxZS)CollectionReferencea A reference to a collection in a Firestore database. The collection may already exist or this class can facilitate creation of documents within the collection. Args: path (Tuple[str, ...]): The components in the collection path. This is a series of strings representing each collection and sub-collection ID, as well as the document IDs for any documents that contain a sub-collection. kwargs (dict): The keyword arguments for the constructor. The only supported keyword is ``client`` and it must be a :class:`~google.cloud.firestore_v1.client.Client` if provided. It represents the client that created this collection reference. Raises: ValueError: if * the ``path`` is empty * there are an even number of elements * a collection ID in ``path`` is not a string * a document ID in ``path`` is not a string TypeError: If a keyword other than ``client`` is used. c,tt| |i|yN)superr__init__)selfpathkwargs __class__s `/var/www/html/truck-me/venv/lib/python3.12/site-packages/google/cloud/firestore_v1/collection.pyrzCollectionReference.__init__Cs !414B6Bc,tj|S)zeQuery factory. Returns: :class:`~google.cloud.firestore_v1.query.Query` ) query_modQueryrs r#_queryzCollectionReference._queryFs t$$r$cHtj|jS)zAggregationQuery factory. Returns: :class:`~google.cloud.firestore_v1.aggregation_query.AggregationQuery` )r AggregationQueryr)r(s r#_aggregation_queryz&CollectionReference._aggregation_queryNs ++DKKM::r$cHtj|jS)zxVectorQuery factory. Returns: :class:`~google.cloud.firestore_v1.vector_query.VectorQuery` )r VectorQueryr)r(s r# _vector_queryz!CollectionReference._vector_queryVs '' 66r$Ncr|j||||\}}|j|fi|}|j|fS)aCreate a document in the Firestore database with the provided data. Args: document_data (dict): Property names and values to use for creating the document. document_id (Optional[str]): The document identifier within the current collection. If not provided, an ID will be automatically assigned by the server (the assigned ID will be a random 20 character string composed of digits, uppercase and lowercase letters). 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. Returns: Tuple[:class:`google.protobuf.timestamp_pb2.Timestamp`, :class:`~google.cloud.firestore_v1.document.DocumentReference`]: Pair of * The ``update_time`` when the document was created/overwritten. * A document reference for the created document. Raises: :class:`google.cloud.exceptions.Conflict`: If ``document_id`` is provided and the document already exists. ) _prep_addcreate update_time)r document_data document_idr timeout document_refr! write_results r#addzCollectionReference.add^sND $~~       f +|**=CFC ''55r$) read_timecj||||\}}jjjd|jjd|}fd|DS)aList all subdocuments of the current collection. Args: page_size (Optional[int]]): The maximum number of documents in each page of results from this request. Non-positive values are ignored. Defaults to a sensible value set by the API. 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. 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: Sequence[:class:`~google.cloud.firestore_v1.collection.DocumentReference`]: iterator of subdocuments of the current collection. If the collection does not exist at the time of `snapshot`, the iterator will be empty )requestmetadatac36K|]}t|ywr)r).0irs r# z5CollectionReference.list_documents..sA1%dA.As)_prep_list_documents_client_firestore_apilist_documents _rpc_metadata)r page_sizer r6r:r<r!iterators` r#rFz"CollectionReference.list_documentssm:33 ugy >4<<..== \\//   BAAr$c@|jj|Sr)r) _chunkify)r chunk_sizes r#rKzCollectionReference._chunkifys{{}&&z22r$)explain_optionsr:cp|j||\}}|||d<|||d<|jdd|i|S)asRead the documents in this collection. This sends a ``RunQuery`` RPC and returns a list of documents returned in the stream of ``RunQueryResponse`` messages. Args: transaction (Optional[:class:`~google.cloud.firestore_v1.transaction.transaction.Transaction`]): An existing transaction that this query will run in. 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. 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). Returns: QueryResultsList[DocumentSnapshot]: The documents in this collection that match the query. rMr:rrB)_prep_get_or_streamgetrrr r6rMr:rr!s r#rPzCollectionReference.getsUN00@ v  &(7F$ %  "+F; uyy;[;F;;r$cp|j||\}}|r||d<|||d<|jdd|i|S)aRead the documents in this collection. This sends a ``RunQuery`` RPC and then returns an iterator which consumes each document returned in the stream of ``RunQueryResponse`` messages. .. note:: The underlying stream of responses will time out after the ``max_rpc_timeout_millis`` value set in the GAPIC client configuration for the ``RunQuery`` API. Snapshots not consumed from the iterator before that point will be lost. 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.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: `StreamGenerator[DocumentSnapshot]`: A generator of the query results. rMr:rrB)rOstreamrQs r#rSzCollectionReference.streamsR^00@ v (7F$ %  "+F; u||> >v>>r$cl|j}tj||tjS)aMonitor the documents in this collection. This starts a watch on this collection using a background thread. The provided callback is run on the snapshot of the documents. Args: callback (Callable[[:class:`~google.cloud.firestore.collection.CollectionSnapshot`], NoneType]): a callback to run when a change occurs. Example: from google.cloud import firestore_v1 db = firestore_v1.Client() collection_ref = db.collection(u'users') def on_snapshot(collection_snapshot, changes, read_time): for doc in collection_snapshot.documents: print(u'{} => {}'.format(doc.id, doc.to_dict())) # Watch this collection collection_watch = collection_ref.on_snapshot(on_snapshot) # Terminate this watch collection_watch.unsubscribe() )r)r for_queryrr)rcallbackrs r# on_snapshotzCollectionReference.on_snapshots(4 uh0I0IJJr$)returnNone)rXzquery_mod.Query)rXzaggregation.AggregationQuery)rXzvector_query.VectorQuery) r4dictr5zUnion[str, None]r retries.Retry | object | Noner6Union[float, None]rXzTuple[Any, Any]) rHzUnion[int, None]r r[r6r\r:Optional[datetime.datetime]rXzGenerator[Any, Any, None])rLint) rz$Union[transaction.Transaction, None]r r[r6r\rMOptional[ExplainOptions]r:r]rXz"QueryResultsList[DocumentSnapshot]) rz!Optional[transaction.Transaction]r r[r6zOptional[float]rMr_r:r]rXz!StreamGenerator[DocumentSnapshot])rVrrXr)__name__ __module__ __qualname____doc__rr)r,r/r methodDEFAULTr9rFrKrPrSrW __classcell__)r"s@r#rr)s2C%;7)-/7/F/F&* )6)6&)6- )6 $ )6  )6Z'+/7/F/F&* &B 26 &B#&B-&B$ &B / &B #&BP3 =A/7/F/F&* -< 5915-<9-<--<$ -< 2 -</-< ,-/7/F/F#' 5? 59155?65?-5?! 5? 2 5?/5? +5?nKr$r)&rc __future__rtypingrrrrrr r google.api_corer r retriesgoogle.cloud.firestore_v1r rrrrr&)google.cloud.firestore_v1.base_collectionrr'google.cloud.firestore_v1.query_resultsrgoogle.cloud.firestore_v1.watchrdatetime'google.cloud.firestore_v1.base_documentr'google.cloud.firestore_v1.query_profiler*google.cloud.firestore_v1.stream_generatorrr'rrBr$r#rss]O"RRR$,VV8E1HFJLK1)//BLKr$