GithubFollow/One-KVM
1
0
Fork 0
mirror of https://github.com/mofeng-git/One-KVM.git synced 2026-02-02 19:11:54 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

validators, tests

Browse Source
This commit is contained in:
Devaev Maxim
2019-04-06 05:32:02 +03:00
parent 73e04b71ed
commit 1d75b738a0
44 changed files with 1616 additions and 311 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
kvmd/yamlconf/loader.py
View File

@@ -26,7 +26,6 @@ from typing import IO
from typing import Any
import yaml
import yaml.loader
import yaml.nodes
@@ -37,17 +36,17 @@ def load_yaml_file(path: str) -> Any:
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))
raise ValueError("Incorrect YAML syntax in file %r" % (path))
class _YamlLoader(yaml.loader.Loader): # pylint: disable=too-many-ancestors
class _YamlLoader(yaml.SafeLoader):
def __init__(self, yaml_file: IO) -> None:
yaml.loader.Loader.__init__(self, yaml_file)
super().__init__(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
path = os.path.join(self.__root, self.construct_scalar(node))
return load_yaml_file(path)
_YamlLoader.add_constructor("!include", _YamlLoader.include) # pylint: disable=no-member
_YamlLoader.add_constructor("!include", _YamlLoader.include)