Mercurial > libervia-backend
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