From 072f0cc723be59dabff11b789c71f8cfb8dcdc14 Mon Sep 17 00:00:00 2001 From: Goffi Date: Thu, 23 Mar 2023 15:39:48 +0100 Subject: [PATCH] plugin lang detect: don't detect the language if the body is empty --- sat/plugins/plugin_exp_lang_detect.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) 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