| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253 |
- import sentry_sdk
- from sentry_sdk.integrations import Integration
- from sentry_sdk.integrations.wsgi import SentryWsgiMiddleware
- from sentry_sdk.utils import ensure_integration_enabled, event_from_exception
- from sentry_sdk.integrations import DidNotEnable
- try:
- from trytond.exceptions import TrytonException # type: ignore
- from trytond.wsgi import app # type: ignore
- except ImportError:
- raise DidNotEnable("Trytond is not installed.")
- # TODO: trytond-worker, trytond-cron and trytond-admin intergations
- class TrytondWSGIIntegration(Integration):
- identifier = "trytond_wsgi"
- origin = f"auto.http.{identifier}"
- def __init__(self) -> None:
- pass
- @staticmethod
- def setup_once() -> None:
- app.wsgi_app = SentryWsgiMiddleware(
- app.wsgi_app,
- span_origin=TrytondWSGIIntegration.origin,
- )
- @ensure_integration_enabled(TrytondWSGIIntegration)
- def error_handler(e: Exception) -> None:
- if isinstance(e, TrytonException):
- return
- else:
- client = sentry_sdk.get_client()
- event, hint = event_from_exception(
- e,
- client_options=client.options,
- mechanism={"type": "trytond", "handled": False},
- )
- sentry_sdk.capture_event(event, hint=hint)
- # Expected error handlers signature was changed
- # when the error_handler decorator was introduced
- # in Tryton-5.4
- if hasattr(app, "error_handler"):
- @app.error_handler
- def _(app, request, e): # type: ignore
- error_handler(e)
- else:
- app.error_handlers.append(error_handler)
|
