o /i!@s:dZddlZddlmZddlmZGdddejZdS)atFutures for extended long-running operations returned from Google Cloud APIs. These futures can be used to synchronously wait for the result of a long-running operations using :meth:`ExtendedOperation.result`: .. code-block:: python extended_operation = my_api_client.long_running_method() extended_operation.result() Or asynchronously using callbacks and :meth:`Operation.add_done_callback`: .. code-block:: python extended_operation = my_api_client.long_running_method() def my_callback(ex_op): print(f"Operation {ex_op.name} completed") extended_operation.add_done_callback(my_callback) N) exceptions)pollingcseZdZdZejffdd ZeddZeddZ edd Z ed d Z d d Z dddZ ddZddZdddZddZeddZZS)ExtendedOperationaAAn ExtendedOperation future for interacting with a Google API Long-Running Operation. Args: extended_operation (proto.Message): The initial operation. refresh (Callable[[], type(extended_operation)]): A callable that returns the latest state of the operation. cancel (Callable[[], None]): A callable that tries to cancel the operation. polling Optional(google.api_core.retry.Retry): The configuration used for polling. This can be used to control how often :meth:`done` is polled. If the ``timeout`` argument to :meth:`result` is specified it will override the ``polling.timeout`` property. retry Optional(google.api_core.retry.Retry): DEPRECATED use ``polling`` instead. If specified it will override ``polling`` parameter to maintain backward compatibility. Note: Most long-running API methods use google.api_core.operation.Operation This class is a wrapper for a subset of methods that use alternative Long-Running Operation (LRO) semantics. Note: there is not a concrete type the extended operation must be. It MUST have fields that correspond to the following, POSSIBLY WITH DIFFERENT NAMES: * name: str * status: Union[str, bool, enum.Enum] * error_code: int * error_message: str c sDtjdd|i|||_||_||_d|_t|_| dS)NrF) super__init___extended_operation_refresh_cancel _cancelled threadingLock_completion_lock_handle_refreshed_operation)selfextended_operationrefreshcancelrkwargs __class__rV/var/www/passon-env/lib/python3.10/site-packages/google/api_core/extended_operation.pyrIs  zExtendedOperation.__init__cC|jjSN)rnamerrrrrjzExtendedOperation.namecCrr)rstatusrrrrrnrzExtendedOperation.statuscCrr)r error_coderrrrrrrzExtendedOperation.error_codecCrr)r error_messagerrrrrvrzExtendedOperation.error_messagecCs t|j|Sr)getattrr)rrrrr __getattr__zs zExtendedOperation.__getattr__NcCs|||jjSr)_refresh_and_updaterdonerretryrrrr#}s zExtendedOperation.donecCs|rdS|d|_dS)NFT)r#r r rrrrrs zExtendedOperation.cancelcCs|jsdS||jjS)NF)r r"rr#rrrr cancelledszExtendedOperation.cancelledcCs2|jjs|r |j|dn||_|dSdS)N)r%)rr#r rr$rrrr"s  z%ExtendedOperation._refresh_and_updatecCs|jm|jjs WddS|jr:|jr:g}t|dr(t|jdr(|jj}tj |j|j|j|d}| |n&|js@|jrSt d|jd|j}| |n| dWddSWddSWddS1sswYdS)Nerrorerrors) status_codemessageresponser(zUnexpected error z: ) rrr#rrhasattrr'r(rfrom_http_status set_exceptionGoogleAPICallError set_result)rr( exceptionrrrrs4     "z-ExtendedOperation._handle_refreshed_operationcKs||||fi|S)aA Return an instantiated ExtendedOperation (or child) that wraps * a refresh callable * a cancel callable (can be a no-op) * an initial result .. note:: It is the caller's responsibility to set up refresh and cancel with their correct request argument. The reason for this is that the services that use Extended Operations have rpcs that look something like the following: // service.proto service MyLongService { rpc StartLongTask(StartLongTaskRequest) returns (ExtendedOperation) { option (google.cloud.operation_service) = "CustomOperationService"; } } service CustomOperationService { rpc Get(GetOperationRequest) returns (ExtendedOperation) { option (google.cloud.operation_polling_method) = true; } } Any info needed for the poll, e.g. a name, path params, etc. is held in the request, which the initial client method is in a much better position to make made because the caller made the initial request. TL;DR: the caller sets up closures for refresh and cancel that carry the properly configured requests. Args: refresh (Callable[Optional[Retry]][type(extended_operation)]): A callable that returns the latest state of the operation. cancel (Callable[][Any]): A callable that tries to cancel the operation on a best effort basis. extended_operation (Any): The initial response of the long running method. See the docstring for ExtendedOperation.__init__ for requirements on the type and fields of extended_operation r)clsrrrrrrrmakes+zExtendedOperation.maker)__name__ __module__ __qualname____doc__rDEFAULT_POLLINGrpropertyrrrrr!r#rr&r"r classmethodr3 __classcell__rrrrr-s( !      r)r7r google.api_corergoogle.api_core.futurer PollingFuturerrrrrs