Add SmartHomeError exception

DewGew · Dec 15, 2023 · 66091c5 · 66091c5
1 parent 9cd70bd
commit 66091c5
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/modules/helpers.py b/modules/helpers.py
@@ -124,3 +124,16 @@ def generateCsrfToken():
 
 def generate_API_key():
     return random_string(16)
+
+class SmartHomeError(Exception):
+    def __init__(self, code, msg):
+        """Log error code."""
+        super().__init__(msg)
+        self.code = code
+
+class SmartHomeErrorNoChallenge(Exception):
+    def __init__(self, code, desc, msg):
+        """Log error code."""
+        super().__init__(msg)
+        self.code = code
+        self.desc = desc