openfeature.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435
  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():
  15. # type: () -> None
  16. # Register the hook within the global openfeature hooks list.
  17. api.add_hooks(hooks=[OpenFeatureHook()])
  18. class OpenFeatureHook(Hook):
  19. def after(self, hook_context, details, hints):
  20. # type: (Any, Any, Any) -> None
  21. if isinstance(details.value, bool):
  22. add_feature_flag(details.flag_key, details.value)
  23. def error(self, hook_context, exception, hints):
  24. # type: (HookContext, Exception, HookHints) -> None
  25. if isinstance(hook_context.default_value, bool):
  26. add_feature_flag(hook_context.flag_key, hook_context.default_value)