o }l~ie@sdZddlmZddlZddlmZmZmZddlm Z ddl m Z ddl mZddlmZGd d d ejd ZGd d d eZdS)z=Helpers for batch requests to the Google Cloud Firestore API.) annotationsN)AnyDictUnion)retry)_helpers)BaseDocumentReference)writec@sxeZdZdZd$ddZddZd%d d Zd&ddZej ddZ d'ddZ d(d)ddZ d*d+d d!Z d*d,d"d#ZdS)- BaseBatchaAccumulate write operations to be sent in a batch. This has the same set of methods for write operations that :class:`~google.cloud.firestore_v1.document.DocumentReference` does, e.g. :meth:`~google.cloud.firestore_v1.document.DocumentReference.create`. Args: client (:class:`~google.cloud.firestore_v1.client.Client`): The client that created this batch. returnNonecCs"||_g|_i|_d|_d|_dSN)_client _write_pbs_document_references write_results commit_time)selfclientrk/var/www/newdalilibackend/backend/venv/lib/python3.10/site-packages/google/cloud/firestore_v1/base_batch.py__init__(s  zBaseBatch.__init__cCs t|jSr )lenrrrrr__len__/s zBaseBatch.__len__ referencercCs |j|jvSr )_document_pathr)rrrrr __contains__2s zBaseBatch.__contains__ write_pbslist[write_pb.Write]cCs|j|dS)aAdd `Write`` protobufs to this transaction. This method intended to be over-ridden by subclasses. Args: write_pbs (List[google.cloud.firestore_v1. write_pb2.Write]): A list of write protobufs to be added. N)rextend)rrrrr_add_write_pbs5s zBaseBatch._add_write_pbscCst)zySends all accumulated write operations to the server. The details of this write depend on the implementing class.)NotImplementedErrorrrrrcommit@szBaseBatch.commit document_datadict[str, Any]cCs(t|j|}||j|j<||dS)aAdd a "change" to this batch to create a document. If the document given by ``reference`` already exists, then this batch will fail when :meth:`commit`-ed. Args: reference (:class:`~google.cloud.firestore_v1.document.DocumentReference`): A document reference to be created in this batch. document_data (dict): Property names and values to use for creating a document. N)rpbs_for_createrrr!)rrr$rrrrcreateFs zBaseBatch.createFdictmergeUnion[bool, list]cCsB|dur t|j||}nt|j|}||j|j<||dS)aAdd a "change" to replace a document. See :meth:`google.cloud.firestore_v1.document.DocumentReference.set` for more information on how ``option`` determines how the change is applied. Args: reference (:class:`~google.cloud.firestore_v1.document.DocumentReference`): A document reference that will have values set in this batch. document_data (dict): Property names and values to use for replacing a document. merge (Optional[bool] or Optional[List]): If True, apply merging instead of overwriting the state of the document. FN)rpbs_for_set_with_mergerpbs_for_set_no_mergerr!)rrr$r)rrrrsetXs z BaseBatch.setN field_updatesoption_helpers.WriteOption | NonecCs>|jjdkr tdt|j||}||j|j<||dS)aAdd a "change" to update a document. See :meth:`google.cloud.firestore_v1.document.DocumentReference.update` for more information on ``field_updates`` and ``option``. Args: reference (:class:`~google.cloud.firestore_v1.document.DocumentReference`): A document reference that will be updated in this batch. field_updates (dict): Field names or paths to update and values to update with. option (Optional[:class:`~google.cloud.firestore_v1.client.WriteOption`]): A write option to make assertions / preconditions on the server state of the document before applying changes. ExistsOptionz5you must not pass an explicit write option to update.N) __class____name__ ValueErrorrpbs_for_updaterrr!)rrr.r/rrrrupdatezs  zBaseBatch.updatecCs*t|j|}||j|j<||gdS)ayAdd a "change" to delete a document. See :meth:`google.cloud.firestore_v1.document.DocumentReference.delete` for more information on how ``option`` determines how the change is applied. Args: reference (:class:`~google.cloud.firestore_v1.document.DocumentReference`): A document reference that will be deleted in this batch. option (Optional[:class:`~google.cloud.firestore_v1.client.WriteOption`]): A write option to make assertions / preconditions on the server state of the document before applying changes. N)r pb_for_deleterrr!)rrr/write_pbrrrdeletes zBaseBatch.delete)r r )rr)rrr r )rrr$r%r r )F)rrr$r(r)r*r r r )rrr.r%r/r0r r )rrr/r0r r )r3 __module__ __qualname____doc__rrrr!abcabstractmethodr#r'r-r6r9rrrrr s      &  r ) metaclassc@seZdZdZd ddZdS) BaseWriteBatchzBase class for a/sync implementations of the `commit` RPC. `commit` is useful for lower volumes or when the order of write operations is important.r2retries.Retry | retries.AsyncRetry | object | Nonetimeout float | NonecCs&|jj|jdd}t||}||fS)z+Shared setup for async/sync :meth:`commit`.N)databasewrites transaction)r_database_stringrrmake_retry_timeout_kwargs)rrrBrequestkwargsrrr _prep_commits  zBaseWriteBatch._prep_commitN)rrArBrC)r3r:r;r<rKrrrrr@sr@)r< __future__rr=typingrrrgoogle.api_corerretriesgoogle.cloud.firestore_v1r'google.cloud.firestore_v1.base_documentrgoogle.cloud.firestore_v1.typesr r8ABCMetar r@rrrrs