VSf9 dZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZm Z m!Z!ddl"m#Z#GddejHe Z%GddeZ&de e%gefde&fdZ'de#de(de)dejTfdZ+eefde,de,de,de,fdZ-y)zEHelpers for applying Google Cloud Firestore changes in a transaction.N)gapic_v1) retry_async) _BaseTransactionalBaseTransaction MAX_ATTEMPTS _CANT_BEGIN_CANT_ROLLBACK _CANT_COMMIT_WRITE_READ_ONLY_INITIAL_SLEEP _MAX_SLEEP _MULTIPLIER_EXCEED_ATTEMPTS_TEMPLATE) exceptions) async_batch)_helpers)types)AsyncDocumentReference)DocumentSnapshot) AsyncQuery)AnyAsyncGeneratorCallable Coroutine)Clientc (eZdZdZedfdfd Zdeddffd Zddeddfd Z dd Z defd Z e jjdfd ed ej dedeeeffdZe jjdfd ej dedeeeffdZxZS)AsyncTransactionaDAccumulate read-and-write operations to be sent in a transaction. Args: client (:class:`~google.cloud.firestore_v1.client.Client`): The client that created this transaction. max_attempts (Optional[int]): The maximum number of attempts for the transaction (i.e. allowing retries). Defaults to :attr:`~google.cloud.firestore_v1.transaction.MAX_ATTEMPTS`. read_only (Optional[bool]): Flag indicating if the transaction should be read-only or should allow writes. Defaults to :data:`False`. FreturnNcZtt| |tj|||yN)superr__init__r)selfclient max_attempts read_only __class__s jC:\Users\user\Documents\project_loop\venv\Lib\site-packages\google/cloud/firestore_v1/async_transaction.pyr"zAsyncTransaction.__init__Bs% .v6  |Y? write_pbscb|jrtttt||y)aAdd `Write`` protobufs to this transaction. Args: write_pbs (List[google.cloud.proto.firestore.v1. write.Write]): A list of write protobufs to be added. Raises: ValueError: If this transaction is read-only. N) _read_only ValueErrorr r!r_add_write_pbs)r#r*r's r(r.zAsyncTransaction._add_write_pbsFs( ??-. . 4Y?r)retry_idcpK|jr*tj|j}t ||j j j|j j|j|d|j jd{}|j|_y7w)zBegin the transaction. Args: retry_id (Optional[bytes]): Transaction ID of a transaction to be retried. Raises: ValueError: If the current transaction has already begun. )databaseoptionsrequestmetadataN) in_progressrformat_idr-_client_firestore_apibegin_transaction_database_string_options_protobuf _rpc_metadata transaction)r#r/msgtransaction_responses r(_beginzAsyncTransaction._beginUs   $$TXX.CS/ !%)\\%@%@%R%R LL9911(;\\// &S&  (33 sBB6B4B6cZK|jstt |jjj |jj |jd|jjd{|jy7#|jwxYww)zRoll back the transaction. Raises: ValueError: If no transaction is in progress. google.api_core.exceptions.GoogleAPICallError: If the rollback fails. )r1r?r3N) r6r-r r9r:rollbackr<r8r> _clean_up)r#s r( _rollbackzAsyncTransaction._rollbackls^, , ,,--66 $ = =#'8833 7   NN   NN s/B+AB>B?BB+BB((B+cK|jsttt|j|j |j d{}|jt|jS7)w)aTransactionally commit the changes accumulated. 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 transaction. A write result contains an ``update_time`` field. Raises: ValueError: If no transaction is in progress. N) r6r-r _commit_with_retryr9 _write_pbsr8rElist write_results)r#commit_responses r(_commitzAsyncTransaction._commits`\* * 2 LL$//488!   O1122  sA A8 A6 *A8 referencesretrytimeoutcKtj||}|jj|fd|i|d{S7w)atRetrieves multiple documents from Firestore. Args: references (List[.AsyncDocumentReference, ...]): Iterable of document references to be retrieved. 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. Yields: .DocumentSnapshot: The next document snapshot that fulfills the query, or :data:`None` if the document does not exist. r?N)rmake_retry_timeout_kwargsr9get_all)r#rNrOrPkwargss r(rSzAsyncTransaction.get_allsA(33E7C)T\\))*Q$Q&QQQQs9AAAcKtj||}t|tr(|jj |gfd|i|d{St|t r|jdd|i|d{Std7;7w)aW Retrieve a document or a query result from the database. Args: ref_or_query The document references or query object to return. 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. Yields: .DocumentSnapshot: The next document snapshot that fulfills the query, or :data:`None` if the document does not exist. r?NzSValue for argument "ref_or_query" must be a AsyncDocumentReference or a AsyncQuery.) rrR isinstancerr9rSrstreamr-)r# ref_or_queryrOrPrTs r(getzAsyncTransaction.gets(33E7C l$: ;---,04:  j 1,,,HHHH He  Is$A B  B +B 8B 9B  B rNr )__name__ __module__ __qualname____doc__rr"rJr.bytesrBrFrMrmethodDEFAULTretries AsyncRetryfloatrrrrSrZ __classcell__r's@r(rr4s -9E@ @ @ @4U4d4..3t32%-OO$;$; RR!!R R (#- . R4%-OO$;$; !!  (#- . r)rc:eZdZdZdfd ZdedefdZdZxZ S)_AsyncTransactionalalProvide a callable object to use as a transactional decorater. This is surfaced via :func:`~google.cloud.firestore_v1.async_transaction.transactional`. Args: to_wrap (Coroutine[[:class:`~google.cloud.firestore_v1.async_transaction.AsyncTransaction`, ...], Any]): A coroutine that should be run (and retried) in a transaction. rc,tt| |yr )r!rir")r#to_wrapr's r(r"z_AsyncTransactional.__init__s !41':r)r?cK|j|j|jd{|j|_|j|j|_|j |g|i|d{S7P7w)a}Begin transaction and call the wrapped coroutine. Args: transaction (:class:`~google.cloud.firestore_v1.async_transaction.AsyncTransaction`): A transaction to execute the coroutine within. args (Tuple[Any, ...]): The extra positional arguments to pass along to the wrapped coroutine. kwargs (Dict[str, Any]): The extra keyword arguments to pass along to the wrapped coroutine. Returns: Any: result of the wrapped coroutine. Raises: Exception: Any failure caused by ``to_wrap``. )r/N)rErBr/r8 current_idrk)r#r?argsrTs r( _pre_commitz_AsyncTransactional._pre_commits{*   $-- 888&// ==  OODM!T\\+????? 9 @s"0BBA B>B?BBcK|j|jstjnd}d} t |j D]<}|j |g|i|d{} |jd{|cStj|j } t| |7M76#|$r }|}Yd}~zd}~wwxYw#t$r|jd{7wxYww)aExecute the wrapped callable within a transaction. Args: transaction (:class:`~google.cloud.firestore_v1.transaction.Transaction`): A transaction to execute the callable within. args (Tuple[Any, ...]): The extra positional arguments to pass along to the wrapped callable. kwargs (Dict[str, Any]): The extra keyword arguments to pass along to the wrapped callable. Returns: Any: The result of the wrapped callable. Raises: ValueError: If the transaction does not succeed in ``max_attempts``. rVN) _resetr,rAbortedrange _max_attemptsrorMrr7r- BaseExceptionrF) r#r?rnrTretryable_exceptionslast_excattemptresultexcr@s r(__call__z_AsyncTransactional.__call__s& (3(>(>Z  B   !:!:; #/t// MdMfMM#%--///!M  #,22;3L3LMCS/x /N/+#"H# '') ) )   sv1C21C %B3&C +B7>B5?B7C C2-C 5B77C<C>C CC C/'C*(C//C2r[) r\r]r^r_r"rrror{rfrgs@r(riris*;@+@ @<0r)rirkrct|S)aDecorate a callable so that it runs in a transaction. Args: to_wrap (Callable[[:class:`~google.cloud.firestore_v1.transaction.Transaction`, ...], Any]): A callable that should be run (and retried) in a transaction. Returns: Callable[[:class:`~google.cloud.firestore_v1.transaction.Transaction`, ...], Any]: the wrapped callable. )ri)rks r(async_transactionalr}0s w ''r)r$r*transaction_idcKt} |jj|j||d|jd{S7#t j $rYnwxYwt|d{7}nw)a.Call ``Commit`` on the GAPIC client with retry / sleep. Retries the ``Commit`` RPC on Unavailable. Usually this RPC-level retry is handled by the underlying GAPICd client, but in this case it doesn't because ``Commit`` is not always idempotent. But here we know it is "idempotent"-like because it has a transaction ID. We also need to do our own retry to special-case the ``INVALID_ARGUMENT`` error. Args: client (:class:`~google.cloud.firestore_v1.client.Client`): A client with GAPIC client and configuration details. write_pbs (List[:class:`google.cloud.proto.firestore.v1.write.Write`, ...]): A ``Write`` protobuf instance to be committed. transaction_id (bytes): ID of an existing transaction that this commit will run in. Returns: :class:`google.cloud.firestore_v1.types.CommitResponse`: The protobuf response from ``Commit``. Raises: ~google.api_core.exceptions.GoogleAPICallError: If a non-retryable exception is encountered. )r1writesr?r3N)r r:commitr<r>rServiceUnavailable_sleep)r$r*r~ current_sleeps r(rHrHBs6#M  ..55 & 7 7'#1  -- 6 ,,   %]333  sFA88A AA A8A A A8A  A81A42A8r max_sleep multipliercKtjd|}tj|d{t ||z|S7w)aSleep and produce a new sleep time. .. _Exponential Backoff And Jitter: https://www.awsarchitectureblog.com/ 2015/03/backoff.html Select a duration between zero and ``current_sleep``. It might seem counterintuitive to have so much jitter, but `Exponential Backoff And Jitter`_ argues that "full jitter" is the best strategy. Args: current_sleep (float): The current "max" for sleep interval. max_sleep (Optional[float]): Eventual "max" sleep time multiplier (Optional[float]): Multiplier for exponential backoff. Returns: float: Newly doubled ``current_sleep`` or ``max_sleep`` (whichever is smaller) gN)randomuniformasynciosleepmin)rrr actual_sleeps r(rrosA,>>#}5L -- %%% zM)9 55&s/AAA).r_rrgoogle.api_corerrrc*google.cloud.firestore_v1.base_transactionrrrrr r r r r rrrgoogle.cloud.firestore_v1rrr(google.cloud.firestore_v1.async_documentrr%google.cloud.firestore_v1.async_queryrtypingrrrr google.cloud.firestore_v1.clientrAsyncWriteBatchrrir}rJr`CommitResponserHrerrVr)r(rsL $2    '1.+KE<;;4Z{22OZz\,\~( '(#- .(($*4 *4#*45:*4 *4\.8[66%*6EJ6 6r)