57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
from __future__ import absolute_import
|
|
|
|
from sentry_sdk.hub import Hub
|
|
from sentry_sdk.integrations import Integration
|
|
from sentry_sdk.scope import add_global_event_processor
|
|
|
|
from sentry_sdk._types import MYPY
|
|
|
|
if MYPY:
|
|
from typing import Any
|
|
from typing import Dict
|
|
from typing import Tuple
|
|
from typing import Iterator
|
|
|
|
from sentry_sdk._types import Event
|
|
|
|
|
|
_installed_modules = None
|
|
|
|
|
|
def _generate_installed_modules():
|
|
# type: () -> Iterator[Tuple[str, str]]
|
|
try:
|
|
import pkg_resources
|
|
except ImportError:
|
|
return
|
|
|
|
for info in pkg_resources.working_set:
|
|
yield info.key, info.version
|
|
|
|
|
|
def _get_installed_modules():
|
|
# type: () -> Dict[str, str]
|
|
global _installed_modules
|
|
if _installed_modules is None:
|
|
_installed_modules = dict(_generate_installed_modules())
|
|
return _installed_modules
|
|
|
|
|
|
class ModulesIntegration(Integration):
|
|
identifier = "modules"
|
|
|
|
@staticmethod
|
|
def setup_once():
|
|
# type: () -> None
|
|
@add_global_event_processor
|
|
def processor(event, hint):
|
|
# type: (Event, Any) -> Dict[str, Any]
|
|
if event.get("type") == "transaction":
|
|
return event
|
|
|
|
if Hub.current.get_integration(ModulesIntegration) is None:
|
|
return event
|
|
|
|
event["modules"] = dict(_get_installed_modules())
|
|
return event
|