misskey2telegram/modules/translate.py
2023-01-27 20:36:41 +08:00

35 lines
1.4 KiB
Python

from mipac.errors import NoSuchRenoteTargetError, APIError
from pyrogram import Client, filters
from pyrogram.types import CallbackQuery
from misskey_init import get_misskey_bot
from models.filters import timeline_filter
# translate:note_id
@Client.on_callback_query(filters.regex(r"^translate:(\w+)$") & timeline_filter)
async def translate_callback(_: Client, callback_query: CallbackQuery):
note_id = callback_query.matches[0].group(1)
try:
misskey_bot = get_misskey_bot(callback_query.from_user.id)
result = await misskey_bot.core.api.note.action.translate(
note_id=note_id,
target_lang="zh-CN",
)
if len(result.text) < 50:
await callback_query.answer(result.text, show_alert=True)
else:
await callback_query.message.reply(
f"翻译结果:<code>{result.text}</code>",
quote=True,
)
await callback_query.answer("翻译成功", show_alert=True)
except NoSuchRenoteTargetError:
await callback_query.answer("该嘟文不存在", show_alert=True)
except APIError:
return await callback_query.answer("该嘟文无法翻译,可能是无需翻译", show_alert=True)
except Exception as e:
if callback_query.message:
await callback_query.message.reply(f"翻译失败:{e}", quote=True)
await callback_query.answer("翻译失败", show_alert=True)