def messages(self, auto_handle_message=True, timeout_return_none=False):
self.start_time = time.time()
self.check_ws_ready()
while self._running:
# check tab max timeout
now = time.time()
if now - self.start_time > self.TAB_TIMEOUT:
raise TabTimeout()
try:
message = self.ws.recv()
parsed_message = json.loads(message)
if auto_handle_message:
self.handle_messages(parsed_message)
yield parsed_message
except websocket.WebSocketTimeoutException:
if timeout_return_none:
yield None
continue
except websocket.WebSocketConnectionClosedException:
break
except KeyboardInterrupt:
break
评论列表
文章目录