summaryrefslogtreecommitdiff
path: root/kvmd/yamlconf/loader.py
diff options
context:
space:
mode:
Diffstat (limited to 'kvmd/yamlconf/loader.py')
-rw-r--r--kvmd/yamlconf/loader.py20
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()