o ˷eU2@sdZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z Gd d d ejZdd d Zd dZddZdddZddZdddZdddZdS)a6Futures for long-running operations returned from Google Cloud APIs. These futures can be used to synchronously wait for the result of a long-running operation using :meth:`Operation.result`: .. code-block:: python operation = my_api_client.long_running_method() result = 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 = future.result() operation.add_done_callback(my_callback) N) exceptions)protobuf_helpers)polling)operations_pb2) json_format)code_pb2cseZdZdZdejffdd ZeddZeddZ e d d Z d d Z ejfd dZ ejfddZddZddZZS) OperationaA 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`. NcsFtt|j|d||_||_||_||_||_t |_ | dS)Nretry) superr__init__ _operation_refresh_cancel _result_type_metadata_type threadingLock_completion_lock_set_result_from_operation)self operationrefreshcancel result_type metadata_typer  __class__P/var/www/ideatree/venv/lib/python3.10/site-packages/google/api_core/operation.pyr Fs   zOperation.__init__cCs|jS)zAgoogle.longrunning.Operation: The current long-running operation.)r rrrrrYszOperation.operationcCs"|jdsdSt|j|jjS)z8google.protobuf.Message: the current operation metadata.metadataN)r HasFieldr from_any_pbrr!r rrrr!^s  zOperation.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. )rr FromString)rpayloadrrr deserializehs zOperation.deserializecCs|jg|jjr |jr WddS|jdr)t|j|jj}| |n1|jdrHt j |jj j |jj j|jj f|jd}||nt d}||WddSWddSWddS1smwYdS)zASet the result or exception from the operation if it is complete.Nresponseerror) status_codemessageerrorsr'zLUnexpected state: Long-running operation had neither response nor error set.)rr done _result_setr"rr#rr' set_resultrfrom_grpc_statusr(coder* set_exceptionGoogleAPICallError)rr' exceptionrrrrts4       "z$Operation._set_result_from_operationcCs&|jjs|j|d|_|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)r r,rrrr rrr_refresh_and_updates zOperation._refresh_and_updatecCs|||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. )r5r r,r4rrrr,s zOperation.donecCs|rdS|dS)zAttempt to cancel the operation. Returns: bool: True if the cancel RPC was made, False if the operation is already complete. FT)r,rr rrrrszOperation.cancelcCs$||jdo|jjjtjkS)z$True if the operation was cancelled.r()r5r r"r(r0r CANCELLEDr rrr cancelleds zOperation.cancelled)__name__ __module__ __qualname____doc__r DEFAULT_RETRYr propertyrr! classmethodr&rr5r,rr7 __classcell__rrrrr2s     !   rcCs6d|}|dur ||}|d|d}t|tS)aRefresh an operation using a JSON/HTTP client. Args: api_request (Callable): A callable used to make an API request. This should generally be :meth:`google.cloud._http.Connection.api_request`. operation_name (str): The name of the operation. retry (google.api_core.retry.Retry): (Optional) retry policy Returns: google.longrunning.operations_pb2.Operation: The operation. z operations/{}NGETmethodpath)formatr ParseDictrr) api_requestoperation_namer rC api_responserrr _refresh_https  rIcCsd|}|d|ddS)a"Cancel an operation using a JSON/HTTP client. Args: api_request (Callable): A callable used to make an API request. This should generally be :meth:`google.cloud._http.Connection.api_request`. operation_name (str): The name of the operation. zoperations/{}:cancelPOSTrAN)rD)rFrGrCrrr _cancel_https rKcKsFt|t}tt||j}tt||j}t||||fi|S)aCreate an operation future using a HTTP/JSON client. This interacts with the long-running operations `service`_ (specific to a given API) via `HTTP/JSON`_. .. _HTTP/JSON: https://cloud.google.com/speech/reference/rest/ v1beta1/operations#Operation Args: operation (dict): Operation as a dictionary. api_request (Callable): A callable used to make an API request. This should generally be :meth:`google.cloud._http.Connection.api_request`. result_type (:func:`type`): The protobuf result type. kwargs: Keyword args passed into the :class:`Operation` constructor. Returns: ~.api_core.operation.Operation: The operation future to track the given operation. ) rrErr functoolspartialrInamerK)rrFrkwargsoperation_protorrrrrfrom_http_jsonsrQcCs*tj|d}|j}|dur||}||S)a{Refresh an operation using a gRPC client. Args: operations_stub (google.longrunning.operations_pb2.OperationsStub): The gRPC operations stub. operation_name (str): The name of the operation. retry (google.api_core.retry.Retry): (Optional) retry policy Returns: google.longrunning.operations_pb2.Operation: The operation. rNN)rGetOperationRequest GetOperation)operations_stubrGr request_pbrpcrrr _refresh_grpcs rXcCstj|d}||dS)zCancel an operation using a gRPC client. Args: operations_stub (google.longrunning.operations_pb2.OperationsStub): The gRPC operations stub. operation_name (str): The name of the operation. rRN)rCancelOperationRequestCancelOperation)rUrGrVrrr _cancel_grpcs r[cKs>tjt||j|d}tjt||j|d}t||||fi|S)aCreate an operation future using a gRPC client. This interacts with the long-running operations `service`_ (specific to a given API) via gRPC. .. _service: https://github.com/googleapis/googleapis/blob/ 050400df0fdb16f63b63e9dee53819044bffc857/ google/longrunning/operations.proto#L38 Args: operation (google.longrunning.operations_pb2.Operation): The operation. operations_stub (google.longrunning.operations_pb2.OperationsStub): The operations stub. 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!)rLrMrXrNr[r)rrUr grpc_metadatarOrrrrr from_grpc"  r^cKs>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\)rLrM get_operationrNcancel_operationr)roperations_clientrr]rOrrrrr from_gapicBr_rc)N)r;rLrgoogle.api_corerrgoogle.api_core.futurergoogle.longrunningrgoogle.protobufr google.rpcr PollingFuturerrIrKrQrXr[r^rcrrrrs$