60 lines
2.3 KiB
Python
60 lines
2.3 KiB
Python
import time
|
|
import html
|
|
import googletrans
|
|
from pyrogram import Client, filters
|
|
from .. import config, help_dict, log_errors, public_log_errors
|
|
|
|
PROBLEM_CODES = set(i for i in googletrans.LANGUAGES if '-' in i)
|
|
ZWS = '\u200B'
|
|
|
|
@Client.on_message(~filters.sticker & ~filters.via_bot & ~filters.edited & filters.me & filters.command(['tr', 'translate'], prefixes=config['config']['prefixes']))
|
|
@log_errors
|
|
@public_log_errors
|
|
async def translate(client, message):
|
|
reply = message.reply_to_message
|
|
if getattr(reply, 'empty', True):
|
|
await message.reply_text('Reply required')
|
|
return
|
|
text = reply.text or reply.caption
|
|
if not text:
|
|
await message.reply_text('Text required')
|
|
return
|
|
src_lang = 'auto'
|
|
dest_lang = 'en'
|
|
lang = ' '.join(message.command[1:]).lower()
|
|
for i in PROBLEM_CODES:
|
|
if lang.startswith(i):
|
|
lang = lang[len(i) + 1:]
|
|
if lang:
|
|
src_lang = i
|
|
dest_lang = lang
|
|
else:
|
|
dest_lang = i
|
|
break
|
|
else:
|
|
lang = lang.split('-', 1)
|
|
if len(lang) == 1 or not lang[-1]:
|
|
dest_lang = lang.pop(0) or dest_lang
|
|
else:
|
|
src_lang, dest_lang = lang
|
|
def _translate():
|
|
while True:
|
|
try:
|
|
# https://github.com/ssut/py-googletrans/issues/234#issuecomment-736314530
|
|
return googletrans.Translator(service_urls=['translate.googleapis.com']).translate(text, src=src_lang, dest=dest_lang)
|
|
except AttributeError:
|
|
time.sleep(0.5)
|
|
result = await client.loop.run_in_executor(None, _translate)
|
|
if result.text == text:
|
|
await message.reply_text('They\'re the same')
|
|
else:
|
|
text_ping = f'Translated from {result.src} to {result.dest}:\n{result.text[:4000]}'
|
|
text_pingnt = f'Translated from {result.src} to {result.dest}:\n{result.text.replace("@", "@" + ZWS)[:4000]}'
|
|
reply = await message.reply_text(text_pingnt, parse_mode=None, disable_web_page_preview=True)
|
|
if text_ping != text_pingnt:
|
|
await reply.edit_text(text_ping, parse_mode=None, disable_web_page_preview=True)
|
|
|
|
help_dict['translate'] = ('Translate',
|
|
'''{prefix}translate <i>(as reply to text)</i> <i>[src]-[dest]</i> - Translates text and stuff
|
|
Aliases: {prefix}tr''')
|