Misha1tigr
/
modular-bot-framework-for-telegram
forked from dymik739/modular-bot-framework-for-telegram
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request 'transliteration-decoder: add automatic photo caption detection and let bot core deal with exceptions' (#11) from dev into master 

Reviewed-on: dymik739/modular-bot-framework-for-telegram#11
This commit is contained in:
dymik739 2023-09-19 18:22:50 +03:00
parent 34439e944b 09def1cd63
commit 4502e7fbe1
1 changed files with 25 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
modules/transliteration-decoder/index.py
View File

@ -4,37 +4,36 @@ command_length = len(command)
def escaped_string(unescaped_string): def escaped_string(unescaped_string):
result_string = str(unescaped_string) result_string = str(unescaped_string)
for i in ["/", "<", ">"]: for i in ["<", ">"]:
result_string = result_string.replace(i, f"\\{i}") result_string = result_string.replace(i, f"\\{i}")
return result_string return result_string
if (command[0] in self.aliases) and (1 <= command_length <= 3): if (command[0] in self.aliases) and (1 <= command_length <= 3):
try: models = json.loads(readfile(self.path + "translate_models.json"))
models = json.loads(readfile(self.path + "translate_models.json"))
if command_length == 1: if command_length == 1:
chosen_model = "cz-ua" chosen_model = "cz-ua"
else: else:
chosen_model = command[1] chosen_model = command[1]
if command_length == 3: if command_length == 3:
text_to_decode = command[2] text_to_decode = command[2]
else: else:
text_to_decode = self.MESSAGE['reply_to_message']['text'] if self.MESSAGE['reply_to_message'].caption:
text_to_decode = self.MESSAGE['reply_to_message'].caption
decoded_text = text_to_decode elif self.MESSAGE['reply_to_message'].text:
if chosen_model not in models: text_to_decode = self.MESSAGE['reply_to_message'].text
self.RESPONSE = f"Такого варіанту транслітерації не існує. Доступні варіанти: " \
f"{', '.join(list(models.keys()))}"
else:
for i in models[chosen_model]:
decoded_text = decoded_text.replace(i[0], i[1])
decoded_text = decoded_text.replace(i[0].capitalize(), i[1].capitalize())
decoded_text = decoded_text.replace(i[0].upper(), i[1].upper())
self.RESPONSE = f"<b><u>Результат</u></b>\n{escaped_string(decoded_text)}"
self.FORMAT = "HTML"
except Exception as e: decoded_text = text_to_decode
print(f"[translit-decoder] Got exception: {e}") if chosen_model not in models:
self.RESPONSE = f"Такого варіанту транслітерації не існує. Доступні варіанти: " \
f"{', '.join(list(models.keys()))}"
else:
for i in models[chosen_model]:
decoded_text = decoded_text.replace(i[0], i[1])
decoded_text = decoded_text.replace(i[0].capitalize(), i[1].capitalize())
decoded_text = decoded_text.replace(i[0].upper(), i[1].upper())
self.RESPONSE = f"<b><u>Результат</u></b>\n{escaped_string(decoded_text)}"
self.FORMAT = "HTML"