0UT1S's

opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/

By Order of Mr.0UT1S

N/A

[DIR] __pycache__

N/A

N/A

N/A

N/A

[DIR] opentelemetry

N/A

N/A

N/A

9.95 KB

_asgi_common.py

3.11 KB

_wsgi_common.py

7.38 KB

12.59 KB

9.21 KB

911 bytes

5.70 KB

7.67 KB

12.46 KB

4.65 KB

6.37 KB

1.61 KB

17.53 KB

5.06 KB

4.31 KB

6.46 KB

4.59 KB

clickhouse_driver.py

5.12 KB

cloud_resource_context.py

7.60 KB

9.05 KB

1.38 KB

5.45 KB

2.35 KB

1.95 KB

9.28 KB

4.61 KB

8.54 KB

8.08 KB

gnu_backtrace.py

2.83 KB

4.08 KB

4.92 KB

5.73 KB

5.32 KB

huggingface_hub.py

6.38 KB

17.30 KB

launchdarkly.py

1.90 KB

11.30 KB

13.09 KB

3.77 KB

820 bytes

15.19 KB

1.27 KB

4.47 KB

6.23 KB

7.19 KB

7.26 KB

4.06 KB

5.18 KB

rust_tracing.py

8.87 KB

12.66 KB

1.76 KB

3.09 KB

4.27 KB

25.69 KB

10.37 KB

1.20 KB

8.62 KB

13.79 KB

2.43 KB

3.92 KB

7.05 KB

1.61 KB

1.77 KB

1.05 KB

10.50 KB

import enum

from sentry_sdk.integrations import Integration, DidNotEnable
from sentry_sdk.integrations.logging import (
    BreadcrumbHandler,
    EventHandler,
    _BaseHandler,
)

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from logging import LogRecord
    from typing import Optional, Tuple, Any

try:
    import loguru
    from loguru import logger
    from loguru._defaults import LOGURU_FORMAT as DEFAULT_FORMAT
except ImportError:
    raise DidNotEnable("LOGURU is not installed")

class LoggingLevels(enum.IntEnum):
    TRACE = 5
    DEBUG = 10
    INFO = 20
    SUCCESS = 25
    WARNING = 30
    ERROR = 40
    CRITICAL = 50

SENTRY_LEVEL_FROM_LOGURU_LEVEL = {
    "TRACE": "DEBUG",
    "DEBUG": "DEBUG",
    "INFO": "INFO",
    "SUCCESS": "INFO",
    "WARNING": "WARNING",
    "ERROR": "ERROR",
    "CRITICAL": "CRITICAL",
}

DEFAULT_LEVEL = LoggingLevels.INFO.value
DEFAULT_EVENT_LEVEL = LoggingLevels.ERROR.value
# We need to save the handlers to be able to remove them later
# in tests (they call `LoguruIntegration.__init__` multiple times,
# and we can't use `setup_once` because it's called before
# than we get configuration).
_ADDED_HANDLERS = (None, None)  # type: Tuple[Optional[int], Optional[int]]

class LoguruIntegration(Integration):
    identifier = "loguru"

def __init__(
        self,
        level=DEFAULT_LEVEL,
        event_level=DEFAULT_EVENT_LEVEL,
        breadcrumb_format=DEFAULT_FORMAT,
        event_format=DEFAULT_FORMAT,
    ):
        # type: (Optional[int], Optional[int], str | loguru.FormatFunction, str | loguru.FormatFunction) -> None
        global _ADDED_HANDLERS
        breadcrumb_handler, event_handler = _ADDED_HANDLERS

if breadcrumb_handler is not None:
            logger.remove(breadcrumb_handler)
            breadcrumb_handler = None
        if event_handler is not None:
            logger.remove(event_handler)
            event_handler = None

if level is not None:
            breadcrumb_handler = logger.add(
                LoguruBreadcrumbHandler(level=level),
                level=level,
                format=breadcrumb_format,
            )

if event_level is not None:
            event_handler = logger.add(
                LoguruEventHandler(level=event_level),
                level=event_level,
                format=event_format,
            )

_ADDED_HANDLERS = (breadcrumb_handler, event_handler)

@staticmethod
    def setup_once():
        # type: () -> None
        pass  # we do everything in __init__

class _LoguruBaseHandler(_BaseHandler):
    def _logging_to_event_level(self, record):
        # type: (LogRecord) -> str
        try:
            return SENTRY_LEVEL_FROM_LOGURU_LEVEL[
                LoggingLevels(record.levelno).name
            ].lower()
        except (ValueError, KeyError):
            return record.levelname.lower() if record.levelname else ""

class LoguruEventHandler(_LoguruBaseHandler, EventHandler):
    """Modified version of :class:`sentry_sdk.integrations.logging.EventHandler` to use loguru's level names."""

def __init__(self, *args, **kwargs):
        # type: (*Any, **Any) -> None
        if kwargs.get("level"):
            kwargs["level"] = SENTRY_LEVEL_FROM_LOGURU_LEVEL.get(
                kwargs.get("level", ""), DEFAULT_LEVEL
            )

super().__init__(*args, **kwargs)

class LoguruBreadcrumbHandler(_LoguruBaseHandler, BreadcrumbHandler):
    """Modified version of :class:`sentry_sdk.integrations.logging.BreadcrumbHandler` to use loguru's level names."""

def __init__(self, *args, **kwargs):
        # type: (*Any, **Any) -> None
        if kwargs.get("level"):
            kwargs["level"] = SENTRY_LEVEL_FROM_LOGURU_LEVEL.get(
                kwargs.get("level", ""), DEFAULT_LEVEL
            )

super().__init__(*args, **kwargs)