platypush/platypush
platypush/platypush
2
2
Fork 4
Code Issues 30 Pull requests Projects 4 Releases Packages Wiki Activity

Added _on_error handler to StreamingRoute.

Browse source
This commit is contained in:
Fabio Manganiello 2023-11-05 01:03:40 +01:00
parent 7c780e6650
commit 0e2738d849
Signed by: blacklight
GPG key ID: D90FBA7F76362774
1 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
platypush/backend/http/app/streaming/_base.py
View file

@ -6,6 +6,7 @@ from typing import Optional
from tornado.web import RequestHandler, stream_request_body
from platypush.backend.http.app.utils import logger
from platypush.backend.http.app.utils.auth import AuthStatus, get_auth_status
from ..mixins import PubSubMixin
@ -121,3 +122,20 @@ class StreamingRoute(RequestHandler, PubSubMixin, ABC):
return self.request.connection.stream.closed() # type: ignore
return True
def _on_error(self, e: Exception):
"""
Handler for uncaught exceptions.
"""
if isinstance(e, FileNotFoundError):
self.set_status(404, str(e))
elif isinstance(e, NotImplementedError):
self.set_status(422, str(e))
else:
logger().exception(e)
if isinstance(e, AssertionError):
self.set_status(400, str(e))
else:
self.set_status(500, str(e))
self.finish(json.dumps({"error": str(e)}))