Source code for labgrid.remote.config

from pprint import pprint

import os

import attr
import jinja2

from ..util.yaml import load
from ..exceptions import NoConfigFoundError




@attr.s(eq=False)
class ResourceConfig:
    filename = attr.ib(validator=attr.validators.instance_of(str))
    template_env = attr.ib(default=attr.Factory(dict), validator=attr.validators.instance_of(dict))



    def __attrs_post_init__(self):
        env = jinja2.Environment(
            loader=jinja2.FileSystemLoader(os.path.dirname(self.filename)),
            line_statement_prefix="#",
            line_comment_prefix="##",
        )
        try:
            template = env.get_template(os.path.basename(self.filename))
        except jinja2.TemplateNotFound:
            raise NoConfigFoundError(f"{self.filename} could not be found")
        rendered = template.render(self.template_env)
        pprint(("rendered", rendered))
        self.data = load(rendered)
        pprint(("loaded", self.data))