From 76474d6efdedcc9254e0e165d50bf2fc831df31e Mon Sep 17 00:00:00 2001
From: Eric Callahan <arksine.code@gmail.com>
Date: Tue, 26 Apr 2022 08:20:24 -0400
Subject: [PATCH] flash_can: update for latest changes

Signed-off-by:  Eric Callahan <arksine.code@gmail.com>
---
 scripts/flash_can.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/scripts/flash_can.py b/scripts/flash_can.py
index c430d91..bd559aa 100755
--- a/scripts/flash_can.py
+++ b/scripts/flash_can.py
@@ -46,6 +46,7 @@ NACK = bytearray(CMD_HEADER + b"\xf0")
 
 # Klipper Admin Defs (for jumping to bootloader)
 KLIPPER_ADMIN_ID = 0x3f0
+KLIPPER_SET_NODE_CMD = 0x01
 KLIPPER_REBOOT_CMD = 0x02
 
 # CAN Admin Defs
@@ -329,11 +330,11 @@ class CanSocket:
                 curtime = self._loop.time()
             if resp[0] != CANBUS_RESP_NEED_NODEID:
                 continue
-            app = "unknown"
-            if resp[-1] == 1:
-                app = "CanBoot"
-            elif resp[-1] == 0:
-                app = "Klipper"
+            app_names = {
+                KLIPPER_SET_NODE_CMD: "Klipper",
+                CANBUS_CMD_SET_NODEID: "CanBoot"
+            }
+            app = app_names.get(resp[7], "Unknown")
             data = resp[1:7]
             output_line(f"Detected UUID: {data.hex()}, Application: {app}")
             uuid = sum([v << ((5 - i) * 8) for i, v in enumerate(data)])