VSfDUdZddlmZddlmZddlmZmZm Z m Z m Z e e d<e e d<e e d<e e d<e e d <ee d <ee d <e e d <ee d <e e d<dZ dZe e d<dZe e d <ej%dZe e d<ej%dZe e d<dZe e d<dZee d < dZee d < dZee d < dZe e d <dZe e d<GddeZGddeZy)zEHelpers for applying Google Cloud Firestore changes in a transaction.)retry)types)Any CoroutineNoReturnOptionalUnion _CANT_BEGIN _CANT_COMMIT_CANT_RETRY_READ_ONLY_CANT_ROLLBACK_EXCEED_ATTEMPTS_TEMPLATE_INITIAL_SLEEP _MAX_SLEEP_MISSING_ID_TEMPLATE _MULTIPLIER_WRITE_READ_ONLYz@The transaction has already begun. Current transaction ID: {!r}.z:The transaction has no transaction ID, so it cannot be {}.z rolled back committedz8Cannot perform write operation in read-only transaction.g?g>@g@z.Failed to commit transaction in {:d} attempts.z,Only read-write transactions can be retried.c 6eZdZdZedfddZdefdZdee dfde e jjfdZed Zed Zdd Zddefd Zdefd ZdeeeeeefffdZ ddedej4dedefdZ ddej4dedefdZy)BaseTransactionaAccumulate read-and-write operations to be sent in a transaction. Args: 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`. FreturnNc.||_||_d|_yN) _max_attempts _read_only_id)self max_attempts read_onlys iC:\Users\user\Documents\project_loop\venv\Lib\site-packages\google/cloud/firestore_v1/base_transaction.py__init__zBaseTransaction.__init__?s)#ctrNotImplementedError)r write_pbss r!_add_write_pbszBaseTransaction._add_write_pbsD!!r#retry_idc"|O|jrtttjtjj |S|jr2tjtjj Sy)aConvert the current object to protobuf. The ``retry_id`` value is used when retrying a transaction that failed (e.g. due to contention). It is intended to be the "first" transaction that failed (i.e. if multiple retries are needed). Args: retry_id (Union[bytes, NoneType]): Transaction ID of a transaction to be retried. Returns: Optional[google.cloud.firestore_v1.types.TransactionOptions]: The protobuf ``TransactionOptions`` if ``read_only==True`` or if there is a transaction ID to be retried, else :data:`None`. Raises: ValueError: If ``retry_id`` is not :data:`None` but the transaction is read-only. N)retry_transaction) read_write)r )r ValueErrorr rTransactionOptions ReadWriteReadOnlyrr*s r!_options_protobufz!BaseTransaction._options_protobufGs,   !677++ 33==&.>  __++22;;= r#c|jduS)zDetermine if this transaction has already begun. Returns: bool: Indicates if the transaction has started. Nrrs r! in_progresszBaseTransaction.in_progressmsxxt##r#c|jS)zGet the current transaction ID. Returns: Optional[bytes]: The transaction ID (or :data:`None` if the current transaction is not in progress). r5r6s r!idzBaseTransaction.idvs xxr#c g|_d|_y)zClean up the instance after :meth:`_rollback`` or :meth:`_commit``. This intended to occur on success or failure of the associated RPCs. N) _write_pbsrr6s r! _clean_upzBaseTransaction._clean_ups r#ctrr%r2s r!_beginzBaseTransaction._beginr)r#ctrr%r6s r! _rollbackzBaseTransaction._rollbackr)r#ctrr%r6s r!_commitzBaseTransaction._commitr)r# referencesrtimeoutctrr%)rrCrrDs r!get_allzBaseTransaction.get_all "!r#ctrr%)r ref_or_queryrrDs r!getzBaseTransaction.getrGr#rNr)NN)__name__ __module__ __qualname____doc__ MAX_ATTEMPTSr"rr(r bytesrrcommonr/r3propertyr7r9r<r>r@listrrrBretriesRetryfloatrFrJr#r!rr3s %1E "8"$eTk*$ %,,11 2$L$$"x""8""tYsC~%>>?" $ ""}}" "  " $ "}}" "  "r#rc2eZdZdZddZddZdefdZdZy) _BaseTransactionalaYProvide a callable object to use as a transactional decorater. This is surfaced via :func:`~google.cloud.firestore_v1.transaction.transactional`. Args: to_wrap (Callable[[:class:`~google.cloud.firestore_v1.transaction.Transaction`, ...], Any]): A callable that should be run (and retried) in a transaction. rNc0||_d|_ d|_yr)to_wrap current_idr*)rr\s r!r"z_BaseTransactional.__init__s : Ir#c d|_d|_y)zUnset the transaction IDs.N)r]r*r6s r!_resetz_BaseTransactional._resets r#ctrr%r transactionargskwargss r! _pre_commitz_BaseTransactional._pre_commitr)r#ctrr%ras r!__call__z_BaseTransactional.__call__r)r#rK) rLrMrNrOr"r_rrergrXr#r!rZrZs#J "8""r#rZN)rOgoogle.api_corerrUgoogle.cloud.firestore_v1rtypingrrrrr str__annotations__rWrPr rformatr r rrrrrr objectrrZrXr#r!rosL,+<<    AU SUXcX*11-@@(// < c<R#RL EE UN!Q3QKsKl"fl"^""r#