o /in@spdZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z Gdd d ej Z d d d ZdS) aBAsyncIO futures for long-running operations returned from Google Cloud APIs. These futures can be used to await for the result of a long-running operation using :meth:`AsyncOperation.result`: .. code-block:: python operation = my_api_client.long_running_method() result = await operation.result() Or asynchronously using callbacks and :meth:`Operation.add_done_callback`: .. code-block:: python operation = my_api_client.long_running_method() def my_callback(future): result = await future.result() operation.add_done_callback(my_callback) N) exceptions)protobuf_helpers) async_future)operations_pb2)code_pb2cseZdZdZdejffdd ZeddZeddZ e d d Z d d Z ejfd dZ ejfddZddZddZZS)AsyncOperationaA Future for interacting with a Google API Long-Running Operation. Args: operation (google.longrunning.operations_pb2.Operation): The initial operation. refresh (Callable[[], ~.api_core.operation.Operation]): A callable that returns the latest state of the operation. cancel (Callable[[], None]): A callable that tries to cancel the operation. result_type (func:`type`): The protobuf type for the operation's result. metadata_type (func:`type`): The protobuf type for the operation's metadata. retry (google.api_core.retry.Retry): The retry configuration used when polling. This can be used to control how often :meth:`done` is polled. Regardless of the retry's ``deadline``, it will be overridden by the ``timeout`` argument to :meth:`result`. NcsBtj|d||_||_||_||_||_t|_ | dS)Nretry) super__init__ _operation_refresh_cancel _result_type_metadata_type threadingLock_completion_lock_set_result_from_operation)self operationrefreshcancel result_type metadata_typer  __class__S/var/www/passon-env/lib/python3.10/site-packages/google/api_core/operation_async.pyr Es   zAsyncOperation.__init__cCs|jS)zAgoogle.longrunning.Operation: The current long-running operation.)r rrrrrXszAsyncOperation.operationcCs"|jdsdSt|j|jjS)z8google.protobuf.Message: the current operation metadata.metadataN)r HasFieldr from_any_pbrr rrrrr ]s  zAsyncOperation.metadatacCs tj|S)zDeserialize a ``google.longrunning.Operation`` protocol buffer. Args: payload (bytes): A serialized operation protocol buffer. Returns: ~.operations_pb2.Operation: An Operation protobuf object. )r Operation FromString)clspayloadrrr deserializegs zAsyncOperation.deserializecCs|je|jjr |jr WddS|jdr+t|j|jj}| |n-|jdrFt j |jj j |jj f|jd}||nt d}||WddSWddSWddS1skwYdS)zASet the result or exception from the operation if it is complete.Nresponseerror)errorsr(zLUnexpected state: Long-running operation had neither response nor error set.)rr done_futurer!rr"rr( set_resultrGoogleAPICallErrorr)message set_exception)rr( exceptionrrrrss2       "z)AsyncOperation._set_result_from_operationcs.|jjs|j|dIdH|_|dSdS)zRefresh the operation and update the result if needed. Args: retry (google.api_core.retry.Retry): (Optional) How to retry the RPC. rN)r r+r rrr rrr_refresh_and_updates  z"AsyncOperation._refresh_and_updatecs||IdH|jjS)zChecks to see if the operation is complete. Args: retry (google.api_core.retry.Retry): (Optional) How to retry the RPC. Returns: bool: True if the operation is complete, False otherwise. N)r3r r+r2rrrr+s zAsyncOperation.donecs*|IdH}|r dS|IdHdS)zAttempt to cancel the operation. Returns: bool: True if the cancel RPC was made, False if the operation is already complete. NFT)r+r)rresultrrrrs zAsyncOperation.cancelcs,|IdH|jdo|jjjtjkS)z$True if the operation was cancelled.Nr))r3r r!r)coder CANCELLEDrrrr cancelleds  zAsyncOperation.cancelled)__name__ __module__ __qualname____doc__r DEFAULT_RETRYr propertyrr classmethodr'rr3r+rr7 __classcell__rrrrr1s       rcKs>tj|j|j|d}tj|j|j|d}t||||fi|S)aCreate an operation future from a gapic client. This interacts with the long-running operations `service`_ (specific to a given API) via a gapic client. .. _service: https://github.com/googleapis/googleapis/blob/ 050400df0fdb16f63b63e9dee53819044bffc857/ google/longrunning/operations.proto#L38 Args: operation (google.longrunning.operations_pb2.Operation): The operation. operations_client (google.api_core.operations_v1.OperationsClient): The operations client. result_type (:func:`type`): The protobuf result type. grpc_metadata (Optional[List[Tuple[str, str]]]): Additional metadata to pass to the rpc. kwargs: Keyword args passed into the :class:`Operation` constructor. Returns: ~.api_core.operation.Operation: The operation future to track the given operation. )r ) functoolspartial get_operationnamecancel_operationr)roperations_clientr grpc_metadatakwargsrrrrr from_gapicsrH)N)r;r@rgoogle.api_corerrgoogle.api_core.futurergoogle.longrunningr google.rpcr AsyncFuturerrHrrrrs