repl_installer: Schedule reboot

This allows us to cleanly exit the script and disconnect before the hard reboot happens.

Author: laurensvalk

pybricksdev/repl_installer.py

@@ -141,11 +141,14 @@ async def install(self, firmware_archive_path):
         await self.exec_line("from _pybricks.install import install; install()")
 
         # Remove installation files
+        self.print_output = False
         await self.exec_line("import uos")
         await self.exec_line("uos.remove('_pybricks/__init__.py')")
         await self.exec_line("uos.remove('_pybricks/install.py')")
         await self.exec_line("uos.remove('_pybricks/firmware.bin')")
 
+        # Disconnect.
+        await self.disconnect()
 
 
 if __name__ == "__main__":

pybricksdev/resources/install_pybricks.py

@@ -10,6 +10,7 @@
 import utime
 import uhashlib
 import uos
+import micropython
 
 
 FLASH_START = 0x8000000
@@ -196,4 +197,12 @@ def install():
     firmware.appl_image_store(overall_checksum.to_bytes(4, 'little'))
 
     # Check result
-    print(firmware.info())
+    if firmware.info()["valid"]:
+        print("Success! The firmware will be installed when it reboots.")
+    else:
+        print("Firmware image not accepted. It will not be installed.")
+
+    # Reboot soon, giving some time to softly disconnect.
+    print("Rebooting soon! Please wait.")
+    timer = umachine.Timer()
+    timer.init(period=1500, mode=umachine.Timer.ONE_SHOT, callback=lambda x: umachine.reset())