Refactor event handling in EventHandler class

pyflowlauncher/event.py

@@ -23,8 +23,8 @@ def add_events(self, events: Iterable[Callable[..., Any]]):
     def add_exception_handler(self, exception: Type[Exception], handler: Callable[..., Any]):
         self._handlers[exception] = handler
 
-    def _call_event(self, event: str, *args, **kwargs) -> Callable[..., Any] | None:
-        return self._events[event](*args, **kwargs)
+    def get_event(self, event: str) -> Callable[..., Any]:
+        return self._events[event]
 
     async def _await_maybe(self, result: Any) -> Any:
         if asyncio.iscoroutine(result):
@@ -33,7 +33,7 @@ async def _await_maybe(self, result: Any) -> Any:
 
     async def trigger_event(self, event: str, *args, **kwargs) -> Any:
         try:
-            result = self._call_event(event, *args, **kwargs)
+            result = self.get_event(event)(*args, **kwargs)
             return await self._await_maybe(result)
         except Exception as e:
             handler = self._handlers.get(type(e), None)