feat: Add cancel_session method to class SessionFieldFilter (#430)

aneoconsulting · Oct 16, 2023 · 3b4b1ff · 3b4b1ff
2 parents 475c14a + 080161a
commit 3b4b1ff
Showing 1 changed file with 10 additions and 1 deletion.
diff --git 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))
+