transliteration-decoder: add !dc command trigger, allow cz-ua model to be default and make code more readable

This commit is contained in:
dymik739 2023-04-26 13:12:15 +03:00
parent c4f6e952bc
commit 45b0a5973c
3 changed files with 16 additions and 6 deletions

View File

@ -1,11 +1,18 @@
if self.MESSAGE['text'].split()[0] == "!decode" and (2 <= len(self.MESSAGE['text'].split(" ", 2)) <= 3): cmd = self.MESSAGE['text'].split(" ", 2)
l = len(cmd)
if (cmd[0] in self.aliases) and (1 <= l <= 3):
try: try:
models = json.loads(readfile(self.path + "translate_models.json")) models = json.loads(readfile(self.path + "translate_models.json"))
chosen_model = self.MESSAGE['text'].split(" ", 2)[1]
if len(self.MESSAGE['text'].split(" ", 2)) == 3: if l == 1:
t = self.MESSAGE['text'].split(" ", 2)[2] chosen_model = "cz-ua"
elif len(self.MESSAGE['text'].split(" ", 2)) == 2: else:
chosen_model = cmd[1]
if l == 3:
t = cmd[2]
else:
t = self.MESSAGE['reply_to_message']['text'] t = self.MESSAGE['reply_to_message']['text']
translated_t = t translated_t = t
@ -18,5 +25,6 @@ if self.MESSAGE['text'].split()[0] == "!decode" and (2 <= len(self.MESSAGE['text
translated_t = translated_t.replace(i[0].upper(), i[1].upper()) translated_t = translated_t.replace(i[0].upper(), i[1].upper())
self.RESPONCE = f"Результат: {translated_t}" self.RESPONCE = f"Результат: {translated_t}"
except Exception as e: except Exception as e:
print(f"[translit-decoder] Got exception: {e}") print(f"[translit-decoder] Got exception: {e}")

View File

@ -2,5 +2,6 @@
"version": 1, "version": 1,
"index_file": "index.py", "index_file": "index.py",
"start_on_boot": true, "start_on_boot": true,
"alias": "translit-decoder" "alias": "translit-decoder",
"predefine": "predefine.py"
} }

View File

@ -0,0 +1 @@
self.aliases = ["!decode", "!dc"]