update_firmware.py

from power_api import SixfabPower, Definition
import time
import sys, os

api = SixfabPower()

firmware_path = "./firmwares/sixfab_pms_firmware_v0.3.3.bin"

error_occured = 0
try:
    for step in api.update_firmware(firmware_path):
        print(f"{step}%")
        
except:
    error_occured = 1
    print("raised error")

finally:
    update_successful = 0
    if error_occured is 0:
        counter = 0
        
        for i in range(15):
            print('.', sep=' ', end=' ', flush=True)
            time.sleep(1)

        while(1):
            try:
                sys.stdout = open(os.devnull, 'w')
                fw_ver = api.get_firmware_ver()
                sys.stdout = sys.__stdout__
            except:
                counter += 1
                print('.', sep=' ', end=' ', flush=True)
            else:
               update_successful = 1
            finally:
                if update_successful is 1:
                    print("\nUpdate is Successful to FW Ver: " + str(fw_ver))
                    break
                else: 
                    if counter >= 10:
                        print("Update is Failed!")
                        break
            
            time.sleep(2)