mirror of https://github.com/coddrago/Heroku
Update translate.py
parent
da363a1813
commit
4524b6858e
|
@ -56,7 +56,7 @@ class Translator(loader.Module):
|
||||||
try:
|
try:
|
||||||
await utils.answer(
|
await utils.answer(
|
||||||
message,
|
message,
|
||||||
await self.translate(
|
await self._client.translate(
|
||||||
message.peer_id,
|
message.peer_id,
|
||||||
message,
|
message,
|
||||||
lang,
|
lang,
|
||||||
|
@ -68,33 +68,3 @@ class Translator(loader.Module):
|
||||||
logger.exception("Unable to translate text")
|
logger.exception("Unable to translate text")
|
||||||
await utils.answer(message, self.strings("error"))
|
await utils.answer(message, self.strings("error"))
|
||||||
|
|
||||||
async def translate(self, peer, message, to_lang, raw_text, entities) -> str:
|
|
||||||
msg_id = herokutl.utils.get_message_id(message) or 0
|
|
||||||
if not msg_id:
|
|
||||||
return None
|
|
||||||
|
|
||||||
if not isinstance(message, types.Message):
|
|
||||||
message = (await self.get_messages(peer, ids=[msg_id]))[0]
|
|
||||||
|
|
||||||
result = await self._client(
|
|
||||||
functions.messages.TranslateTextRequest(
|
|
||||||
peer=peer,
|
|
||||||
id=[msg_id],
|
|
||||||
text=[
|
|
||||||
types.TextWithEntities(
|
|
||||||
raw_text or message.raw_text,
|
|
||||||
entities or message.entities or [],
|
|
||||||
)
|
|
||||||
],
|
|
||||||
to_lang=to_lang,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
return (
|
|
||||||
extensions.html.unparse(
|
|
||||||
result.result[0].text,
|
|
||||||
result.result[0].entities,
|
|
||||||
)
|
|
||||||
if result and result.result
|
|
||||||
else ""
|
|
||||||
)
|
|
Loading…
Reference in New Issue