mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2025-12-12 01:00:29 +08:00
refactoring
This commit is contained in:
parent
db3f622023
commit
a74044b086
@ -179,7 +179,7 @@ def _init_config(config_path: str, override_options: list[str], **load_flags: bo
|
|||||||
scheme = _get_config_scheme()
|
scheme = _get_config_scheme()
|
||||||
try:
|
try:
|
||||||
yaml_merge(raw_config, (raw_config.pop("override", {}) or {}))
|
yaml_merge(raw_config, (raw_config.pop("override", {}) or {}))
|
||||||
yaml_merge(raw_config, build_raw_from_options(override_options), "raw command line options")
|
yaml_merge(raw_config, build_raw_from_options(override_options), "raw CLI options")
|
||||||
_patch_raw(raw_config)
|
_patch_raw(raw_config)
|
||||||
config = make_config(raw_config, scheme)
|
config = make_config(raw_config, scheme)
|
||||||
|
|
||||||
|
|||||||
@ -19,10 +19,25 @@
|
|||||||
# #
|
# #
|
||||||
# ========================================================================== #
|
# ========================================================================== #
|
||||||
|
|
||||||
from typing import Optional
|
|
||||||
|
|
||||||
|
|
||||||
# =====
|
# =====
|
||||||
|
def yaml_merge(dest: dict, src: dict, src_name: str="") -> None:
|
||||||
|
""" Merges the source dictionary into the destination dictionary. """
|
||||||
|
|
||||||
|
# Checking if destination is None
|
||||||
|
if dest is None:
|
||||||
|
# We can't merge into a None
|
||||||
|
raise ValueError(f"Could not merge {src_name or 'config'} into None. The destination cannot be None")
|
||||||
|
|
||||||
|
# Checking if source is None or empty
|
||||||
|
if not src:
|
||||||
|
# If src is None or empty, there's nothing to merge
|
||||||
|
return
|
||||||
|
|
||||||
|
_merge(dest, src)
|
||||||
|
|
||||||
|
|
||||||
|
# ======
|
||||||
def _merge(dest: dict, src: dict) -> None:
|
def _merge(dest: dict, src: dict) -> None:
|
||||||
for key in src:
|
for key in src:
|
||||||
if key in dest:
|
if key in dest:
|
||||||
@ -30,19 +45,3 @@ def _merge(dest: dict, src: dict) -> None:
|
|||||||
_merge(dest[key], src[key])
|
_merge(dest[key], src[key])
|
||||||
continue
|
continue
|
||||||
dest[key] = src[key]
|
dest[key] = src[key]
|
||||||
|
|
||||||
|
|
||||||
def yaml_merge(dest: dict, src: dict, source_name: Optional[str]=None) -> None:
|
|
||||||
""" Merges the source dictionary into the destination dictionary. """
|
|
||||||
|
|
||||||
# Checking if destination is None
|
|
||||||
if dest is None:
|
|
||||||
# We can't merge into a None
|
|
||||||
raise ValueError(f"Could not merge {source_name} into None. The destination cannot be None")
|
|
||||||
|
|
||||||
# Checking if source is None or empty
|
|
||||||
if src is None:
|
|
||||||
# If src is None or empty, there's nothing to merge
|
|
||||||
return
|
|
||||||
|
|
||||||
_merge(dest, src)
|
|
||||||
|
|||||||
@ -21,6 +21,7 @@
|
|||||||
|
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from kvmd.yamlconf import merger
|
from kvmd.yamlconf import merger
|
||||||
|
|
||||||
|
|
||||||
@ -108,7 +109,7 @@ def test_non_dict_values_in_source() -> None:
|
|||||||
|
|
||||||
|
|
||||||
def test_empty_base() -> None:
|
def test_empty_base() -> None:
|
||||||
base = {}
|
base: dict = {}
|
||||||
incoming = {"key1": "value1"}
|
incoming = {"key1": "value1"}
|
||||||
merger.yaml_merge(base, incoming)
|
merger.yaml_merge(base, incoming)
|
||||||
assert base == {"key1": "value1"}
|
assert base == {"key1": "value1"}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user