erebion/libervia-backend
Watch
1
0
Fork
You've already forked libervia-backend
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

plugin lang detect: don't detect the language if the body is empty

Browse source
This commit is contained in:
Goffi 2023-03-23 15:39:48 +01:00
commit 072f0cc723
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
sat/plugins/plugin_exp_lang_detect.py
View file

@ -69,12 +69,13 @@ class LangDetect(object):
host.trigger.add("messageReceived", self.messageReceivedTrigger)
host.trigger.add("sendMessage", self.MessageSendTrigger)
def addLanguage(self, mess_data):
def add_language(self, mess_data):
message = mess_data["message"]
if len(message) == 1 and list(message.keys())[0] == "":
msg = list(message.values())[0]
lang = identifier.classify(msg)[0]
mess_data["message"] = {lang: msg}
msg = list(message.values())[0].strip()
if msg:
lang = identifier.classify(msg)[0]
mess_data["message"] = {lang: msg}
return mess_data
def messageReceivedTrigger(self, client, message_elt, post_treat):
@ -84,7 +85,7 @@ class LangDetect(object):
NAME, CATEGORY, profile_key=client.profile
)
if lang_detect:
post_treat.addCallback(self.addLanguage)
post_treat.addCallback(self.add_language)
return True
def MessageSendTrigger(self, client, data, pre_xml_treatments, post_xml_treatments):
@ -92,5 +93,5 @@ class LangDetect(object):
NAME, CATEGORY, profile_key=client.profile
)
if lang_detect:
self.addLanguage(data)
self.add_language(data)
return True