Method should always be available

Garulf · Jan 25, 2024 · fa85194 · fa85194
1 parent 95b9f1b
commit fa85194
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pyflowlauncher/event.py b/pyflowlauncher/event.py
@@ -22,13 +22,13 @@ def add_methods(self, methods: Iterable[Callable[..., Any]]):
             self.add_method(method)
 
     def add_exception_handler(self, exception: Exception, handler: Callable[..., Any]):
-        self._handlers[exception.__class__.__name__] = handler
+        self._handlers[exception] = handler
 
     def __call__(self, method: str, *args, **kwargs):
         try:
             return self._methods[method](*args, **kwargs)
         except Exception as e:
-            handler = self._handlers.get(e.__class__.__name__, None)
+            handler = self._handlers.get(e, None)
             if handler:
                 return handler(e)
-            raise e
+            # raise e
diff --git a/pyflowlauncher/plugin.py b/pyflowlauncher/plugin.py
@@ -62,7 +62,7 @@ def settings(self) -> dict:
 
     def run(self) -> None:
         request = self._client.recieve()
-        method = request.get('method')
+        method = request["method"]
         parameters = request.get('parameters', [])
         feedback = self._event_handler(method, *parameters)
         # Inject settings if changed