From 767aa34fc05eba5403c03163279c978f34a9cba6 Mon Sep 17 00:00:00 2001
From: SiHuan <sihuan@sakuya.love>
Date: Tue, 25 Aug 2020 18:00:20 +0800
Subject: [PATCH] add talk

---
 worker/talk.py | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 worker/talk.py

diff --git a/worker/talk.py b/worker/talk.py
new file mode 100644
index 0000000..43e2580
--- /dev/null
+++ b/worker/talk.py
@@ -0,0 +1,41 @@
+from zzcore import StdAns
+
+from pydub import AudioSegment
+from base64 import b64encode
+import requests, io
+
+class Ans(StdAns):
+    def GETMSG(self):
+        if len(self.parms) < 2:
+            text = "想让我说点什么呢？"
+        else:
+            text = self.raw_msg['message'][5:]
+
+        try:
+           b = tts(text)
+           msg = f'[CQ:record,file=base64://{str(b)}]'
+        except:
+            msg = '什么东西坏掉了,...咱不能说话了!'
+        return msg
+
+def tts(text):
+    url = "https://tts.baidu.com/text2audio"
+    params = {
+        'cuid': 'baike',
+        'lan': 'zh',
+        'ctp': 1,
+        'pdt': 301,
+        'vol': 5,
+        'rate': 32,
+        'per': 5118,
+        'tex': text,
+    }
+
+    r = requests.get(url=url, params=params)
+    s = io.BytesIO(r.content)
+    r.close()
+    # s.seek(0,0)
+    song = AudioSegment.from_mp3(s)
+    amr = song.export(format="amr", parameters=["-ar", "8000"])
+    b = b64encode(amr.read()).decode("utf-8")
+    return b
\ No newline at end of file