o  iz@s`dZddlmZddlmZddlmZddlmZddl m Z ddl m Z Gdd d e Z d S) z=Helpers for batch requests to the Google Cloud Firestore API.) annotations)gapic_v1)retry)_helpers) BaseBatch)BatchWriteResponsecs@eZdZdZdfdd Zejjdfdd d ZdddZ Z S)BulkWriteBatchaAccumulate write operations to be sent in a batch. Use this over `WriteBatch` for higher volumes (e.g., via `BulkWriter`) and when the order of operations within a given batch is unimportant. Because the order in which individual write operations are applied to the database is not guaranteed, `batch_write` RPCs can never contain multiple operations to the same document. If calling code detects a second write operation to a known document reference, it should first cut off the previous batch and send it, then create a new batch starting with the latest write operation. In practice, the [Async]BulkWriter classes handle this. 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. returnNonecstt|j|ddS)N)client)superr__init__)selfr  __class__/var/www/snowflake_co_dev_github/snow_flake_back_end_deploy/env/lib/python3.10/site-packages/google/cloud/firestore_v1/bulk_batch.pyr .szBulkWriteBatch.__init__Nrretries.Retry | object | Nonetimeout float | NonercCsH|||\}}|jj}|jd||jjd|}g|_t|j|_|S)aWrites the changes accumulated in this batch. Write operations are not guaranteed to be applied in order and must not contain multiple writes to any given document. Preferred over `commit` for performance reasons if these conditions are acceptable. 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: :class:`google.cloud.firestore_v1.write.BatchWriteResponse`: Container holding the write results corresponding to the changes committed, returned in the same order as the changes were applied to this batch. An individual write result contains an ``update_time`` field. )requestmetadataNr) _prep_commit_client_firestore_api batch_write _rpc_metadata _write_pbslist write_results)rrrrkwargs_api save_responserrrcommit1s zBulkWriteBatch.commitcCs&|jj|jdd}t||}||fS)N)databasewriteslabels)r_database_stringrrmake_retry_timeout_kwargs)rrrrr rrrrWs  zBulkWriteBatch._prep_commit)r r )rrrrr r)rrrr) __name__ __module__ __qualname____doc__r rmethodDEFAULTr#r __classcell__rrrrrs &rN)r, __future__rgoogle.api_corerrretriesgoogle.cloud.firestore_v1r$google.cloud.firestore_v1.base_batchr)google.cloud.firestore_v1.types.firestorerrrrrrs