Add mpremote session validation

Author: malcolm-sparkfun

MicroPython_Firmware_Uploader/MicroPython_Firmware_Uploader.py

@@ -634,6 +634,7 @@ def on_finished(self, status, action_type, job_id) -> None:
         # If the reset is finished, re-enable the UX
         if action_type == AUxEsptoolResetESP32.ACTION_ID:
             self.writeMessage("Reset complete...")
+            self.writeMessage("DONE: Firmware file copied to ESP32 device.\n")
             self.disable_interface(False)
 
         if action_type == AUxRp2UploadRp2.ACTION_ID:
@@ -851,7 +852,7 @@ def on_discover_btn_pressed(self) -> None:
             return
 
         mpr = self.get_mpremote_session()
-        if mpr.is_connected():
+        if mpr is not None:
             try:
                 boardName = mpr.get_short_board_name()
                 if boardName is None:
@@ -1059,7 +1060,14 @@ def get_mpremote_session(self) -> MPRemoteSession:
         else:
             portName = self.port_button.text()
 
-        return MPRemoteSession(portName)
+        mpr = MPRemoteSession(portName)
+
+        if mpr.validate_session():
+            self.writeMessage("MPRemote session validated.\n")
+            return mpr
+        else:
+            self.writeMessage("MPRemote session failed to validate.\n")
+            return None
 
     def do_upload_rp2(self, fwFile = None) -> None:
         """Upload the firmware to the RP2."""
@@ -1074,7 +1082,7 @@ def do_upload_rp2(self, fwFile = None) -> None:
 
         # If we are able to use mpremote to enter bootloader mode, do so
         # Otherwise, prompt the user to press the correct button sequence to enter bootloader mode
-        if mpr.is_connected():
+        if mpr is not None:
             self.writeMessage("Able to automatically enter bootloader mode...\n")
             mpr.enter_bootloader()
             self.writeMessage("Entering bootloader mode...\n")

MicroPython_Firmware_Uploader/mpremote_utils.py

@@ -117,7 +117,7 @@ def exec_command(self, command: str) -> None:
         self.args.expr = [command]
         commands.do_exec(state=self.state, args=self.args)
 
-    def exec_command_with_output(self, command: str, timeout: Optional[float] = 3) -> str:
+    def exec_command_with_output(self, command: str, timeout: Optional[float] = 2) -> str:
 
         self.args.expr = [command]
         capture = StdoutCapture()
@@ -146,6 +146,23 @@ def execute():
         capture.stop()
         return capture.get_output()
 
+    def validate_session(self) -> None:
+        # Ensure that we can ping the device by checking sys.implementation.name
+        # This is a good way to check if the device is connected and responsive
+        try:
+            if self.is_connected():
+                res = self.exec_command_with_output("import sys; print(sys.implementation.name)")
+                if res is None:
+                    return False
+                if res.strip() == "micropython":
+                    return True
+        except Exception as e:
+            #print("Error validating session:", e)
+            pass
+        
+        return False
+
+    
     def eval_command(self, command: str) -> None:
         self.args.expr = self.args.eval = [command]
         commands.do_eval(state=self.state, args=self.args)