Skip to content

Commit d63887c

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

File tree

2 files changed

+33
-0
lines changed

2 files changed

+33
-0
lines changed

Diff for: 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):

Diff for: ydb/operation.py

+32
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,18 @@ 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(
35+
kind=kind,
36+
page_size=page_size,
37+
page_token=page_token
38+
)
39+
40+
2941
class OperationClient(object):
3042
def __init__(self, driver):
3143
self._driver = driver
@@ -48,6 +60,15 @@ def forget(self, operation_id, settings=None):
4860
settings,
4961
)
5062

63+
def list(self, kind: str, page_size: int, page_token: str, settings=None):
64+
return self._driver(
65+
_list_operations_request(kind, page_size, page_token),
66+
_apis.OperationService.Stub,
67+
_apis.OperationService.ListOperations,
68+
_list_operations_response,
69+
settings,
70+
)
71+
5172

5273
class Operation(object):
5374
__slots__ = ("id", "_driver", "self_cls")
@@ -99,3 +120,14 @@ def get(self, settings=None):
99120
settings,
100121
(self._driver,),
101122
)
123+
124+
def list(self, kind: str, page_size: int, page_token: str, settings=None):
125+
self._ensure_implements()
126+
return self._driver(
127+
_list_operations_request(kind, page_size, page_token),
128+
_apis.OperationService.Stub,
129+
_apis.OperationService.ListOperations,
130+
self.__class__,
131+
settings,
132+
(self._driver,),
133+
)

0 commit comments

Comments
 (0)