nodetool-core 0.7.0__py3-none-any.whl

nodetool/config/configuration.py

@@ -0,0 +1,82 @@
+from dataclasses import dataclass
+
+
+@dataclass
+class Setting:
+    package_name: str
+    env_var: str
+    group: str
+    description: str
+    enum: list[str] | None
+
+
+@dataclass
+class Secret:
+    package_name: str
+    env_var: str
+    group: str
+    description: str
+
+
+_registry: list[Setting] = []
+_secrets_registry: list[Secret] = []
+
+
+def register_secret(
+    package_name: str,
+    env_var: str,
+    group: str,
+    description: str,
+) -> list[Secret]:
+    """Register a new secret."""
+    secret = Secret(package_name=package_name, env_var=env_var, group=group, description=description)
+    _secrets_registry.append(secret)
+    return list(_secrets_registry)
+
+
+def register_setting(
+    package_name: str,
+    env_var: str,
+    group: str,
+    description: str,
+    enum: list[str] | None = None,
+) -> list[Setting]:
+    """Register a new setting.
+
+    Parameters
+    ----------
+    package_name: str
+        Name of the package registering the setting.
+    env_var: str
+        The environment variable name.
+    group: str
+        Group the setting belongs to.
+    description: str
+        Human readable description of the setting.
+    enum: List[str] | None
+        List of possible values for the setting.
+
+    Returns
+    -------
+    List[Setting]
+        The list of all registered settings.
+    """
+    setting = Setting(
+        package_name=package_name,
+        env_var=env_var,
+        group=group,
+        description=description,
+        enum=enum,
+    )
+    _registry.append(setting)
+    return list(_registry)
+
+
+def get_settings_registry() -> list[Setting]:
+    """Return the list of all registered settings."""
+    return list(_registry)
+
+
+def get_secrets_registry() -> list[Secret]:
+    """Return the list of all registered secrets."""
+    return list(_secrets_registry)

nodetool/config/env_guard.py

@@ -0,0 +1,34 @@
+import os
+import sys
+from typing import Any
+
+
+def _is_running_under_pytest() -> bool:
+    """Detect pytest presence from command-line arguments or environment.
+
+    This checks both sys.argv (for main pytest process) and PYTEST_CURRENT_TEST
+    environment variable (which is set in all pytest workers, including xdist workers).
+    """
+    # Check if PYTEST_CURRENT_TEST is set (works for all pytest workers)
+    if "PYTEST_CURRENT_TEST" in os.environ:
+        return True
+    # Fallback to checking sys.argv for main process
+    return any(arg and "pytest" in str(arg).lower() for arg in sys.argv)
+
+
+RUNNING_PYTEST = _is_running_under_pytest()
+
+
+def get_system_env_value(key: str, default: Any = None) -> Any:
+    """Return an environment variable value.
+
+    Tests may monkeypatch os.environ to drive configuration, so we always
+    read from the current process environment instead of short-circuiting
+    when pytest is detected.
+    """
+    return os.environ.get(key, default)
+
+
+def get_system_env() -> dict[str, str]:
+    """Return the current process environment."""
+    return dict(os.environ)