fix!: restrict exported names with __all__

Signed-off-by: Federico Bond <federicobond@gmail.com>

openfeature/api.py

@@ -13,6 +13,21 @@
 from openfeature.provider._registry import provider_registry
 from openfeature.provider.metadata import Metadata
 
+__all__ = [
+    "get_client",
+    "set_provider",
+    "clear_providers",
+    "get_provider_metadata",
+    "get_evaluation_context",
+    "set_evaluation_context",
+    "add_hooks",
+    "clear_hooks",
+    "get_hooks",
+    "shutdown",
+    "add_handler",
+    "remove_handler",
+]
+
 _evaluation_context = EvaluationContext()
 
 _hooks: typing.List[Hook] = []

openfeature/client.py

@@ -30,6 +30,11 @@
 from openfeature.provider import FeatureProvider, ProviderStatus
 from openfeature.provider._registry import provider_registry
 
+__all__ = [
+    "ClientMetadata",
+    "OpenFeatureClient",
+]
+
 logger = logging.getLogger("openfeature")
 
 GetDetailCallable = typing.Union[

openfeature/evaluation_context.py

@@ -1,6 +1,8 @@
 import typing
 from dataclasses import dataclass, field
 
+__all__ = ["EvaluationContext"]
+
 
 @dataclass
 class EvaluationContext:

openfeature/event.py

@@ -7,6 +7,8 @@
 from openfeature.exception import ErrorCode
 from openfeature.provider import ProviderStatus
 
+__all__ = ["ProviderEvent", "ProviderEventDetails", "EventDetails", "EventHandler"]
+
 
 class ProviderEvent(Enum):
     PROVIDER_READY = "PROVIDER_READY"

openfeature/exception.py

@@ -4,6 +4,19 @@
 from collections.abc import Mapping
 from enum import Enum
 
+__all__ = [
+    "OpenFeatureError",
+    "ProviderNotReadyError",
+    "ProviderFatalError",
+    "FlagNotFoundError",
+    "GeneralError",
+    "ParseError",
+    "TypeMismatchError",
+    "TargetingKeyMissingError",
+    "InvalidContextError",
+    "ErrorCode",
+]
+
 
 class OpenFeatureError(Exception):
     """

openfeature/flag_evaluation.py

@@ -11,6 +11,16 @@
     from openfeature.hook import Hook, HookHints
 
 
+__all__ = [
+    "FlagType",
+    "Reason",
+    "FlagMetadata",
+    "FlagEvaluationDetails",
+    "FlagEvaluationOptions",
+    "FlagResolutionDetails",
+]
+
+
 class FlagType(StrEnum):
     BOOLEAN = "BOOLEAN"
     STRING = "STRING"

openfeature/hook/__init__.py

@@ -12,6 +12,8 @@
     from openfeature.client import ClientMetadata
     from openfeature.provider.metadata import Metadata
 
+__all__ = ["HookType", "HookContext", "HookHints", "Hook"]
+
 
 class HookType(Enum):
     BEFORE = "before"

openfeature/provider/__init__.py

@@ -7,6 +7,8 @@
 
 from .metadata import Metadata
 
+__all__ = ["ProviderStatus", "FeatureProvider", "Metadata"]
+
 
 class ProviderStatus(Enum):
     NOT_READY = "NOT_READY"

openfeature/provider/provider.py

@@ -8,6 +8,8 @@
 from openfeature.provider import FeatureProvider
 from openfeature.provider.metadata import Metadata
 
+__all__ = ["AbstractProvider"]
+
 
 class AbstractProvider(FeatureProvider):
     def initialize(self, evaluation_context: EvaluationContext) -> None: