trustchain-oc1-orchestral/authentik
Archived
10
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

root: fix error_handler for websocket

Browse source 
Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
Jens Langhammer 2021-08-23 17:12:11 +02:00
parent 73c55b56a0
commit 4308136108
1 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
authentik/root/asgi/error_handler.py
View file

@ -19,14 +19,21 @@ class ASGIErrorHandler:
return await self.app(scope, receive, send) return await self.app(scope, receive, send)
except Exception as exc: # pylint: disable=broad-except except Exception as exc: # pylint: disable=broad-except
LOGGER.warning("Fatal ASGI exception", exc=exc) LOGGER.warning("Fatal ASGI exception", exc=exc)
return await self.error_handler(send) return await self.error_handler(scope, send)
async def error_handler(self, send: Send) -> None: async def error_handler(self, scope: Scope, send: Send) -> None:
"""Return a generic error message""" """Return a generic error message"""
if scope.get("scheme", "http") == "http":
return await send(
{
"type": "http.request",
"body": b"Internal server error",
"more_body": False,
}
)
return await send( return await send(
{ {
"type": "http.request", "type": "websocket.send",
"body": b"Internal server error", "text": b"Internal server error",
"more_body": False,
} }
) )