o  i9 @sHdZddlmZddlmZddlmZddlmZGdddeZ dS) z=Helpers for batch requests to the Google Cloud Firestore API.) annotations)gapic_v1) retry_async)BaseWriteBatchcsFeZdZdZdfdd Zejjdfdd d ZddZ ddZ Z S)AsyncWriteBatchaAccumulate write operations to be sent in a batch. This has the same set of methods for write operations that :class:`~google.cloud.firestore_v1.async_document.AsyncDocumentReference` does, e.g. :meth:`~google.cloud.firestore_v1.async_document.AsyncDocumentReference.create`. Args: client (:class:`~google.cloud.firestore_v1.async_client.AsyncClient`): 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/async_batch.pyr $szAsyncWriteBatch.__init__Nretry"retries.AsyncRetry | object | Nonetimeout float | NonelistcsX|||\}}|jjjd||jjd|IdH}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.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. )requestmetadataNr) _prep_commit_client_firestore_apicommit _rpc_metadata _write_pbsr write_results commit_time)r rrrkwargscommit_responseresultsrrrr's  zAsyncWriteBatch.commitcs|SNr)r rrr __aenter__HszAsyncWriteBatch.__aenter__cs |dur|IdHdSdSr#)r)r exc_type exc_value tracebackrrr __aexit__KszAsyncWriteBatch.__aexit__)rr)rrrrrr) __name__ __module__ __qualname____doc__r rmethodDEFAULTrr$r( __classcell__rrr rrs  !rN) r, __future__rgoogle.api_corerrretries$google.cloud.firestore_v1.base_batchrrrrrrs