B fndZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z Gdd ej Z 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_pb2ceZdZdZdejffd ZedZedZ e dZ dZ ejfdZ ejfd Zd Zd ZxZS) 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`. Nct|||_||_||_||_||_tj|_ | 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 n/home/panchajanya/Development/work/volatility3/lib/python3.11/site-packages/google/api_core/operation_async.pyrzAsyncOperation.__init__Esm u%%%#  '+ ) 0 0 '')))))c|jS)zAgoogle.longrunning.Operation: The current long-running operation.)rrs rrzAsyncOperation.operationXs r c|jdsdStj|j|jjS)z8google.protobuf.Message: the current operation metadata.metadataN)rHasFieldr from_any_pbrr$r"s rr$zAsyncOperation.metadata]sC'' 33 4+  !9   r c@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)clspayloads r deserializezAsyncOperation.deserializegs'227;;;r cr|j5|jjr|jr ddddS|jdr:t j|j|jj}| |n|jdrLtj |jj j |jj f|j}||n)tj d}||ddddS#1swxYwYdS)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.)rrdone_futurer%rr&rr. set_resultrGoogleAPICallErrorr/message set_exception)rr. exceptions rrz)AsyncOperation._set_result_from_operationss  " . .?' 4<+<+<+>+>   . . . . . . . . '' 33 .+7%t'?))))))'22 .&9O)1 O13!_ ""9----&9. ""9---/ . . . . . . . . . . . . . . . . . .s'D,C#D,,D03D0cK|jjs7||d{V|_|dSdS)zRefresh the operation and update the result if needed. Args: retry (google.api_core.retry.Retry): (Optional) How to retry the RPC. r N)rr1rrrr s r_refresh_and_updatez"AsyncOperation._refresh_and_updates^# .$(MMM$>$>>>>>>>DO  + + - - - - - . .r cTK||d{V|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)r:rr1r9s rr1zAsyncOperation.dones9&&u---------##r czK|d{V}|rdS|d{VdS)zAttempt to cancel the operation. Returns: bool: True if the cancel RPC was made, False if the operation is already complete. NFT)r1r)rresults rrzAsyncOperation.cancelsYyy{{""""""  5,,.. 4r cK|d{V|jdo|jjjt jkS)z$True if the operation was cancelled.Nr/)r:rr%r/coder CANCELLEDr"s r cancelledzAsyncOperation.cancelleds[&&((((((((( O $ $W - - A%*h.@@ r )__name__ __module__ __qualname____doc__r DEFAULT_RETRYrpropertyrr$ classmethodr,rr:r1rrA __classcell__)rs@rr r 1s2(******&X  X  < <[ <...</;.H . . . . ,9 $ $ $ $          r r c tj|j|j|}tj|j|j|}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_metadatakwargsrrs r from_gapicrSsn.'G  *F )Wfk L LV L LLr )N)rErKrgoogle.api_corerrgoogle.api_core.futurergoogle.longrunningr google.rpcr AsyncFuturer rSr rrZs0&&&&&&,,,,,,//////------L L L L L \-L L L ^!M!M!M!M!M!Mr