o }l~il:@s0dZddlmZddlmZmZmZmZmZm Z m Z ddl m Z m Z mZddlmZmZddlmZddlmZmZddlmZdd lmZdd lmZmZmZmZm Z m!Z!m"Z"m#Z#errdd l$Z$dd l%m&Z&dd l'm(Z(ddl)m*Z*edZ+e dZ,Gdddej-e"Z.Gddde#e e+e,fZ/dddZ0d S)zEHelpers for applying Google Cloud Firestore changes in a transaction.) annotations) TYPE_CHECKINGAnyAsyncGenerator AwaitableCallableGenericOptional) Concatenate ParamSpecTypeVar) exceptionsgapic_v1) retry_async)_helpers async_batch)AsyncDocumentReference) AsyncQuery) _CANT_BEGIN _CANT_COMMIT_CANT_ROLLBACK_EXCEED_ATTEMPTS_TEMPLATE_WRITE_READ_ONLY MAX_ATTEMPTSBaseTransaction_BaseTransactionalN)AsyncStreamGenerator)DocumentSnapshot)ExplainOptionsTPcseZdZdZedfd)fdd Zd*fd d Zd+d,ddZd)ddZd-ddZ e j j d fd dd.ddZ e j j d fd d dd/d'd(ZZS)0AsyncTransactionaOAccumulate 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`. FreturnNonecs"tt||t|||dSN)superr!__init__r)selfclient max_attempts read_only __class__r/var/www/newdalilibackend/backend/venv/lib/python3.10/site-packages/google/cloud/firestore_v1/async_transaction.pyr&IszAsyncTransaction.__init__ write_pbslistcs"|jrtttt||dS)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'r/r+r-r.r3Ms zAsyncTransaction._add_write_pbsNretry_id bytes | NonecsT|jrt|j}t||jjj|jj| |d|jj dIdH}|j |_dS)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_idr2_client_firestore_apibegin_transaction_database_string_options_protobuf _rpc_metadata transaction)r'r4msgtransaction_responser-r-r._begin\s  zAsyncTransaction._begincsR|jsttz|jjj|jj|jd|jjdIdHW| dS| w)zRoll back the transaction. Raises: ValueError: If no transaction is in progress. google.api_core.exceptions.GoogleAPICallError: If the rollback fails. )r6rDr8N) r;r2rr>r?rollbackrAr=rC _clean_up)r'r-r-r. _rollbackss zAsyncTransaction._rollbackcs^|jstt|jjj|jj|j|jd|jj dIdH}| t |j |_ |j |_ |j S)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. )r6writesrDr8N)r;r2rr>r?commitrA _write_pbsr=rCrIr0 write_results commit_time)r'commit_responser-r-r._commits   zAsyncTransaction._commit) read_time referencesretry"retries.AsyncRetry | object | Nonetimeout float | NonerRdatetime.datetime | None%AsyncGenerator[DocumentSnapshot, Any]cs<t||}|dur||d<|jj|fd|i|IdHS)aRetrieves 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. read_time (Optional[datetime.datetime]): If set, reads documents as they were at the given time. This must be a timestamp within the past one hour, or if Point-in-Time Recovery is enabled, can additionally be a whole minute timestamp within the past 7 days. If no timezone is specified in the :class:`datetime.datetime` object, it is assumed to be UTC. Yields: .DocumentSnapshot: The next document snapshot that fulfills the query, or :data:`None` if the document does not exist. NrRrD)rmake_retry_timeout_kwargsr>get_all)r'rSrTrVrRkwargsr-r-r.r[s  zAsyncTransaction.get_all)explain_optionsrR ref_or_query#AsyncDocumentReference | AsyncQueryOptional[float]r]Optional[ExplainOptions]Optional[datetime.datetime]NAsyncGenerator[DocumentSnapshot, Any] | AsyncStreamGenerator[DocumentSnapshot]cst||}|dur||d<t|tr,|durtd|jj|gfd|i|IdHSt|trC|dur9||d<|jdd|i|Std)aP 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. read_time (Optional[datetime.datetime]): If set, reads documents as they were at the given time. This must be a timestamp within the past one hour, or if Point-in-Time Recovery is enabled, can additionally be a whole minute timestamp within the past 7 days. If no timezone is specified in the :class:`datetime.datetime` object, it is assumed to be UTC. 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. NrRz^When type of `ref_or_query` is `AsyncDocumentReference`, `explain_options` cannot be provided.rDr]zSValue for argument "ref_or_query" must be a AsyncDocumentReference or a AsyncQuery.r-) rrZ isinstancerr2r>r[rstream)r'r^rTrVr]rRr\r-r-r.gets. &  zAsyncTransaction.get)r"r#)r/r0r"r#r$)r4r5r"r#)r"r0) rSr0rTrUrVrWrRrXr"rY) r^r_rTrUrVr`r]rarRrbr"rc)__name__ __module__ __qualname____doc__rr&r3rGrJrQrmethodDEFAULTr[rf __classcell__r-r-r+r.r!;s$     "r!cs6eZdZdZdfdd ZdddZdddZZS)_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. to_wrap8Callable[Concatenate[AsyncTransaction, P], Awaitable[T]]r"r#cstt||dSr$)r%rnr&)r'ror+r-r.r& sz_AsyncTransactional.__init__rDr!argsP.argsr\P.kwargsrcsV||j|jdIdH|j|_|jdur|j|_|j|g|Ri|IdHS)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: T: result of the wrapped coroutine. Raises: Exception: Any failure caused by ``to_wrap``. )r4N)rIrGr4r= current_idro)r'rDrqr\r-r-r. _pre_commits z_AsyncTransactional._pre_commitc s||js tjnd}d}zCt|jD]2}|j|g|Ri|IdH}z |IdH|WWS|yG}z|}WYd}~qd}~wwt |j} t | |t ya| IdHw)aExecute the wrapped callable within a transaction. Args: transaction (:class:`~google.cloud.firestore_v1.async_transaction.AsyncTransaction`): 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: T: The result of the wrapped callable. Raises: ValueError: If the transaction does not succeed in ``max_attempts``. r-N) _resetr1r Abortedrange _max_attemptsrurQrr<r2 BaseExceptionrJ) r'rDrqr\retryable_exceptionslast_excattemptresultexcrEr-r-r.__call__0s*   z_AsyncTransactional.__call__)rorpr"r#)rDr!rqrrr\rsr"r)rgrhrirjr&rurrmr-r-r+r.rns  rnrorpr"cCst|S)aDecorate a callable so that it runs in a transaction. Args: to_wrap (Callable[[:class:`~google.cloud.firestore_v1.async_transaction.AsyncTransaction`, ...], Awaitable[Any]]): A callable that should be run (and retried) in a transaction. Returns: Callable[[:class:`~google.cloud.firestore_v1.transaction.Transaction`, ...], Awaitable[Any]]: the wrapped callable. )rn)ror-r-r.async_transactionalesr)rorpr"rp)1rj __future__rtypingrrrrrrr typing_extensionsr r r google.api_corer rrretriesgoogle.cloud.firestore_v1rr(google.cloud.firestore_v1.async_documentr%google.cloud.firestore_v1.async_queryr*google.cloud.firestore_v1.base_transactionrrrrrrrrdatetime0google.cloud.firestore_v1.async_stream_generatorr'google.cloud.firestore_v1.base_documentr'google.cloud.firestore_v1.query_profilerrr AsyncWriteBatchr!rnrr-r-r-r.s* $    (    Hc