From 8810aa9a193d1bb63f26d4b5f9c58e80d1b4aa35 Mon Sep 17 00:00:00 2001
From: xel <xel1995@163.com>
Date: Thu, 21 Feb 2019 15:22:28 +0800
Subject: [PATCH] fix auto selece com port bug

---
 kflash.py | 26 +++++++++++++++-----------
 1 file changed, 15 insertions(+), 11 deletions(-)

diff --git a/kflash.py b/kflash.py
index ab04240..4adc0b1 100755
--- a/kflash.py
+++ b/kflash.py
@@ -803,17 +803,21 @@ def flash_firmware(self, firmware_bin: bytes, aes_key: bytes = None, address_off
         print(INFO_MSG,'ANSI colors not used',BASH_TIPS['DEFAULT'])
 
     if args.port == "DEFAULT":
-        try:
-            if args.Board == "goE":
-                list_port_info = list(serial.tools.list_ports.grep("0403")) #Take the second one
-                print(INFO_MSG,"COM Port Auto Detected, Selected ",list_port_info[1].device,BASH_TIPS['DEFAULT'])
-                _port = list_port_info[1].device
-            else:
-                list_port_info = list(serial.tools.list_ports.grep(VID_LIST_FOR_AUTO_LOOKUP)) #Take the first one within the list
-                _port = list_port_info[0].device
-        except StopIteration:
-            print(ERROR_MSG,"No vaild COM Port found in Auto Detect, Check Your Connection or Specify One by"+BASH_TIPS['GREEN']+'`--port/-p`',BASH_TIPS['DEFAULT'])
-            sys.exit(1)
+        if args.Board == "goE":
+            list_port_info = list(serial.tools.list_ports.grep("0403")) #Take the second one
+            if(len(list_port_info)==0):
+                print(ERROR_MSG,"No vaild COM Port found in Auto Detect, Check Your Connection or Specify One by"+BASH_TIPS['GREEN']+'`--port/-p`',BASH_TIPS['DEFAULT'])
+                sys.exit(1)
+            list_port_info.sort()
+            print(INFO_MSG,"COM Port Auto Detected, Selected ",list_port_info[1].device,BASH_TIPS['DEFAULT'])
+            _port = list_port_info[1].device
+        else:
+            try:
+                list_port_info = next(serial.tools.list_ports.grep(VID_LIST_FOR_AUTO_LOOKUP)) #Take the first one within the list
+                _port = list_port_info.device
+            except StopIteration:
+                print(ERROR_MSG,"No vaild COM Port found in Auto Detect, Check Your Connection or Specify One by"+BASH_TIPS['GREEN']+'`--port/-p`',BASH_TIPS['DEFAULT'])
+                sys.exit(1)
     else:
         _port = args.port
         print(INFO_MSG,"COM Port Selected Manually: ",_port,BASH_TIPS['DEFAULT'])