| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061 |
- from typing import TYPE_CHECKING
- from sentry_sdk.feature_flags import add_feature_flag
- from sentry_sdk.integrations import DidNotEnable, Integration
- try:
- import ldclient
- from ldclient.hook import Hook, Metadata
- if TYPE_CHECKING:
- from ldclient import LDClient
- from ldclient.hook import EvaluationSeriesContext
- from ldclient.evaluation import EvaluationDetail
- from typing import Any
- except ImportError:
- raise DidNotEnable("LaunchDarkly is not installed")
- class LaunchDarklyIntegration(Integration):
- identifier = "launchdarkly"
- def __init__(self, ld_client=None):
- # type: (LDClient | None) -> None
- """
- :param client: An initialized LDClient instance. If a client is not provided, this
- integration will attempt to use the shared global instance.
- """
- try:
- client = ld_client or ldclient.get()
- except Exception as exc:
- raise DidNotEnable("Error getting LaunchDarkly client. " + repr(exc))
- if not client.is_initialized():
- raise DidNotEnable("LaunchDarkly client is not initialized.")
- # Register the flag collection hook with the LD client.
- client.add_hook(LaunchDarklyHook())
- @staticmethod
- def setup_once():
- # type: () -> None
- pass
- class LaunchDarklyHook(Hook):
- @property
- def metadata(self):
- # type: () -> Metadata
- return Metadata(name="sentry-flag-auditor")
- def after_evaluation(self, series_context, data, detail):
- # type: (EvaluationSeriesContext, dict[Any, Any], EvaluationDetail) -> dict[Any, Any]
- if isinstance(detail.value, bool):
- add_feature_flag(series_context.key, detail.value)
- return data
- def before_evaluation(self, series_context, data):
- # type: (EvaluationSeriesContext, dict[Any, Any]) -> dict[Any, Any]
- return data # No-op.
|
