mirror of https://github.com/coddrago/Heroku
Update translate.py
parent
da363a1813
commit
4524b6858e
|
@ -56,7 +56,7 @@ class Translator(loader.Module):
|
|||
try:
|
||||
await utils.answer(
|
||||
message,
|
||||
await self.translate(
|
||||
await self._client.translate(
|
||||
message.peer_id,
|
||||
message,
|
||||
lang,
|
||||
|
@ -68,33 +68,3 @@ class Translator(loader.Module):
|
|||
logger.exception("Unable to translate text")
|
||||
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