From ac7aabc5f9c0625aa3133e5fbba86223173b4019 Mon Sep 17 00:00:00 2001
From: rafal-gorecki <rafal.gorecki@husarion.com>
Date: Thu, 23 May 2024 15:06:31 +0200
Subject: [PATCH] Clean output flashing script

---
 flash-firmware.py | 34 ++++++++++++++++++++--------------
 1 file changed, 20 insertions(+), 14 deletions(-)

diff --git a/flash-firmware.py b/flash-firmware.py
index 73a5ce7..df3b874 100755
--- a/flash-firmware.py
+++ b/flash-firmware.py
@@ -26,6 +26,8 @@ def __init__(self, binary_file):
         self.binary_file = binary_file
         sys_arch = str(sh.uname("-m")).strip()
 
+        self.max_approach_no = 3
+
         print(f"System architecture: {sys_arch}")
 
         if sys_arch == "armv7l":
@@ -71,20 +73,24 @@ def exit_bootloader_mode(self):
 
     def try_flash_operation(self, operation_name, flash_args):
         self.enter_bootloader_mode()
-        
-        try:
-            if operation_name == "Flashing":
-                sh.stm32flash(self.port, *flash_args, _out=sys.stdout)
-                time.sleep(0.2)
-                print("Success! The robot firmware has been uploaded.")
-            else:                    
-                sh.stm32flash(self.port, *flash_args)
-                time.sleep(0.2)
-        except Exception as e:
-            stderr = e.stderr.decode('utf-8')
-            if stderr:
-                print(f"ERROR {operation_name} went wrong: {stderr}")
-            
+
+        print(f"{operation_name} operation started.")
+        for i in range(self.max_approach_no):
+            try:
+                if operation_name == "Flashing":
+                    sh.stm32flash(self.port, *flash_args, _out=sys.stdout)
+                    print("Success! The robot firmware has been uploaded.")
+                else:
+                    sh.stm32flash(self.port, *flash_args)
+                break
+            except Exception as e:
+                stderr = e.stderr.decode('utf-8')
+                if stderr:
+                    print(f"[ERROR] [{operation_name}]: \n\t{stderr}.")
+                else:
+                    break
+            time.sleep(0.2) # Delay between attempts
+
         self.exit_bootloader_mode()