forked from getsentry/sentry-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathatexit.py
More file actions
51 lines (38 loc) · 1.51 KB
/
Copy pathatexit.py
File metadata and controls
51 lines (38 loc) · 1.51 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
import atexit
import os
import sys
from typing import TYPE_CHECKING
import sentry_sdk
from sentry_sdk.integrations import Integration
from sentry_sdk.utils import logger
if TYPE_CHECKING:
from typing import Any, Optional
def default_callback(pending: int, timeout: int) -> None:
"""This is the default shutdown callback that is set on the options.
It prints out a message to stderr that informs the user that some events
are still pending and the process is waiting for them to flush out.
"""
def echo(msg: str) -> None:
sys.stderr.write(msg + "\n")
echo("Sentry is attempting to send %i pending events" % pending)
echo("Waiting up to %s seconds" % timeout)
echo("Press Ctrl-%s to quit" % (os.name == "nt" and "Break" or "C"))
sys.stderr.flush()
class AtexitIntegration(Integration):
identifier = "atexit"
def __init__(self, callback: "Optional[Any]" = None) -> None:
if callback is None:
callback = default_callback
self.callback = callback
@staticmethod
def setup_once() -> None:
@atexit.register
def _shutdown() -> None:
client = sentry_sdk.get_client()
integration = client.get_integration(AtexitIntegration)
if integration is None:
return
logger.debug("atexit: got shutdown signal")
logger.debug("atexit: shutting down client")
sentry_sdk.get_isolation_scope().end_session()
client.close(callback=integration.callback)