diff sat/plugins/plugin_exp_lang_detect.py @ 2624:56f94936df1e

code style reformatting using black
author Goffi <goffi@goffi.org>
date Wed, 27 Jun 2018 20:14:46 +0200
parents 26edcf3a30eb
children 003b8b4b56a7
line wrap: on
line diff
--- a/sat/plugins/plugin_exp_lang_detect.py	Wed Jun 27 07:51:29 2018 +0200
+++ b/sat/plugins/plugin_exp_lang_detect.py	Wed Jun 27 20:14:46 2018 +0200
@@ -20,13 +20,16 @@
 from sat.core.i18n import _, D_
 from sat.core.constants import Const as C
 from sat.core.log import getLogger
+
 log = getLogger(__name__)
 from sat.core import exceptions
 
 try:
     from langid.langid import LanguageIdentifier, model
 except ImportError:
-    raise exceptions.MissingModule(u'Missing module langid, please download/install it with "pip install langid")')
+    raise exceptions.MissingModule(
+        u'Missing module langid, please download/install it with "pip install langid")'
+    )
 
 identifier = LanguageIdentifier.from_modelstring(model, norm_probs=False)
 
@@ -39,7 +42,7 @@
     C.PI_DEPENDENCIES: [],
     C.PI_MAIN: "LangDetect",
     C.PI_HANDLER: "no",
-    C.PI_DESCRIPTION: _("""Detect and set message language when unknown""")
+    C.PI_DESCRIPTION: _("""Detect and set message language when unknown"""),
 }
 
 CATEGORY = D_(u"Misc")
@@ -53,14 +56,12 @@
     </category>
     </individual>
     </params>
-    """.format(category_name=CATEGORY,
-               name=NAME,
-               label=_(LABEL),
-              )
+    """.format(
+    category_name=CATEGORY, name=NAME, label=_(LABEL)
+)
 
 
 class LangDetect(object):
-
     def __init__(self, host):
         log.info(_(u"Language detection plugin initialization"))
         self.host = host
@@ -69,8 +70,8 @@
         host.trigger.add("sendMessage", self.MessageSendTrigger)
 
     def addLanguage(self, mess_data):
-        message = mess_data['message']
-        if len(message) == 1 and message.keys()[0] == '':
+        message = mess_data["message"]
+        if len(message) == 1 and message.keys()[0] == "":
             msg = message.values()[0]
             lang = identifier.classify(msg)[0]
             mess_data["message"] = {lang: msg}
@@ -79,13 +80,17 @@
     def MessageReceivedTrigger(self, client, message_elt, post_treat):
         """ Check if source is linked and repeat message, else do nothing  """
 
-        lang_detect = self.host.memory.getParamA(NAME, CATEGORY, profile_key=client.profile)
+        lang_detect = self.host.memory.getParamA(
+            NAME, CATEGORY, profile_key=client.profile
+        )
         if lang_detect:
             post_treat.addCallback(self.addLanguage)
         return True
 
     def MessageSendTrigger(self, client, data, pre_xml_treatments, post_xml_treatments):
-        lang_detect = self.host.memory.getParamA(NAME, CATEGORY, profile_key=client.profile)
+        lang_detect = self.host.memory.getParamA(
+            NAME, CATEGORY, profile_key=client.profile
+        )
         if lang_detect:
             self.addLanguage(data)
         return True