openfeature.py 1.1 KB

12345678910111213141516171819202122232425262728293031323334
  1. from typing import TYPE_CHECKING, Any
  2. from sentry_sdk.feature_flags import add_feature_flag
  3. from sentry_sdk.integrations import DidNotEnable, Integration
  4. try:
  5. from openfeature import api
  6. from openfeature.hook import Hook
  7. if TYPE_CHECKING:
  8. from openfeature.hook import HookContext, HookHints
  9. except ImportError:
  10. raise DidNotEnable("OpenFeature is not installed")
  11. class OpenFeatureIntegration(Integration):
  12. identifier = "openfeature"
  13. @staticmethod
  14. def setup_once() -> None:
  15. # Register the hook within the global openfeature hooks list.
  16. api.add_hooks(hooks=[OpenFeatureHook()])
  17. class OpenFeatureHook(Hook):
  18. def after(self, hook_context: "Any", details: "Any", hints: "Any") -> None:
  19. if isinstance(details.value, bool):
  20. add_feature_flag(details.flag_key, details.value)
  21. def error(
  22. self, hook_context: "HookContext", exception: Exception, hints: "HookHints"
  23. ) -> None:
  24. if isinstance(hook_context.default_value, bool):
  25. add_feature_flag(hook_context.flag_key, hook_context.default_value)