flows: fix error when stage has incorrect type

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
2021-06-16 22:52:00 +02:00 · 2021-06-16 22:52:00 +02:00 · b53c94d76a
parent d4419d66c1
commit b53c94d76a
1 changed files with 5 additions and 1 deletions
--- a/authentik/flows/views.py
+++ b/authentik/flows/views.py
@ -164,7 +164,11 @@ class FlowExecutorView(APIView):
            current_stage=self.current_stage,
            flow_slug=self.flow.slug,
        )
-        stage_cls = self.current_stage.type
+        try:
+            stage_cls = self.current_stage.type
+        except NotImplementedError as exc:
+            self._logger.debug("Error getting stage type", exc=exc)
+            return self.stage_invalid()
        self.current_stage_view = stage_cls(self)
        self.current_stage_view.args = self.args
        self.current_stage_view.kwargs = self.kwargs