58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
from __future__ import absolute_import
|
|
|
|
import os
|
|
import sys
|
|
import atexit
|
|
|
|
from sentry_sdk.hub import Hub
|
|
from sentry_sdk.utils import logger
|
|
from sentry_sdk.integrations import Integration
|
|
|
|
from sentry_sdk._types import MYPY
|
|
|
|
if MYPY:
|
|
|
|
from typing import Any
|
|
from typing import Optional
|
|
|
|
|
|
def default_callback(pending, timeout):
|
|
"""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):
|
|
# type: (str) -> None
|
|
sys.stderr.write(msg + "\n")
|
|
|
|
echo("Sentry is attempting to send %i pending error messages" % 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=None):
|
|
# type: (Optional[Any]) -> None
|
|
if callback is None:
|
|
callback = default_callback
|
|
self.callback = callback
|
|
|
|
@staticmethod
|
|
def setup_once():
|
|
# type: () -> None
|
|
@atexit.register
|
|
def _shutdown():
|
|
logger.debug("atexit: got shutdown signal")
|
|
hub = Hub.main
|
|
integration = hub.get_integration(AtexitIntegration)
|
|
if integration is not None:
|
|
logger.debug("atexit: shutting down client")
|
|
|
|
# If an integration is there, a client has to be there.
|
|
client = hub.client # type: Any
|
|
client.close(callback=integration.callback)
|