Skip to content

Commit 6b9c807

Browse files
committed
[operation] add ListOperations api implementation
1 parent 8aad508 commit 6b9c807

File tree

2 files changed

+29
-0
lines changed

2 files changed

+29
-0
lines changed

ydb/_apis.py

+1
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ class OperationService(object):
6666
ForgetOperation = "ForgetOperation"
6767
GetOperation = "GetOperation"
6868
CancelOperation = "CancelOperation"
69+
ListOperations = "ListOperations"
6970

7071

7172
class SchemeService(object):

ydb/operation.py

+28
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,14 @@ def _get_operation_request(self):
2626
return request
2727

2828

29+
def _list_operations_response(rpc_state, response): # pylint: disable=W0613
30+
issues._process_response(response)
31+
32+
33+
def _list_operations_request(kind: str, page_size: int, page_token: str):
34+
return _apis.ydb_operation.ListOperationsRequest(kind=kind, page_size=page_size, page_token=page_token)
35+
36+
2937
class OperationClient(object):
3038
def __init__(self, driver):
3139
self._driver = driver
@@ -48,6 +56,15 @@ def forget(self, operation_id, settings=None):
4856
settings,
4957
)
5058

59+
def list(self, kind: str, page_size: int, page_token: str, settings=None):
60+
return self._driver(
61+
_list_operations_request(kind, page_size, page_token),
62+
_apis.OperationService.Stub,
63+
_apis.OperationService.ListOperations,
64+
_list_operations_response,
65+
settings,
66+
)
67+
5168

5269
class Operation(object):
5370
__slots__ = ("id", "_driver", "self_cls")
@@ -99,3 +116,14 @@ def get(self, settings=None):
99116
settings,
100117
(self._driver,),
101118
)
119+
120+
def list(self, kind: str, page_size: int, page_token: str, settings=None):
121+
self._ensure_implements()
122+
return self._driver(
123+
_list_operations_request(kind, page_size, page_token),
124+
_apis.OperationService.Stub,
125+
_apis.OperationService.ListOperations,
126+
self.__class__,
127+
settings,
128+
(self._driver,),
129+
)

0 commit comments

Comments
 (0)