home-assistant-addon/script/generate.py

63 lines
1.8 KiB
Python
Raw Normal View History

2019-04-24 17:09:50 +02:00
#!/usr/bin/env python3
import argparse
import yaml
from pathlib import Path
from enum import Enum
from shutil import copyfile
import sys
2019-04-24 17:09:50 +02:00
2019-04-24 17:09:50 +02:00
class Channel(Enum):
stable = "stable"
beta = "beta"
dev = "dev"
2019-04-24 17:09:50 +02:00
def main(args):
parser = argparse.ArgumentParser(
description="Generate ESPHome Home Assistant config.json"
)
parser.add_argument("channels", nargs="+", type=Channel, choices=list(Channel))
args = parser.parse_args(args)
2019-04-24 17:09:50 +02:00
root = Path(__file__).parent.parent
templ = root / "template"
2019-04-24 17:09:50 +02:00
with open(templ / "addon_config.yaml", "r") as f:
config = yaml.safe_load(f)
2019-04-24 17:09:50 +02:00
copyf = config["copy_files"]
2019-04-24 17:09:50 +02:00
for channel in args.channels:
conf = config[f"esphome-{channel.value}"]
base_image = conf.pop("base_image", None)
dir_ = root / conf.pop("directory")
2022-03-24 05:23:36 +01:00
path = dir_ / "config.yaml"
with open(path, "w") as f:
2022-03-24 05:23:36 +01:00
yaml.dump(conf, f, indent=2, sort_keys=True, explicit_start=True)
for file_, conf_ in copyf.items():
if Path.exists(templ / channel.value / file_):
copyfile(templ / channel.value / file_, dir_ / file_)
else:
copyfile(templ / file_, dir_ / file_)
path = dir_ / "FILES ARE GENERATED DO NOT EDIT"
with open(path, "w") as f:
2021-02-14 20:45:10 +01:00
f.write("Any edits should be made to the files in the 'template' directory")
if channel == Channel.dev:
2022-03-24 05:23:36 +01:00
path = dir_ / "build.yaml"
build_conf = {
"build_from": {
arch: base_image.format(arch=arch) for arch in conf["arch"]
}
}
with open(path, "w") as f:
2022-03-24 05:23:36 +01:00
yaml.dump(build_conf, f, indent=2, sort_keys=True, explicit_start=True)
2019-04-24 17:09:50 +02:00
if __name__ == "__main__":
main(sys.argv[1:])