diff options
Diffstat (limited to 'kvmd/yamlconf/loader.py')
-rw-r--r-- | kvmd/yamlconf/loader.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/kvmd/yamlconf/loader.py b/kvmd/yamlconf/loader.py index 9185d143..bc38d7c1 100644 --- a/kvmd/yamlconf/loader.py +++ b/kvmd/yamlconf/loader.py @@ -27,6 +27,8 @@ from typing import Any import yaml import yaml.nodes +import yaml.resolver +import yaml.constructor from .. import tools @@ -41,6 +43,7 @@ def load_yaml_file(path: str) -> Any: raise ValueError(f"Invalid YAML in the file {path!r}:\n{tools.efmt(err)}") from None +# ===== class _YamlLoader(yaml.SafeLoader): def __init__(self, yaml_file: IO) -> None: super().__init__(yaml_file) @@ -52,3 +55,20 @@ class _YamlLoader(yaml.SafeLoader): _YamlLoader.add_constructor("!include", _YamlLoader.include) + + +# ===== +def _disable_some_bools() -> None: + # https://stackoverflow.com/questions/36463531 + resolvers = yaml.resolver.Resolver.yaml_implicit_resolvers + for key in "oOyYnN": + resolvers[key] = [ + resolver + for resolver in resolvers[key] + if resolver[0] != "tag:yaml.org,2002:bool" + ] + if len(resolvers) == 0: + del resolvers[key] + + +_disable_some_bools() |