diff --git a/app/config.py b/app/config.py index 45b3e3e..49f3f8c 100644 --- a/app/config.py +++ b/app/config.py @@ -3,4 +3,7 @@ import json def load_config() -> dict: with open("config.json", "r") as config_file: - return json.load(config_file) + config = json.load(config_file) + config["output"] = config["output"] if "output" in config else "static" + config["purge"] = config["purge"] if "purge" in config else [] + diff --git a/main.py b/main.py index ad0d711..b30c325 100644 --- a/main.py +++ b/main.py @@ -6,21 +6,19 @@ from app.generator import DefaultGeneratorProvider config = load_config() -output = config["output"] if "output" in config else "static" -purge = config["purge"] if "purge" in config else [] -generator_configs = config["configs"] - generator_provider = DefaultGeneratorProvider() -for path in purge: +for path in config["purge"]: if isdir(path): shutil.rmtree(path) elif isfile(path): os.remove(path) -for config in generator_configs: - for generator_name in config["generators"]: +output = config["output"] + +for generator_config in config["configs"]: + for generator_name in generator_config["generators"]: generator = generator_provider.get(generator_name) if generator is None: continue - generator.generate(config, output) + generator.generate(generator_config, output)