"""Create local device, calibration and secrets files without overwriting any.""" import argparse import base64 import os from pathlib import Path import re import secrets ROOT = Path(__file__).resolve().parents[1] def create(directory, name, friendly_name): if not re.fullmatch(r'[a-z][a-z0-9-]{0,30}', name) or name.endswith('-'): raise ValueError('Gebruik een unieke naam van 1–31 tekens: kleine letters, cijfers en koppeltekens.') import json device = (ROOT / 'device.example.yaml').read_text().replace('"cyd-new"', json.dumps(name), 1) device = device.replace('"Mijn CYD"', json.dumps(friendly_name, ensure_ascii=False), 1) secret_text = (ROOT / 'secrets.yaml.example').read_text() replacements = [base64.b64encode(secrets.token_bytes(32)).decode(), secrets.token_urlsafe(24), secrets.token_urlsafe(18)] for value in replacements: secret_text = secret_text.replace('WORDT_UNIEK_GEGENEREERD', value, 1) files = {'device.yaml': device, 'calibration.yaml': (ROOT / 'calibration.example.yaml').read_text(), 'secrets.yaml': secret_text} directory = Path(directory) if not directory.is_dir(): raise ValueError('De doelmap moet al bestaan en de uitgepakte code bevatten.') for name in files: if (directory / name).exists(): raise ValueError(f'{directory / name} bestaat al; niets overschreven. Gebruik een nieuwe kopie van de map.') created = [] try: for name, contents in files.items(): path = directory / name fd = os.open(path, os.O_WRONLY | os.O_CREAT | os.O_EXCL, 0o600) created.append(path) with os.fdopen(fd, 'w', encoding='utf-8') as handle: handle.write(contents) except Exception: for path in created: path.unlink(missing_ok=True) raise return created def main(): parser = argparse.ArgumentParser(description=__doc__) parser.add_argument('--name', required=True, help='Unieke ESPHome-naam, bijvoorbeeld display-keuken') parser.add_argument('--friendly-name', default='Mijn CYD') parser.add_argument('--directory', type=Path, default=ROOT) args = parser.parse_args() try: for path in create(args.directory, args.name, args.friendly_name): print(f'Gemaakt: {path}') print('Vul wifi lokaal in secrets.yaml in. Sleutels zijn uniek gegenereerd en worden niet getoond.') print('Volg nu README.md; flash device.yaml, niet rechtstreeks de basisconfiguratie.') except (ValueError, OSError) as exc: parser.exit(1, f'Fout: {exc}\n') if __name__ == '__main__': main()