a !f @s<dZddlmZddlmZddlmZGdddeZdS)z=Helpers for batch requests to the Google Cloud Firestore API.)gapic_v1)retry)BaseWriteBatchcsTeZdZdZddfdd Zejjdfej e e dddZ d d Z d d ZZS) WriteBatcha Accumulate write operations to be sent in a batch. Use this over `BulkWriteBatch` for lower volumes or when the order of operations within a given batch is important. 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. N)returncstt|j|ddS)N)client)superr__init__)selfr __class__`/var/www/html/python-backend/venv/lib/python3.9/site-packages/google/cloud/firestore_v1/batch.pyr %szWriteBatch.__init__)rtimeoutrcCsP|||\}}|jjjf||jjd|}g|_t|j|_}|j|_|S)aCommit the changes accumulated in this batch. Args: 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: List[:class:`google.cloud.proto.firestore.v1.write.WriteResult`, ...]: The write results corresponding to the changes committed, returned in the same order as the changes were applied to this batch. A write result contains an ``update_time`` field. )requestmetadata) Z _prep_commitZ_clientZ_firestore_apicommitZ _rpc_metadataZ _write_pbslistZ write_resultsZ commit_time)r rrrkwargsZcommit_responseresultsr r rr(s zWriteBatch.commitcCs|SNr )r r r r __enter__GszWriteBatch.__enter__cCs|dur|dSr)r)r exc_type exc_value tracebackr r r__exit__JszWriteBatch.__exit__)__name__ __module__ __qualname____doc__r rmethodDEFAULTretriesRetryfloatrrrr __classcell__r r r rrs  rN)rZgoogle.api_corerrr"Z$google.cloud.firestore_v1.base_batchrrr r r rs