Updated bootloader_version example

themarpe · themarpe · commit 5e82ca7275c3 · 2022-07-11T15:30:51.000+02:00
diff --git a/depthai-core b/depthai-core
@@ -1 +1 @@
-Subproject commit 5136a452dac64624cfd51723494657783a7d6ea1
+Subproject commit 1589735f5f8a808d355d88672f20a3963f39fb1c
diff --git a/examples/bootloader/bootloader_version.py b/examples/bootloader/bootloader_version.py
@@ -9,13 +9,24 @@
     bl = dai.DeviceBootloader(info)
     print(f'Version: {bl.getVersion()}')
 
-    for mem in (dai.DeviceBootloader.Memory.FLASH, dai.DeviceBootloader.Memory.EMMC):
-        memInfo = bl.getMemoryInfo(mem)
-        if not memInfo.available: continue # Memory is not available, skip
-        print(f"Memory '{mem}' size: {memInfo.size}, info: {memInfo.info}")
+    supportedMemTypes = [dai.DeviceBootloader.Memory.FLASH, dai.DeviceBootloader.Memory.EMMC]
+    if bl.getType() == dai.DeviceBootloader.Type.USB:
+        print("USB Bootloader - supports only Flash memory")
+        supportedMemTypes = [dai.DeviceBootloader.Memory.FLASH];
+    else:
+        print("NETWORK Bootloader")
 
-        appInfo = bl.readApplicationInfo(mem)
-        if appInfo.hasApplication:
-            print(f"Application name: {appInfo.applicationName}, firmware version: {appInfo.firmwareVersion}")
+    try:
+        for mem in supportedMemTypes:
+            memInfo = bl.getMemoryInfo(mem)
+            if memInfo.available:
+                print(f"Memory '{mem}' size: {memInfo.size}, info: {memInfo.info}")
+                appInfo = bl.readApplicationInfo(mem)
+                if appInfo.hasApplication:
+                    print(f"Application name: {appInfo.applicationName}, firmware version: {appInfo.firmwareVersion}")
+            else:
+                print(f"Memory '{mem.name}' not available...")
+    except Exception as ex:
+        print(f"Couldn't retrieve memory details: {ex}")
 else:
     print('No devices found')
