parse yes/no/on/off as string

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()