Added yaml constructor that get the realpath of a file

This commit is contained in:
2025-11-09 03:42:55 +01:00
parent 498b0ba480
commit 1931ab71ca

View File

@@ -83,9 +83,23 @@ def template_constructor(environment: Environment):
return inner return inner
def get_loader(): def realpath_constructor(directory: pathlib.Path):
def inner(loader: yaml.SafeLoader, node: yaml.nodes.ScalarNode):
try:
realpath = directory.joinpath(loader.construct_scalar(node)).resolve(
strict=True
)
return str(realpath)
except Exception:
raise yaml.MarkedYAMLError("Failed to get real path", node.start_mark)
return inner
def get_loader(directory: pathlib.Path):
"""Add special constructors to yaml loader""" """Add special constructors to yaml loader"""
loader = yaml.SafeLoader loader = yaml.SafeLoader
loader.add_constructor("!realpath", realpath_constructor(directory))
loader.add_constructor("!schematic", schematic_constructor) loader.add_constructor("!schematic", schematic_constructor)
loader.add_constructor("!patch", template_constructor(PATCHES)) loader.add_constructor("!patch", template_constructor(PATCHES))
@@ -97,7 +111,7 @@ def get_defaults(directory: pathlib.Path, root: pathlib.Path):
"""Compute the defaults from the provided directory and parents.""" """Compute the defaults from the provided directory and parents."""
try: try:
with open(directory.joinpath("_defaults.yaml")) as fyaml: with open(directory.joinpath("_defaults.yaml")) as fyaml:
yml_data = yaml.load(fyaml, Loader=get_loader()) yml_data = yaml.load(fyaml, Loader=get_loader(directory))
except OSError: except OSError:
yml_data = {} yml_data = {}
@@ -122,7 +136,7 @@ def main():
filename = str(fullname.relative_to(NODES).parent) + "/" + fullname.stem filename = str(fullname.relative_to(NODES).parent) + "/" + fullname.stem
with open(fullname) as fyaml: with open(fullname) as fyaml:
yml_data = yaml.load(fyaml, Loader=get_loader()) yml_data = yaml.load(fyaml, Loader=get_loader(fullname.parent))
yml_data = get_defaults(fullname.parent, NODES) | yml_data yml_data = get_defaults(fullname.parent, NODES) | yml_data
yml_data["hostname"] = fullname.stem yml_data["hostname"] = fullname.stem
yml_data["filename"] = filename yml_data["filename"] = filename