switch to using try/except

mobilityhouse · Dec 9, 2024 · 12b793c · 12b793c
1 parent db1a94f
commit 12b793c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ocpp/charge_point.py b/ocpp/charge_point.py
@@ -170,20 +170,20 @@ def _raise_key_error(action, version):
     from ocpp.v201.enums import Action as v201_Action
 
     if version == "1.6":
-        if action in list(v16_Action):
+        try v16_Action(action):
             raise NotImplementedError(
                 details={"cause": f"No handler for {action} registered."}
             )
-        else:
+        except ValueError:
             raise NotSupportedError(
                 details={"cause": f"{action} not supported by OCPP{version}."}
             )
     elif version in ["2.0", "2.0.1"]:
-        if action in list(v201_Action):
+        try v201_Action(action):
             raise NotImplementedError(
                 details={"cause": f"No handler for {action} registered."}
             )
-        else:
+        except ValueError:
             raise NotSupportedError(
                 details={"cause": f"{action} not supported by OCPP{version}."}
             )