From 11058bec8a18a5207514d7587113f7386fbd30fa Mon Sep 17 00:00:00 2001 From: Goffi Date: Sat, 11 Aug 2018 18:24:55 +0200 Subject: [PATCH] plugin XEP-0334: added a method to add hint directly to elements (i.e. on domish.Element instead of message data) --- sat/plugins/plugin_xep_0334.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/sat/plugins/plugin_xep_0334.py b/sat/plugins/plugin_xep_0334.py index 2081540e..081d961c 100644 --- a/sat/plugins/plugin_xep_0334.py +++ b/sat/plugins/plugin_xep_0334.py @@ -85,10 +85,18 @@ class XEP_0334(object): else: log.error(u"Unknown hint: {}".format(hint)) + def addHintElements(self, message_elt, hints): + """Add hints elements to message stanza + + @param message_elt(domish.Element): stanza where hints must be added + @param hints(iterable(unicode)): hints to add + """ + for hint in hints: + message_elt.addElement((NS_HINTS, hint)) + def _sendPostXmlTreatment(self, mess_data): if "hints" in mess_data: - for hint in mess_data["hints"]: - mess_data[u"xml"].addElement((NS_HINTS, hint)) + self.addHintElements(mess_data[u"xml"], mess_data[u"hints"]) return mess_data def sendMessageTrigger(