From 080161a78c410f18643cf56aea50faa5d19acb61 Mon Sep 17 00:00:00 2001 From: jeremyzyngeraneo Date: Mon, 16 Oct 2023 10:27:20 +0200 Subject: [PATCH] feat: Add cancel_session method to class SessionFieldFilter --- packages/python/src/armonik/client/sessions.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/packages/python/src/armonik/client/sessions.py b/packages/python/src/armonik/client/sessions.py index 1df69d6b5..8c7912b9e 100644 --- a/packages/python/src/armonik/client/sessions.py +++ b/packages/python/src/armonik/client/sessions.py @@ -4,7 +4,7 @@ from ..protogen.client.sessions_service_pb2_grpc import SessionsStub from ..protogen.common.submitter_common_pb2 import SessionFilter -from ..protogen.common.sessions_common_pb2 import GetSessionRequest, GetSessionResponse, ListSessionsRequest, ListSessionsResponse, SessionRaw +from ..protogen.common.sessions_common_pb2 import GetSessionRequest, GetSessionResponse, ListSessionsRequest, ListSessionsResponse, SessionRaw, CancelSessionRequest, CancelSessionResponse from ..protogen.common.sessions_filters_pb2 import Filters as rawFilters, FiltersAnd as rawFilterAnd, FilterField as rawFilterField, FilterStatus as rawFilterStatus from ..protogen.common.sessions_fields_pb2 import * from ..common.filter import StringFilter, StatusFilter, DateFilter, NumberFilter, Filter @@ -64,3 +64,12 @@ def list_sessions(self, task_filter: Filter, page: int = 0, page_size: int = 100 ) list_response : ListSessionsResponse = self._client.ListSessions(request) return list_response.total, [Session.from_message(t) for t in list_response.sessions] + + def cancel_session(self, session_id: str) -> None: + """Cancel a session + + Args: + session_id: Id of the session to b cancelled + """ + self._client.CancelSession(CancelSessionRequest(session_id=session_id)) + \ No newline at end of file