Handle missing endpoint in view_class_from_endpoint

Joel Collins · Joel Collins · commit 841d3924aa7e · 2020-04-15T11:36:06.000+01:00
diff --git a/labthings/server/utilities.py b/labthings/server/utilities.py
@@ -55,7 +55,7 @@ def view_class_from_endpoint(endpoint: str):
     Returns:
         View: View class attached to the specified endpoint
     """
-    return current_app.view_functions[endpoint].view_class
+    return getattr(current_app.view_functions.get(endpoint), "view_class", None)
 
 
 def unpack(value):
