2020-02-19 15:31:39 +00:00
|
|
|
|
""" PagerMaid features that uses external HTTP APIs other than Telegram. """
|
|
|
|
|
|
|
|
|
|
from googletrans import Translator, LANGUAGES
|
|
|
|
|
from os import remove
|
2020-07-24 12:04:29 +00:00
|
|
|
|
from requests import get
|
2020-02-19 15:31:39 +00:00
|
|
|
|
from time import sleep
|
|
|
|
|
from threading import Thread
|
|
|
|
|
from bs4 import BeautifulSoup
|
|
|
|
|
from gtts import gTTS
|
|
|
|
|
from re import compile as regex_compile
|
|
|
|
|
from pagermaid import log
|
|
|
|
|
from pagermaid.listener import listener, config
|
|
|
|
|
from pagermaid.utils import clear_emojis, attach_log, fetch_youtube_audio
|
|
|
|
|
|
|
|
|
|
|
2020-08-09 15:17:55 +00:00
|
|
|
|
@listener(is_plugin=False, outgoing=True, command="translate",
|
2020-04-05 13:09:39 +00:00
|
|
|
|
description="通过 Google 翻译将目标消息翻译成指定的语言。(支持回复)",
|
|
|
|
|
parameters="<文本>")
|
2020-02-19 15:31:39 +00:00
|
|
|
|
async def translate(context):
|
|
|
|
|
""" PagerMaid universal translator. """
|
|
|
|
|
translator = Translator()
|
|
|
|
|
reply = await context.get_reply_message()
|
|
|
|
|
message = context.arguments
|
|
|
|
|
lang = config['application_language']
|
|
|
|
|
if message:
|
|
|
|
|
pass
|
|
|
|
|
elif reply:
|
|
|
|
|
message = reply.text
|
|
|
|
|
else:
|
|
|
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
try:
|
2020-02-24 13:10:14 +00:00
|
|
|
|
await context.edit("正在生成翻译中 . . .")
|
2020-02-19 15:31:39 +00:00
|
|
|
|
result = translator.translate(clear_emojis(message), dest=lang)
|
|
|
|
|
except ValueError:
|
|
|
|
|
await context.edit("出错了呜呜呜 ~ 找不到目标语言,请更正配置文件中的错误。")
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
source_lang = LANGUAGES[f'{result.src.lower()}']
|
|
|
|
|
trans_lang = LANGUAGES[f'{result.dest.lower()}']
|
2020-02-20 13:37:19 +00:00
|
|
|
|
result = f"**文本翻译** 源语言 {source_lang.title()}:\n{result.text}"
|
2020-02-19 15:31:39 +00:00
|
|
|
|
|
|
|
|
|
if len(result) > 4096:
|
2020-04-05 13:09:39 +00:00
|
|
|
|
await context.edit("输出超出 TG 限制,正在尝试上传文件。")
|
2020-02-19 15:31:39 +00:00
|
|
|
|
await attach_log(result, context.chat_id, "translation.txt", context.id)
|
|
|
|
|
return
|
|
|
|
|
await context.edit(result)
|
|
|
|
|
if len(result) <= 4096:
|
2020-02-20 13:37:19 +00:00
|
|
|
|
await log(f"把 `{message}` 从 {source_lang} 翻译到了 {trans_lang}")
|
2020-02-19 15:31:39 +00:00
|
|
|
|
else:
|
2020-02-20 13:37:19 +00:00
|
|
|
|
await log(f"把一条消息从 {source_lang} 翻译到了 {trans_lang}.")
|
2020-02-19 15:31:39 +00:00
|
|
|
|
|
|
|
|
|
|
2020-08-09 15:17:55 +00:00
|
|
|
|
@listener(is_plugin=False, outgoing=True, command="tts",
|
2020-02-19 15:31:39 +00:00
|
|
|
|
description="通过 Google文本到语音 基于字符串生成语音消息。",
|
|
|
|
|
parameters="<string>")
|
|
|
|
|
async def tts(context):
|
|
|
|
|
""" Send TTS stuff as voice message. """
|
|
|
|
|
reply = await context.get_reply_message()
|
|
|
|
|
message = context.arguments
|
2020-02-24 13:10:14 +00:00
|
|
|
|
lang = config['application_tts']
|
2020-02-19 15:31:39 +00:00
|
|
|
|
if message:
|
|
|
|
|
pass
|
|
|
|
|
elif reply:
|
|
|
|
|
message = reply.text
|
|
|
|
|
else:
|
|
|
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
await context.edit("生成语音中 . . .")
|
2020-06-06 17:21:33 +00:00
|
|
|
|
gTTS(message, lang=lang)
|
2020-02-19 15:31:39 +00:00
|
|
|
|
except AssertionError:
|
|
|
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|
|
|
|
return
|
|
|
|
|
except ValueError:
|
|
|
|
|
await context.edit('出错了呜呜呜 ~ 找不到目标语言,请更正配置文件中的错误。')
|
|
|
|
|
return
|
|
|
|
|
except RuntimeError:
|
|
|
|
|
await context.edit('出错了呜呜呜 ~ 加载语言数组时出错。')
|
|
|
|
|
return
|
2020-06-06 17:21:33 +00:00
|
|
|
|
google_tts = gTTS(message, lang=lang)
|
2020-02-19 15:31:39 +00:00
|
|
|
|
google_tts.save("vocals.mp3")
|
|
|
|
|
with open("vocals.mp3", "rb") as audio:
|
|
|
|
|
line_list = list(audio)
|
|
|
|
|
line_count = len(line_list)
|
|
|
|
|
if line_count == 1:
|
2020-06-06 17:21:33 +00:00
|
|
|
|
google_tts = gTTS(message, lang=lang)
|
2020-02-19 15:31:39 +00:00
|
|
|
|
google_tts.save("vocals.mp3")
|
|
|
|
|
with open("vocals.mp3", "r"):
|
|
|
|
|
await context.client.send_file(context.chat_id, "vocals.mp3", voice_note=True)
|
|
|
|
|
remove("vocals.mp3")
|
|
|
|
|
if len(message) <= 4096:
|
|
|
|
|
await log(f"生成了一条文本到语音的音频消息 : `{message}`.")
|
|
|
|
|
else:
|
|
|
|
|
await log("生成了一条文本到语音的音频消息。")
|
|
|
|
|
await context.delete()
|
|
|
|
|
|
|
|
|
|
|
2020-08-09 15:17:55 +00:00
|
|
|
|
@listener(is_plugin=False, outgoing=True, command="google",
|
2020-02-19 15:31:39 +00:00
|
|
|
|
description="使用 Google 查询",
|
|
|
|
|
parameters="<query>")
|
2020-07-24 12:04:29 +00:00
|
|
|
|
async def googletest(context):
|
2020-02-19 15:31:39 +00:00
|
|
|
|
""" Searches Google for a string. """
|
2020-07-24 12:04:29 +00:00
|
|
|
|
USER_AGENT = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.14; rv:65.0) Gecko/20100101 Firefox/65.0"
|
|
|
|
|
headers = {"user-agent": USER_AGENT}
|
2020-08-07 05:38:31 +00:00
|
|
|
|
reply = await context.get_reply_message()
|
|
|
|
|
query = context.arguments
|
|
|
|
|
lang = config['application_language']
|
|
|
|
|
if query:
|
|
|
|
|
pass
|
|
|
|
|
elif reply:
|
|
|
|
|
query = reply.text
|
|
|
|
|
else:
|
2020-02-19 15:31:39 +00:00
|
|
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|
|
|
|
return
|
2020-08-07 05:38:31 +00:00
|
|
|
|
|
2020-07-24 12:04:29 +00:00
|
|
|
|
query = query.replace(' ', '+')
|
2020-08-01 23:21:14 +00:00
|
|
|
|
URL = [('https://google.com.hk/search?q=' + query),('https://google.com/search?q=' + query)]
|
2020-02-19 15:31:39 +00:00
|
|
|
|
await context.edit("正在拉取结果 . . .")
|
2020-08-01 23:21:14 +00:00
|
|
|
|
count = 0
|
|
|
|
|
for g in URL:
|
|
|
|
|
count += 1
|
|
|
|
|
resp = get(g, headers=headers)
|
|
|
|
|
if resp.status_code == 200:
|
|
|
|
|
break
|
|
|
|
|
elif count == 2 and not resp.status_code == 200:
|
|
|
|
|
await context.edit("连接到 google服务器 失败")
|
|
|
|
|
return
|
|
|
|
|
else:
|
|
|
|
|
pass
|
|
|
|
|
soup = BeautifulSoup(resp.content, "html.parser")
|
|
|
|
|
results = ""
|
|
|
|
|
count = 0
|
|
|
|
|
for g in soup.find_all('div', class_='r'):
|
|
|
|
|
if count == int(config['result_length']):
|
|
|
|
|
break
|
|
|
|
|
count += 1
|
|
|
|
|
anchors = g.find_all('a')
|
|
|
|
|
if anchors:
|
|
|
|
|
title = g.find('h3').text
|
|
|
|
|
link = anchors[0]['href']
|
|
|
|
|
results += f"\n[{title}]({link}) \n"
|
|
|
|
|
await context.edit(f"**Google** |`{query}`| 🎙 🔍 \n"
|
|
|
|
|
f"{results}",
|
|
|
|
|
link_preview=False)
|
|
|
|
|
await log(f"在Google搜索引擎上查询了 `{query}`")
|
2020-02-19 15:31:39 +00:00
|
|
|
|
|
|
|
|
|
|
2020-08-09 15:17:55 +00:00
|
|
|
|
@listener(is_plugin=False, outgoing=True, command="fetchaudio",
|
2020-02-19 15:31:39 +00:00
|
|
|
|
description="从多个平台获取音频文件。",
|
2020-02-20 13:37:19 +00:00
|
|
|
|
parameters="<url>,<string>")
|
2020-02-19 15:31:39 +00:00
|
|
|
|
async def fetchaudio(context):
|
2020-02-20 13:37:19 +00:00
|
|
|
|
if context.arguments:
|
|
|
|
|
if ',' in context.arguments:
|
|
|
|
|
url, string_2 = context.arguments.split(',', 1)
|
|
|
|
|
else:
|
|
|
|
|
url = context.arguments
|
|
|
|
|
string_2 = "#audio "
|
|
|
|
|
else:
|
|
|
|
|
await context.edit("出错了呜呜呜 ~ 错误的语法。")
|
|
|
|
|
return
|
2020-02-19 15:31:39 +00:00
|
|
|
|
""" Fetches audio from provided URL. """
|
|
|
|
|
reply = await context.get_reply_message()
|
|
|
|
|
reply_id = None
|
|
|
|
|
await context.edit("拉取音频中 . . .")
|
|
|
|
|
if reply:
|
|
|
|
|
reply_id = reply.id
|
|
|
|
|
if url is None:
|
|
|
|
|
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
|
|
|
|
return
|
|
|
|
|
youtube_pattern = regex_compile(r"^(http(s)?://)?((w){3}.)?youtu(be|.be)?(\.com)?/.+")
|
|
|
|
|
if youtube_pattern.match(url):
|
2020-02-20 13:37:19 +00:00
|
|
|
|
if not await fetch_youtube_audio(url, context.chat_id, reply_id, string_2):
|
2020-02-19 15:31:39 +00:00
|
|
|
|
await context.edit("出错了呜呜呜 ~ 原声带下载失败。")
|
|
|
|
|
await log(f"从链接中获取了一条音频,链接: {url}.")
|
2020-02-20 13:56:11 +00:00
|
|
|
|
await context.delete()
|