diff --git a/sat/plugins/plugin_exp_lang_detect.py b/sat/plugins/plugin_exp_lang_detect.py index faa9d4af..5321927d 100644 --- a/sat/plugins/plugin_exp_lang_detect.py +++ b/sat/plugins/plugin_exp_lang_detect.py @@ -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