重唤醒

Author: 陈志和

src/main.py

@@ -185,10 +185,11 @@ def __working_thread_handler(self):
         self.__keyword_spotting_event.wait()
         self.stop_kws()
         t.join()
-        self.start_kws()
+        # self.start_kws()
 
     def __chat_process(self):
         global name
+        self.__protocol.connect_flag = True
         self.power_green_led.on()
         self.start_vad()
         try:
@@ -216,7 +217,8 @@ def __chat_process(self):
                         if is_listen_flag:
                             self.__protocol.listen("stop")
                             is_listen_flag = False
-                    if not self.__protocol.is_state_ok():
+                    if not self.__protocol.is_state_ok or self.__protocol.connect_flag == False:
+                        print("连接断开，退出聊天流程",self.__protocol.connect_flag)
                         break
                     utime.sleep_ms(5)
                     # logger.debug("read opus data length: {}".format(len(data)))

src/protocol.py

@@ -79,7 +79,7 @@ def __init__(self, host=WSS_HOST, debug=WSS_DEBUG):
         self.__json_message_handler = None
         self.__last_text_value = None
         logger.info("OTA:{}".format(WSS_HOST))
-        
+        self.connect_flag = True
 
     def __str__(self):
         return "{}(host=\"{}\")".format(type(self).__name__, self.host)
@@ -162,10 +162,12 @@ def __recv_thread_worker(self):
                 raw = self.recv()
             except Exception as e:
                 logger.info("{} recv thread break, Exception details: {}".format(self, repr(e)))
+                self.connect_flag = False
                 break
 
             if raw is None or raw == "":
                 logger.info("{} recv thread break, Exception details: read none bytes, websocket disconnect".format(self))
+                self.connect_flag = False
                 break
 
             try: