aboutsummaryrefslogtreecommitdiff
path: root/google/api_core/operation.py
diff options
context:
space:
mode:
Diffstat (limited to 'google/api_core/operation.py')
-rw-r--r--google/api_core/operation.py16
1 files changed, 12 insertions, 4 deletions
diff --git a/google/api_core/operation.py b/google/api_core/operation.py
index a806523..b17f753 100644
--- a/google/api_core/operation.py
+++ b/google/api_core/operation.py
@@ -310,8 +310,12 @@ def from_grpc(operation, operations_stub, result_type, grpc_metadata=None, **kwa
~.api_core.operation.Operation: The operation future to track the given
operation.
"""
- refresh = functools.partial(_refresh_grpc, operations_stub, operation.name, metadata=grpc_metadata)
- cancel = functools.partial(_cancel_grpc, operations_stub, operation.name, metadata=grpc_metadata)
+ refresh = functools.partial(
+ _refresh_grpc, operations_stub, operation.name, metadata=grpc_metadata
+ )
+ cancel = functools.partial(
+ _cancel_grpc, operations_stub, operation.name, metadata=grpc_metadata
+ )
return Operation(operation, refresh, cancel, result_type, **kwargs)
@@ -338,6 +342,10 @@ def from_gapic(operation, operations_client, result_type, grpc_metadata=None, **
~.api_core.operation.Operation: The operation future to track the given
operation.
"""
- refresh = functools.partial(operations_client.get_operation, operation.name, metadata=grpc_metadata)
- cancel = functools.partial(operations_client.cancel_operation, operation.name, metadata=grpc_metadata)
+ refresh = functools.partial(
+ operations_client.get_operation, operation.name, metadata=grpc_metadata
+ )
+ cancel = functools.partial(
+ operations_client.cancel_operation, operation.name, metadata=grpc_metadata
+ )
return Operation(operation, refresh, cancel, result_type, **kwargs)