Ѯh~4dZddlmZddlmZmZmZmZmZm Z ddl m Z m Z ddl m ZddlmZmZddlmZddlmZdd lmZmZmZmZmZmZmZmZerdd lm Z dd l!m"Z"dd l#m$Z$Gd dejJeZ&GddeZ' ddZ(y)zEHelpers for applying Google Cloud Firestore changes in a transaction.) annotations) TYPE_CHECKINGAnyAsyncGeneratorCallable CoroutineOptional) exceptionsgapic_v1) retry_async)_helpers async_batch)AsyncDocumentReference) AsyncQuery) _CANT_BEGIN _CANT_COMMIT_CANT_ROLLBACK_EXCEED_ATTEMPTS_TEMPLATE_WRITE_READ_ONLY MAX_ATTEMPTSBaseTransaction_BaseTransactional)AsyncStreamGenerator)DocumentSnapshot)ExplainOptionsceZdZdZedfd fd Zd fd ZdddZd dZddZ e jjdf dd Z e jjdfdd  dd ZxZS)AsyncTransactionaOAccumulate read-and-write operations to be sent in a transaction. Args: client (:class:`~google.cloud.firestore_v1.async_client.AsyncClient`): 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`. FcZtt| |tj|||yN)superr__init__r)selfclient max_attempts read_only __class__s }/home/www/academy-backend.kofcorporation.com/venv/lib/python3.12/site-packages/google/cloud/firestore_v1/async_transaction.pyr!zAsyncTransaction.__init__:s% .v6  |Y?cb|jrtttt||y)a Add `Write`` protobufs to this transaction. Args: write_pbs (List[google.cloud.firestore_v1. write.Write]): A list of write protobufs to be added. Raises: ValueError: If this transaction is read-only. N) _read_only ValueErrorrr r_add_write_pbs)r" write_pbsr&s r'r,zAsyncTransaction._add_write_pbs>s( ??-. . 4Y?r(NcpK|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"retry_idmsgtransaction_responses r'_beginzAsyncTransaction._beginMs   $$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. )r/r=r1N) r4r+rr7r8rollbackr:r6r< _clean_up)r"s r' _rollbackzAsyncTransaction._rollbackds^, , ,,--66 $ = =#'8833 7   NN   NN s/B+AB>B?BB+BB((B+cK|jstt|jjj |jj |j|jd|jjd{}|jt|j|_ |j|_ |jS7Kw)aTransactionally commit the changes accumulated. 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 transaction. A write result contains an ``update_time`` field. Raises: ValueError: If no transaction is in progress. )r/writesr=r1N)r4r+rr7r8commitr: _write_pbsr6r<rDlist write_results commit_time)r"commit_responses r'_commitzAsyncTransaction._commit{s\* * $ ; ; B B LL99//#xx \\// !C!   !/"?"?@*66!!! sBCC A CcKtj||}|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)r make_retry_timeout_kwargsr7get_all)r" referencesretrytimeoutkwargss r'rQzAsyncTransaction.get_allsA(33E7C)T\\))*Q$Q&QQQQs9AAA)explain_optionsc0Ktj||}t|tr5| t d|j j |gfd|i|d{St|tr|||d<|jdd|i|St d7:w)a Retrieve a document or a query result from the database. Args: ref_or_query (AsyncDocumentReference | AsyncQuery): 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. explain_options (Optional[:class:`~google.cloud.firestore_v1.query_profile.ExplainOptions`]): Options to enable query profiling for this query. When set, explain_metrics will be available on the returned generator. Can only be used when running a query, not a document reference. Yields: DocumentSnapshot: The next document snapshot that fulfills the query, or :data:`None` if the document does not exist. Raises: ValueError: if `ref_or_query` is not one of the supported types, or explain_options is provided when `ref_or_query` is a document reference. Nz^When type of `ref_or_query` is `AsyncDocumentReference`, `explain_options` cannot be provided.r=rVzSValue for argument "ref_or_query" must be a AsyncDocumentReference or a AsyncQuery.) r rP isinstancerr+r7rQrstream)r" ref_or_queryrSrTrVrUs r'getzAsyncTransaction.getsB33E7C l$: ;* <.--,04:  j 1*,;()&<&&B4B6B Be sABB;BreturnNone)r-rJr^r_r)r>z bytes | Noner^r_)r^rJ)rRrJrS"retries.AsyncRetry | object | NonerTz float | Noner^z%AsyncGenerator[DocumentSnapshot, Any]) r[z#AsyncDocumentReference | AsyncQueryrSr`rTzOptional[float]rVzOptional[ExplainOptions]r^zNAsyncGenerator[DocumentSnapshot, Any] | AsyncStreamGenerator[DocumentSnapshot])__name__ __module__ __qualname____doc__rr!r,rArErNr methodDEFAULTrQr\ __classcell__r&s@r'rr,s -9E@ @4.."@5=OO4K4K $ RR2R R / R45=OO4K4K#' 2 59 29222! 2 2 2 X2r(rc:eZdZdZdfd Z ddZdZxZS)_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. c,tt| |yr)r rjr!)r"to_wrapr&s r'r!z_AsyncTransactional.__init__s !41':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)rDrAr>r6 current_idrl)r"r=argsrUs 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``. rXN) _resetr*r Abortedrange _max_attemptsrprNrr5r+ BaseExceptionrE) r"r=rorUretryable_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=rr^r)rarbrcrdr!rpr|rgrhs@r'rjrjs*;@+@ @<0r(rjct|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. )rj)rls r'async_transactionalr~Cs w ''r(N)rlz!Callable[[AsyncTransaction], Any]r^rj))rd __future__rtypingrrrrrr google.api_corer r r retriesgoogle.cloud.firestore_v1r r(google.cloud.firestore_v1.async_documentr%google.cloud.firestore_v1.async_queryr*google.cloud.firestore_v1.base_transactionrrrrrrrr0google.cloud.firestore_v1.async_stream_generatorr'google.cloud.firestore_v1.base_documentr'google.cloud.firestore_v1.query_profilerAsyncWriteBatchrrjr~rXr(r'rs|L"TT02;K<   UHFu{22Oup\,\~( .((r(