diff --git a/python_payload/st3m/run.py b/python_payload/st3m/run.py index d3daec0e995e6f1f0ab5158164146c16544cf3db..13a4a281e815b5ad5ea24b8a4c338f5b541676f3 100644 --- a/python_payload/st3m/run.py +++ b/python_payload/st3m/run.py @@ -103,7 +103,7 @@ def run_view(v: View) -> None: reactor.run() -def yeet_local_changes() -> None: +def _yeet_local_changes() -> None: os.remove("/flash/sys/.sys-installed") machine.reset() @@ -125,10 +125,12 @@ def run_main() -> None: [ MenuItemBack(), MenuItemForeground("Settings", menu_settings), - MenuItemNoop("Disk Mode"), + MenuItemAction("Disk Mode (Flash)", machine.disk_mode_flash), + MenuItemAction("Disk Mode (SD)", machine.disk_mode_sd), MenuItemLaunchPersistentView("About", About), - MenuItemAction("Yeet Local Changes", yeet_local_changes), + MenuItemAction("Yeet Local Changes", _yeet_local_changes), MenuItemAction("Reboot", lambda: machine.reset()), + MenuItemAction("Reboot", machine.reset), ], ) menu_main = SunMenu( diff --git a/sim/fakes/machine.py b/sim/fakes/machine.py index 77b2272a1ec254b6a7f5da0b15f3140fb3c47edd..b0efefe90ed5032adf6dcd424539ab9d1a156c2e 100644 --- a/sim/fakes/machine.py +++ b/sim/fakes/machine.py @@ -1,3 +1,6 @@ +import sys + + class Pin: IN = None @@ -14,3 +17,18 @@ class ADC: def read_uv(self): # A half full battery as seen by the ADC return 3.8e6 / 2 + + +def reset(): + print("beep boop i have reset") + sys.exit(0) + + +def disk_mode_flash(): + print("beep boop i'm now in flash disk mode") + sys.exit(0) + + +def disk_mode_sd(): + print("beep boop i'm now in sd card disk mode") + sys.exit(0)