X-Git-Url: https://us-delaporte.pagekite.me/gitweb/dotfiles.git/blobdiff_plain/09daf7343b6b5f929000c4001ff946e19de8914e..aedaada157e561da06b77c23054c8db76c36ce48:/invoke/esptool/tasks.py

diff --git a/invoke/esptool/tasks.py b/invoke/esptool/tasks.py
index 54793ee..546e261 100644
--- a/invoke/esptool/tasks.py
+++ b/invoke/esptool/tasks.py
@@ -9,7 +9,12 @@ TODAY = dt.now().strftime('%Y.%m.%d')
 # Examples for CardputerAdv
 DEVICE = 'Espressif'
 PORT='/dev/ttyACM0' # Linux
+CHIP='ESP32-S3'
 # PORT='COM0' # Windows
+MICROPYTHON=f'$HOME/Downloads/ESP32_GENERIC_S3-20260406-v1.28.0.bin'
+# MicroPython version 
+# from 
+# https://micropython.org/download/ESP32_GENERIC_S3/
 
 @task
 def help(c):
@@ -66,7 +71,8 @@ def listusb(c):
 
 @task
 def chip_id(c):
-    c.run(f'esptool --port {PORT} chip-id')
+    result = c.run(f'esptool --port {PORT} chip-id')
+    assert CHIP in result.stdout
     print('[ok]')
 
 @task
@@ -75,3 +81,18 @@ def check_all(c):
     check_conn(c)
     chip_id(c)
 
+@task
+def erase_flash(c):
+    '''Erasing flash is recommended before flashing fresh.'''
+    c.run(f'esptool --port {PORT} erase-flash')
+
+@task
+def write_flash(c, filepath=MICROPYTHON):
+    '''Write flash - defaults to MicroPython
+
+    Requires .bin file from micropython.org
+    '''
+    c.run(f'esptool --chip {CHIP} --port {PORT} --baud 460800 write-flash -z 0x1000 {filepath}')
+
+
+