rivet/schema.py

import yaml

from config import Config
from renderer import Renderer
from structure import Structure

class InstructionSetSchema:
    def __init__(self, path: str, configPath: str = "config.json") -> None:
        self.config = Config(configPath)
        self.path = path
        self.load()
    
    def load(self) -> None:
        with open(self.path, "r") as f:
            schema = yaml.safe_load(f)

        self.structures = {}
        
        for id_, data in schema["structures"].items():
            id_ = str(id_)
            self.structures[id_] = Structure.load(id_, data)
    
    def save(self, path: str) -> None:
        renderer = Renderer(self.config)
        renderer.render(self)
        renderer.save(path)