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

root: fix error_handler for websocket 

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)
except Exception as exc: # pylint: disable=broad-except
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"""
if scope.get("scheme", "http") == "http":
return await send(
{
"type": "http.request",
"body": b"Internal server error",
"more_body": False,
}
)
return await send(
{
"type": "http.request",
"body": b"Internal server error",
"more_body": False,
"type": "websocket.send",
"text": b"Internal server error",
}
)