write_eeprom.py

# -*- coding: utf-8 -*-
import re
from datetime import datetime
import click
import subprocess
import uuid


uuid4hex = re.compile('[0-9a-f]{8}\-[0-9a-f]{4}\-4[0-9a-f]{3}\-[89ab][0-9a-f]{3}\-[0-9a-f]{12}', re.I)

def is_valid_uuid(uuid):
    return bool(uuid4hex.match(uuid))


@click.command()
@click.argument('version')
@click.argument('serial_number')
def main(version, serial_number):
    
    #1. create eeprom settings from template
    assert re.match(r"\d\.\d", version) is not None, "Version must be of the form x.y"
    assert is_valid_uuid(serial_number), "must be valid hex uuid4"

    major, minor = version.split(".")
    version_as_hex_string = "0x" + major.zfill(3) + minor
    current_time = datetime.utcnow().isoformat()
    product_name = "Pioreactor HAT"

    with open("./pioreactor_eeprom_settings.txt.template", "r") as in_file:
        with open("./pioreactor_eeprom_settings.txt", "w") as out_file:
            eeprom_text = in_file.read().format(product_version=version_as_hex_string, current_time=current_time, serial_number=serial_number, product_name=product_name)
            out_file.write(eeprom_text)
    
    # 2. compile to .eep file.
    output = subprocess.run(["./eepmake", "pioreactor_eeprom_settings.txt", "hat.eep"])
    if output.returncode != 0:
        print("Exiting due to error in `./eepmake pioreactor_eeprom_settings.txt hat.eep`")
        raise click.Abort()

    # 3. write to hat
    output = subprocess.run(["./eepflash.sh", "-w" ,"-f=hat.eep" ,"-t=24c32", "-y"])
    if output.returncode != 0:
        print("Exiting due to error in `./eepflash.sh -w -f=hat.eep -t=24c32 -y`")
        raise click.Abort()

    return serial_number


if __name__ == '__main__':
    main()