Merge environment variables with config file recursively

src/config/bot_config.py

@@ -3,6 +3,7 @@
 
 import contextlib
 import os
+from collections import defaultdict
 from typing import Any
 
 import orjson
@@ -16,17 +17,12 @@
 
 def load_from_env() -> dict[str, dict[str, Any]]:
     _config: dict[str, Any] = {}
-    values = {k: v for k, v in os.environ.items() if k.startswith(f"BOTKIT{SPLIT}")}
-    values = {k[len(f"BOTKIT{SPLIT}") :]: v for k, v in values.items()}
-    current: dict[str, Any] = {}
+    values = {k: v for k, v in os.environ.items() if k.startswith("BOTKIT__")}
     for key, value in values.items():
-        for i, part in enumerate(key.split(SPLIT)):
-            part = part.lower()  # noqa: PLW2901
-            if i == 0:
-                if part not in _config:
-                    _config[part] = {}
-                current = _config[part]
-            elif i == len(key.split(SPLIT)) - 1:
+        parts = key[len("BOTKIT__") :].lower().split("__")
+        current = _config
+        for i, part in enumerate(parts):
+            if i == len(parts) - 1:
                 current[part] = value
             else:
                 if part not in current:
@@ -47,6 +43,9 @@ def load_json_recursive(data: dict[str, Any]) -> dict[str, Any]:
                 data[key] = True
             elif value.lower() == "false":
                 data[key] = False
+            elif value.startswith("0x"):
+                with contextlib.suppress(ValueError):
+                    data[key] = int(value, 16)
             else:
                 with contextlib.suppress(orjson.JSONDecodeError):
                     data[key] = orjson.loads(value)
@@ -59,10 +58,18 @@ def load_json_recursive(data: dict[str, Any]) -> dict[str, Any]:
 elif os.path.exists("config.yml"):
     path = "config.yml"
 
-config: dict[str, dict[str, Any]]
+
+def merge_dicts(dct: dict[str, Any], merge_dct: dict[str, Any]) -> None:
+    for k, v in merge_dct.items():
+        if isinstance(dct.get(k), dict) and isinstance(v, dict):
+            merge_dicts(dct[k], v)
+        else:
+            dct[k] = v
+
+
+config: dict[str, dict[str, Any]] = defaultdict(dict)
 if path:
-    # noinspection PyArgumentEqualDefault
     with open(path, encoding="utf-8") as f:
-        config = yaml.safe_load(f)
-else:
-    config = load_from_env()
+        config.update(yaml.safe_load(f))
+
+merge_dicts(config, load_from_env())

tests/config_test.py

@@ -0,0 +1,115 @@
+# Copyright (c) NiceBots.xyz
+# SPDX-License-Identifier: MIT
+
+# ruff: noqa: S101, S105
+
+import os
+from typing import Any
+
+import yaml
+
+from src.config.bot_config import load_from_env, merge_dicts
+
+
+def test_merge_dicts_basic() -> None:
+    """Test basic dictionary merging."""
+    base = {"a": 1, "b": 2}
+    update = {"b": 3, "c": 4}
+    merge_dicts(base, update)
+    assert base == {"a": 1, "b": 3, "c": 4}
+
+
+def test_merge_dicts_nested() -> None:
+    """Test nested dictionary merging."""
+    base = {"a": {"x": 1, "y": 2}, "b": 3}
+    update = {"a": {"y": 4, "z": 5}, "c": 6}
+    merge_dicts(base, update)
+    assert base == {"a": {"x": 1, "y": 4, "z": 5}, "b": 3, "c": 6}
+
+
+def test_merge_dicts_deep_nested() -> None:
+    """Test deeply nested dictionary merging."""
+    base = {"a": {"x": {"p": 1, "q": 2}, "y": 3}}
+    update = {"a": {"x": {"q": 4, "r": 5}}}
+    merge_dicts(base, update)
+    assert base == {"a": {"x": {"p": 1, "q": 4, "r": 5}, "y": 3}}
+
+
+def test_merge_dicts_with_none() -> None:
+    """Test merging when values are None."""
+    base = {"a": {"x": 1}, "b": None}
+    update = {"a": {"y": 2}, "b": {"z": 3}}
+    merge_dicts(base, update)
+    assert base == {"a": {"x": 1, "y": 2}, "b": {"z": 3}}
+
+
+def test_load_from_env() -> None:
+    """Test loading configuration from environment variables."""
+    # Set up test environment variables
+    test_env = {
+        "BOTKIT__TOKEN": "test-token",
+        "BOTKIT__EXTENSIONS__PING__ENABLED": "true",
+        "BOTKIT__EXTENSIONS__PING__COLOR": "0xFF0000",
+        "BOTKIT__EXTENSIONS__TOPGG__TOKEN": "test-topgg-token",
+    }
+
+    for key, value in test_env.items():
+        os.environ[key] = value
+
+    try:
+        config = load_from_env()
+
+        # Check the loaded configuration
+        assert config["token"] == "test-token"
+        assert config["extensions"]["ping"]["enabled"] is True
+        assert config["extensions"]["ping"]["color"] == 0xFF0000
+        assert config["extensions"]["topgg"]["token"] == "test-topgg-token"
+
+    finally:
+        # Clean up environment variables
+        for key in test_env:
+            os.environ.pop(key, None)
+
+
+def test_config_file_and_env_integration(tmp_path: Any) -> None:
+    """Test integration of config file and environment variables."""
+    # Create a temporary config file
+    config_file = tmp_path / "config.yaml"
+    config_data = {
+        "token": "file-token",
+        "extensions": {"ping": {"enabled": False, "color": 0x00FF00, "message": "Pong!"}, "topgg": {"enabled": True}},
+    }
+
+    with open(config_file, "w", encoding="utf-8") as f:
+        yaml.dump(config_data, f)
+
+    # Set environment variables that should override some values
+    test_env = {
+        "BOTKIT__TOKEN": "env-token",
+        "BOTKIT__EXTENSIONS__PING__ENABLED": "true",
+        "BOTKIT__EXTENSIONS__PING__COLOR": "0xFF0000",
+    }
+
+    for key, value in test_env.items():
+        os.environ[key] = value
+
+    try:
+        # Load config from file
+        with open(config_file, encoding="utf-8") as f:
+            config = yaml.safe_load(f)
+
+        # Merge with environment variables
+        env_config = load_from_env()
+        merge_dicts(config, env_config)
+
+        # Verify the merged configuration
+        assert config["token"] == "env-token"  # Overridden by env
+        assert config["extensions"]["ping"]["enabled"] is True  # Overridden by env
+        assert config["extensions"]["ping"]["color"] == 0xFF0000  # Overridden by env
+        assert config["extensions"]["ping"]["message"] == "Pong!"  # Kept from file
+        assert config["extensions"]["topgg"]["enabled"] is True  # Kept from file
+
+    finally:
+        # Clean up environment variables
+        for key in test_env:
+            os.environ.pop(key, None)