Upload ext_voice.py

ext_voice.py

@@ -21,11 +21,10 @@ from aiohttp import request
 from loguru import logger
 from nonebot.exception import ActionFailed
 import asyncio
+import re
 import requests
 import base64
 import uuid
-import re
-
 
 try:
     from ujson import loads as loadJsonS
@@ -148,37 +147,26 @@ class CustomExtension(Extension):
 
         # todo: 如果需要使用本地的语音合成api，请取消注释下面的代码，并自行改为您的api地址，将填入合成文本的地方改为{text}
         # url = f"http://127.0.0.1:23211/to_voice?text={text}"
-        # 去除base64字符串中的非法字符
-        def call(self, args, context):
-            url = args['url']
-            text = args.get('text')
-            is_base64 = args.get('is_base64', False)
-            raw_text = text
-
-            # 下载语音文件
-            r = requests.get(url)
-
-            # 对base64字符串进行处理
-            if is_base64:
-                pattern = re.compile('[^a-zA-Z0-9+/=]')
-                content = pattern.sub('', r.content.decode())
-                audio_data = base64.b64decode(content + b'=' * (-len(content) % 4), validate=True)
-            else:
-                audio_data = r.content
-
-            # 写入音频文件
-            file_name = f"{voice_path}{uuid.uuid1()}.ogg"
-            with open(file_name, "wb") as f:
-                f.write(audio_data)
-
-            local_url = f"file:///{os.path.abspath(file_name)}"
-
-            if text is not None:
-                return {
-                    'voice': local_url,  # 语音url
-                    'text': f"[语音消息] {raw_text}",  # 文本
-                }
-            return {}
+
+        # 下载语音文件
+        r = requests.get(url)
+        file_name = f"{voice_path}{uuid.uuid1()}.ogg"
+        if is_base64:
+            audio_data = base64.b64decode(r.content + b'=' * (-len(r.content) % 4), validate=True)
+        else:
+            audio_data = r.content
+
+        with open(file_name, "wb") as f:
+            f.write(audio_data)
+
+        local_url = f"file:///{os.path.abspath(file_name)}"
+
+        if text is not None:
+            return {
+                'voice': local_url,             # 语音url
+                'text': f"[语音消息] {raw_text}",    # 文本
+            }
+        return {}
 
     def __init__(self, custom_config: dict):
         super().__init__(ext_config.copy(), custom_config)