Ѯh>dZddlmZddlZddlmZmZddlmZ ddl m Z ddl m Z ddlmZGd d ej" ZGd d eZy)z=Helpers for batch requests to the Google Cloud Firestore API.) annotationsN)DictUnion)retry)_helpers)BaseDocumentReference)writeceZdZdZd dZdZd dZddZejdZ ddZ d ddZ d dd Z d dd Zy ) 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. cJ||_g|_i|_d|_d|_yN)_client _write_pbs_document_references write_results commit_time)selfclients v/home/www/academy-backend.kofcorporation.com/venv/lib/python3.12/site-packages/google/cloud/firestore_v1/base_batch.py__init__zBaseBatch.__init__(s( 02FH!@Dc,t|jSr )lenrrs r__len__zBaseBatch.__len__/s4,,--rc2|j|jvSr )_document_pathr)r references r __contains__zBaseBatch.__contains__2s''4+D+DDDrc:|jj|y)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)r write_pbss r_add_write_pbszBaseBatch._add_write_pbs5s y)rct)zySends all accumulated write operations to the server. The details of this write depend on the implementing class.)NotImplementedErrorrs rcommitzBaseBatch.commit@s "##rctj|j|}||j|j<|j |y)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#)rr document_datar"s rcreatezBaseBatch.createFs@++I,D,DmT >G!!)":":; I&rc|dur"tj|j||}n tj|j|}||j|j<|j |y)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)merger"s rsetz BaseBatch.setVsm,   77((-I!55((-I?H!!)":":; I&rNc|jjdk(r tdtj|j ||}||j |j <|j|y)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#)rr field_updatesoptionr"s rupdatezBaseBatch.updatexsh*    $ $ 6WX X++  $ $mV ?H!!)":":; I&rctj|j|}||j|j<|j |gy)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#)rrr7write_pbs rdeletezBaseBatch.deletesB&)))*B*BFK>G!!)":":; XJ'r)returnNone)rr)r"zlist[write_pb.Write]r=r>)rrr)dictr=r>)F)rrr)r?r.zUnion[bool, list]r=r>r )rrr6r?r7_helpers.WriteOption | Noner=r>)rrr7r@r=r>)r3 __module__ __qualname____doc__rrrr#abcabstractmethodr&r*r/r8r<rrr r s  .E * $$ '($) '( ' '! '  'L/3 '('', '  '@/3(((,(  (rr ) metaclassc eZdZdZ ddZy)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.c~|jj|jdd}tj||}||fS)z+Shared setup for async/sync :meth:`commit`.N)databasewrites transaction)r_database_stringrrmake_retry_timeout_kwargs)rrtimeoutrequestkwargss r _prep_commitzBaseWriteBatch._prep_commits@ 55oo  33E7CrN)rz2retries.Retry | retries.AsyncRetry | object | NonerPz float | None)r3rArBrCrSrFrrrIrIsM A  rrI)rC __future__rrDtypingrrgoogle.api_corerretriesgoogle.cloud.firestore_v1r'google.cloud.firestore_v1.base_documentrgoogle.cloud.firestore_v1.typesr r;ABCMetar rIrFrrr\sAD" -.I=N(#++N(bYr