From 17c6b9e31f9910c30ac660476e7bed8103f3935e Mon Sep 17 00:00:00 2001 From: Devaev Maxim Date: Fri, 16 Nov 2018 06:54:27 +0300 Subject: meta, refactoring --- kvmd/yaml.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 kvmd/yaml.py (limited to 'kvmd/yaml.py') diff --git a/kvmd/yaml.py b/kvmd/yaml.py new file mode 100644 index 00000000..cd7ae4fd --- /dev/null +++ b/kvmd/yaml.py @@ -0,0 +1,31 @@ +import os + +from typing import IO +from typing import Any + +import yaml +import yaml.loader +import yaml.nodes + + +# ===== +def load_yaml_file(path: str) -> Any: + with open(path) as yaml_file: + try: + return yaml.load(yaml_file, _YamlLoader) + except Exception: + # Reraise internal exception as standard ValueError and show the incorrect file + raise ValueError("Incorrect YAML syntax in file '{}'".format(path)) + + +class _YamlLoader(yaml.loader.Loader): # pylint: disable=too-many-ancestors + def __init__(self, yaml_file: IO) -> None: + yaml.loader.Loader.__init__(self, yaml_file) + self.__root = os.path.dirname(yaml_file.name) + + def include(self, node: yaml.nodes.Node) -> str: + path = os.path.join(self.__root, self.construct_scalar(node)) # pylint: disable=no-member + return load_yaml_file(path) + + +_YamlLoader.add_constructor("!include", _YamlLoader.include) # pylint: disable=no-member -- cgit v1.2.3