forked from getsentry/sentry-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtrytond.py
More file actions
52 lines (42 loc) · 1.67 KB
/
Copy pathtrytond.py
File metadata and controls
52 lines (42 loc) · 1.67 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
import sentry_sdk
from sentry_sdk.integrations import DidNotEnable, Integration
from sentry_sdk.integrations.wsgi import SentryWsgiMiddleware
from sentry_sdk.utils import ensure_integration_enabled, event_from_exception
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)