51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
import json
|
|
import os
|
|
import yaml
|
|
|
|
from config import Config
|
|
from renderer import Renderer
|
|
from structure import Structure
|
|
from xml_loader import XMLLoader
|
|
|
|
class UnsupportedFormatException(Exception):
|
|
...
|
|
|
|
class InstructionSetSchema:
|
|
VALID_EXTENSIONS = ("yaml", "json", "xml")
|
|
|
|
def __init__(self, path: str, configPath: str = "config.json", display: bool = False) -> None:
|
|
self.config = Config(configPath)
|
|
self.display = display
|
|
self.path = path
|
|
self.load()
|
|
|
|
def load(self) -> None:
|
|
_, ext = os.path.splitext(self.path)
|
|
ext = ext[1:].lower()
|
|
|
|
if ext not in InstructionSetSchema.VALID_EXTENSIONS:
|
|
fmts = tuple(map(lambda fmt: f".{fmt}", InstructionSetSchema.VALID_EXTENSIONS))
|
|
raise UnsupportedFormatException(f"'.{ext}' files are not supported. Valid formats: {fmts}")
|
|
|
|
with open(self.path, "r") as f:
|
|
if ext == "yaml":
|
|
schema = yaml.safe_load(f)
|
|
|
|
elif ext == "json":
|
|
schema = json.load(f)
|
|
|
|
elif ext == "xml":
|
|
schema = XMLLoader.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, self.display)
|
|
renderer.render(self)
|
|
renderer.save(path)
|
|
if self.display:
|
|
input("Press ENTER to quit") |