PsychoticNinja/oddluck-limnoria-plugins
1
0
Fork 0
mirror of https://github.com/oddluck/limnoria-plugins.git synced 2025-04-25 12:31:07 -05:00
Code Issues Projects Releases Wiki Activity

Azure: add nativeName lookups.

Browse Source
This commit is contained in:
oddluck 2020-08-02 03:58:43 -04:00
parent 83c35b5b37
commit a7ed2b240a
2 changed files with 35 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Azure/config.py
View File

@ -103,7 +103,8 @@ conf.registerChannelValue(
_(
"""
The default reply template for the translate command.
Variables are $text, $sourceISO, $sourceName, $targetISO, $targetName.
Variables are $text, $sourceISO, $sourceName, $sourceNativeName,
$targetISO, $targetName, $targetNativeName.
"""
),
),

51
Azure/plugin.py
View File

@ -69,31 +69,41 @@ class Azure(callbacks.Plugin):
url = "https://api.cognitive.microsofttranslator.com/translate?api-version=3.0&"
if source != "auto":
if not self.languages.get(source):
for lang in self.languages:
if (
source.lower() in self.languages[lang]["name"].lower()
or source.lower() in self.languages[lang]["nativeName"].lower()
):
source = lang
break
if not self.languages.get(target):
for lang in self.languages:
if (
target.lower() in self.languages[lang]["name"].lower()
or target.lower() in self.languages[lang]["nativeName"].lower()
):
target = lang
break
if self.languages.get(source) and self.languages.get(target):
url += "to={0}&from={1}".format(target, source)
else:
for lang in self.languages:
if source.lower() in self.languages[lang]["name"].lower():
source = lang
if target.lower() in self.languages[lang]["name"].lower():
target = lang
if self.languages.get(source) and self.languages.get(target):
url += "to={0}&from={1}".format(target, source)
else:
irc.reply("Invalid language selection.")
return
irc.reply("Invalid language selection.")
return
else:
if not self.languages.get(target):
for lang in self.languages:
if (
target.lower() in self.languages[lang]["name"].lower()
or target.lower() in self.languages[lang]["nativeName"].lower()
):
target = lang
break
if self.languages.get(target):
url += "to={0}".format(target)
else:
for lang in self.languages:
if target.lower() in self.languages[lang]["name"].lower():
target = lang
if self.languages.get(target):
url += "to={0}".format(target)
else:
irc.reply("Invalid language selection.")
return
irc.reply("Invalid language selection.")
return
key = self.registryValue("translate.key")
headers = {"Ocp-Apim-Subscription-Key": key, "Content-type": "application/json"}
body = [{"text": text}]
@ -109,14 +119,19 @@ class Azure(callbacks.Plugin):
results = {
"text": result[0]["translations"][0]["text"],
"targetName": self.languages[target]["name"],
"targetNativeName": self.languages[target]["nativeName"],
"targetISO": target,
}
if result[0].get("detectedLanguage"):
results["sourceName"] = self.languages[
result[0]["detectedLanguage"]["language"]
]["name"]
results["sourceNativeName"] = self.languages[
result[0]["detectedLanguage"]["language"]
]["nativeName"]
else:
results["sourceName"] = self.languages[source]["name"]
results["sourceNativeName"] = self.languages[source]["nativeName"]
irc.reply(template.safe_substitute(results))
translate = wrap(translate, [getopts({"from": "text", "to": "text"}), "text"])