Add timeout for get_audio_msg

clients/python/wcferry/client.py

@@ -1,7 +1,7 @@
 #! /usr/bin/env python3
 # -*- coding: utf-8 -*-
 
-__version__ = "39.0.7.0"
+__version__ = "39.0.7.1"
 
 import atexit
 import base64
@@ -234,15 +234,17 @@ class Wcf():
 
         return ui
 
-    def get_audio_msg(self, id: int, dir: str) -> str:
+    def get_audio_msg(self, id: int, dir: str, timeout: int = 3) -> str:
         """获取语音消息并转成 MP3
         Args:
             id (int): 语音消息 id
             dir (str): MP3 保存目录（目录不存在会出错）
+            timeout (int): 超时时间（秒）
 
         Returns:
             str: 成功返回存储路径；空字符串为失败，原因见日志。
         """
+        def _get_audio_msg(id, dir):
             req = wcf_pb2.Request()
             req.func = wcf_pb2.FUNC_GET_AUDIO_MSG  # FUNC_GET_AUDIO_MSG
             req.am.id = id
@@ -251,6 +253,20 @@ class Wcf():
 
             return rsp.str
 
+        if timeout == 0:
+            return _get_audio_msg(id, dir)
+
+        cnt = 0
+        while cnt < timeout:
+            path = _get_audio_msg(id, dir)
+            if path:
+                return path
+            sleep(1)
+            cnt += 1
+
+        self.LOG.error(f"获取超时")
+        return ""
+
     def send_text(self, msg: str, receiver: str, aters: Optional[str] = "") -> int:
         """发送文本消息