diff --git a/kunlun.py b/kunlun.py index 735c3ea..fb85071 100644 --- a/kunlun.py +++ b/kunlun.py @@ -27,6 +27,10 @@ from bin.factory_mode import ftm_handle def init_send(s_port:serial.Serial, send_str:str): s_info = bytearray() send_str=send_str.encode('utf-8') + key_words={ + b"WQKL":b"Recieving RAM-IMAGE in xmodem : C", + b"HZPG":b"waiting recieve image: C" + } while True: s_port.write(send_str) time.sleep(0.1) @@ -34,7 +38,7 @@ def init_send(s_port:serial.Serial, send_str:str): bytes2read = s_port.in_waiting tmp = s_port.read(bytes2read) s_info += tmp - if(s_info.find(b"Recieving RAM-IMAGE in xmodem : C")>=0): + if(s_info.find(key_words[send_str])>=0): m_ram=True else: m_ram=False @@ -375,6 +379,8 @@ def global_def(): if(args.kunlun_version is None): print("请指定参数 -k") sys.exit(-1) + elif(args.kunlun_version == "4"): + init_str="HZPG" if(args.upload_key is not None): upload_key = args.upload_key else: