import os
import machine
from utime import sleep_ms

# Give time to cancel this boot script
print("Press Ctrl-C to stop new boot script...")
sleep_ms(1000)

root_files = os.listdir()
update_files = ['boot.py.new', 'main.py.new', 'hal.py.new']
files_to_update = []

# Check for FW updates and verify new FW files
for file in update_files:
    if file in root_files:
        print("boot.py: trying to update " + file)
        # Try to load the user code
        try:
            with open(file, 'r') as code:
                compile(code.read(), "snippet", 'exec')
            files_to_update.append(file)
        except:
            print("boot.py: " + file + " compilation failed")
            files_to_update.clear()
            break

# If valid updates replace with new FW
for file in files_to_update:
    os.rename(file, file.replace('.new', ''))

# If updates, reboot to load new FW
if len(files_to_update) != 0:
    machine.reset()