Python: Simplify error class

Signed-off-by: jwijenbergh <jeroenwijenbergh@protonmail.com>

1 files changed, 3 insertions, 6 deletions

diff --git a/wrappers/python/eduvpncommon/error.py b/wrappers/python/eduvpncommon/error.py
index 2ec82e7..ce0cec1 100644
--- a/wrappers/python/eduvpncommon/error.py
+++ b/wrappers/python/eduvpncommon/error.py
@@ -2,15 +2,12 @@ from enum import Enum
 
 class GoError(Exception):
     message_dict: dict
-    code: Enum | None
+    code: Enum
 
     def __init__(self, err: Enum, messages: dict):
         assert err
-        try:
-            self.code = err
-        except ValueError:
-            self.code = None
+        self.code = err
         self.message_dict = messages
 
     def __str__(self):
-        return self.message_dict[self.code] if self.code in self.message_dict else f"unknown error ({self.code})"
+        return self.message_dict[self.code]