Na}{

File Content: `sys_exit.py`

import functools
import sys

import sentry_sdk
from sentry_sdk.utils import capture_internal_exceptions, event_from_exception
from sentry_sdk.integrations import Integration
from sentry_sdk._types import TYPE_CHECKING

if TYPE_CHECKING:
    from collections.abc import Callable
    from typing import NoReturn, Union


class SysExitIntegration(Integration):
    """Captures sys.exit calls and sends them as events to Sentry.

    By default, SystemExit exceptions are not captured by the SDK. Enabling this integration will capture SystemExit
    exceptions generated by sys.exit calls and send them to Sentry.

    This integration, in its default configuration, only captures the sys.exit call if the exit code is a non-zero and
    non-None value (unsuccessful exits). Pass `capture_successful_exits=True` to capture successful exits as well.
    Note that the integration does not capture SystemExit exceptions raised outside a call to sys.exit.
    """

    identifier = "sys_exit"

    def __init__(self, *, capture_successful_exits=False):
        # type: (bool) -> None
        self._capture_successful_exits = capture_successful_exits

    @staticmethod
    def setup_once():
        # type: () -> None
        SysExitIntegration._patch_sys_exit()

    @staticmethod
    def _patch_sys_exit():
        # type: () -> None
        old_exit = sys.exit  # type: Callable[[Union[str, int, None]], NoReturn]

        @functools.wraps(old_exit)
        def sentry_patched_exit(__status=0):
            # type: (Union[str, int, None]) -> NoReturn
            # @ensure_integration_enabled ensures that this is non-None
            integration = sentry_sdk.get_client().get_integration(SysExitIntegration)
            if integration is None:
                old_exit(__status)

            try:
                old_exit(__status)
            except SystemExit as e:
                with capture_internal_exceptions():
                    if integration._capture_successful_exits or __status not in (
                        0,
                        None,
                    ):
                        _capture_exception(e)
                raise e

        sys.exit = sentry_patched_exit


def _capture_exception(exc):
    # type: (SystemExit) -> None
    event, hint = event_from_exception(
        exc,
        client_options=sentry_sdk.get_client().options,
        mechanism={"type": SysExitIntegration.identifier, "handled": False},
    )
    sentry_sdk.capture_event(event, hint=hint)

Name	Size	Permission
__pycache__	---	0755
celery	---	0755
django	---	0755
google_genai	---	0755
grpc	---	0755
openai_agents	---	0755
opentelemetry	---	0755
pydantic_ai	---	0755
redis	---	0755
spark	---	0755
__init__.py	12735 bytes	0644
_asgi_common.py	3187 bytes	0644
_wsgi_common.py	7558 bytes	0644
aiohttp.py	13007 bytes	0644
anthropic.py	14512 bytes	0644
argv.py	911 bytes	0644
ariadne.py	5834 bytes	0644
arq.py	7881 bytes	0644
asgi.py	12800 bytes	0644
asyncio.py	4610 bytes	0644
asyncpg.py	6521 bytes	0644
atexit.py	1652 bytes	0644
aws_lambda.py	18071 bytes	0644
beam.py	5182 bytes	0644
boto3.py	4411 bytes	0644
bottle.py	6615 bytes	0644
chalice.py	4699 bytes	0644
clickhouse_driver.py	6098 bytes	0644
cloud_resource_context.py	7780 bytes	0644
cohere.py	9401 bytes	0644
dedupe.py	1975 bytes	0644
dramatiq.py	7455 bytes	0644
excepthook.py	2408 bytes	0644
executing.py	1994 bytes	0644
falcon.py	9501 bytes	0644
fastapi.py	4589 bytes	0644
flask.py	8740 bytes	0644
gcp.py	8462 bytes	0644
gnu_backtrace.py	2812 bytes	0644
gql.py	4801 bytes	0644
graphene.py	5042 bytes	0644
httpx.py	6086 bytes	0644
huey.py	5443 bytes	0644
huggingface_hub.py	14952 bytes	0644
langchain.py	39957 bytes	0644
langgraph.py	11835 bytes	0644
launchdarkly.py	1934 bytes	0644
litellm.py	10228 bytes	0644
litestar.py	11830 bytes	0644
logging.py	13892 bytes	0644
loguru.py	6534 bytes	0644
mcp.py	20052 bytes	0644
modules.py	820 bytes	0644
openai.py	25131 bytes	0644
openfeature.py	1131 bytes	0644
otlp.py	3129 bytes	0644
pure_eval.py	4605 bytes	0644
pymongo.py	6380 bytes	0644
pyramid.py	7364 bytes	0644
quart.py	7413 bytes	0644
ray.py	5323 bytes	0644
rq.py	5307 bytes	0644
rust_tracing.py	9078 bytes	0644
sanic.py	12960 bytes	0644
serverless.py	1804 bytes	0644
socket.py	3169 bytes	0644
sqlalchemy.py	4344 bytes	0644
starlette.py	26238 bytes	0644
starlite.py	10559 bytes	0644
statsig.py	1227 bytes	0644
stdlib.py	8968 bytes	0644
strawberry.py	14155 bytes	0644
sys_exit.py	2493 bytes	0644
threading.py	7109 bytes	0644
tornado.py	7206 bytes	0644
trytond.py	1651 bytes	0644
typer.py	1815 bytes	0644
unleash.py	1058 bytes	0644
unraisablehook.py	1753 bytes	0644
wsgi.py	10809 bytes	0644

File Content: sys_exit.py

File Content: `sys_exit.py`