refactor(api): delegate module-level functions to default API instance

Author: marcozabel

openfeature/api.py

@@ -1,4 +1,4 @@
-from openfeature import _event_support
+from openfeature._api import _default_api
 from openfeature.client import OpenFeatureClient
 from openfeature.evaluation_context import (
     get_evaluation_context,
@@ -10,7 +10,6 @@
 )
 from openfeature.hook import add_hooks, clear_hooks, get_hooks
 from openfeature.provider import FeatureProvider
-from openfeature.provider._registry import provider_registry
 from openfeature.provider.metadata import Metadata
 from openfeature.transaction_context import (
     get_transaction_context,
@@ -40,32 +39,28 @@
 def get_client(
     domain: str | None = None, version: str | None = None
 ) -> OpenFeatureClient:
-    return OpenFeatureClient(domain=domain, version=version)
+    return _default_api.get_client(domain=domain, version=version)
 
 
 def set_provider(provider: FeatureProvider, domain: str | None = None) -> None:
-    if domain is None:
-        provider_registry.set_default_provider(provider)
-    else:
-        provider_registry.set_provider(domain, provider)
+    _default_api.set_provider(provider, domain)
 
 
 def clear_providers() -> None:
-    provider_registry.clear_providers()
-    _event_support.clear()
+    _default_api.clear_providers()
 
 
 def get_provider_metadata(domain: str | None = None) -> Metadata:
-    return provider_registry.get_provider(domain).get_metadata()
+    return _default_api.get_provider_metadata(domain)
 
 
 def shutdown() -> None:
-    provider_registry.shutdown()
+    _default_api.shutdown()
 
 
 def add_handler(event: ProviderEvent, handler: EventHandler) -> None:
-    _event_support.add_global_handler(event, handler)
+    _default_api.add_handler(event, handler)
 
 
 def remove_handler(event: ProviderEvent, handler: EventHandler) -> None:
-    _event_support.remove_global_handler(event, handler)
+    _default_api.remove_handler(event, handler)

openfeature/evaluation_context/__init__.py

@@ -5,8 +5,6 @@
 from dataclasses import dataclass, field
 from datetime import datetime
 
-from openfeature.exception import GeneralError
-
 __all__ = ["EvaluationContext", "get_evaluation_context", "set_evaluation_context"]
 
 # https://openfeature.dev/specification/sections/evaluation-context#requirement-312
@@ -37,15 +35,16 @@ def merge(self, ctx2: EvaluationContext) -> EvaluationContext:
 
 
 def get_evaluation_context() -> EvaluationContext:
-    return _evaluation_context
+    from openfeature._api import _default_api  # noqa: PLC0415
+
+    return _default_api.get_evaluation_context()
 
 
 def set_evaluation_context(evaluation_context: EvaluationContext) -> None:
-    global _evaluation_context
-    if evaluation_context is None:
-        raise GeneralError(error_message="No api level evaluation context")
-    _evaluation_context = evaluation_context
+    from openfeature._api import _default_api  # noqa: PLC0415
+
+    _default_api.set_evaluation_context(evaluation_context)
 
 
-# need to be at the bottom, because of the definition order
+# Kept for backward compatibility but no longer used; state lives in _default_api.
 _evaluation_context = EvaluationContext()

openfeature/hook/__init__.py

@@ -23,8 +23,6 @@
     "get_hooks",
 ]
 
-_hooks: list[Hook] = []
-
 
 # https://openfeature.dev/specification/sections/hooks/#requirement-461
 HookData = MutableMapping[str, typing.Any]
@@ -152,14 +150,18 @@ def supports_flag_value_type(self, flag_type: FlagType) -> bool:
 
 
 def add_hooks(hooks: list[Hook]) -> None:
-    global _hooks
-    _hooks = _hooks + hooks
+    from openfeature._api import _default_api  # noqa: PLC0415
+
+    _default_api.add_hooks(hooks)
 
 
 def clear_hooks() -> None:
-    global _hooks
-    _hooks = []
+    from openfeature._api import _default_api  # noqa: PLC0415
+
+    _default_api.clear_hooks()
 
 
 def get_hooks() -> list[Hook]:
-    return _hooks
+    from openfeature._api import _default_api  # noqa: PLC0415
+
+    return _default_api.get_hooks()

openfeature/transaction_context/__init__.py

@@ -11,30 +11,29 @@
 
 __all__ = [
     "ContextVarsTransactionContextPropagator",
+    "NoOpTransactionContextPropagator",
     "TransactionContextPropagator",
     "get_transaction_context",
     "set_transaction_context",
     "set_transaction_context_propagator",
 ]
 
-_evaluation_transaction_context_propagator: TransactionContextPropagator = (
-    NoOpTransactionContextPropagator()
-)
-
 
 def set_transaction_context_propagator(
     transaction_context_propagator: TransactionContextPropagator,
 ) -> None:
-    global _evaluation_transaction_context_propagator
-    _evaluation_transaction_context_propagator = transaction_context_propagator
+    from openfeature._api import _default_api  # noqa: PLC0415
+
+    _default_api.set_transaction_context_propagator(transaction_context_propagator)
 
 
 def get_transaction_context() -> EvaluationContext:
-    return _evaluation_transaction_context_propagator.get_transaction_context()
+    from openfeature._api import _default_api  # noqa: PLC0415
+
+    return _default_api.get_transaction_context()
 
 
 def set_transaction_context(evaluation_context: EvaluationContext) -> None:
-    global _evaluation_transaction_context_propagator
-    _evaluation_transaction_context_propagator.set_transaction_context(
-        evaluation_context
-    )
+    from openfeature._api import _default_api  # noqa: PLC0415
+
+    _default_api.set_transaction_context(evaluation_context)