diff options
author | Link Mauve <linkmauve@linkmauve.fr> | 2020-12-07 21:21:43 +0100 |
---|---|---|
committer | Link Mauve <linkmauve@linkmauve.fr> | 2020-12-07 21:21:43 +0100 |
commit | c892fba7785df9f3e1de908c3b9cd4ea06865673 (patch) | |
tree | a2cbc4e2b5a227d0fb30350c031b1ead74805da0 /slixmpp | |
parent | 340b0f0f169221b31dbcab8d9937b427c55fe2da (diff) | |
parent | 758dfb930c9b1f89797a943e4055810386c3acfb (diff) | |
download | slixmpp-c892fba7785df9f3e1de908c3b9cd4ea06865673.tar.gz slixmpp-c892fba7785df9f3e1de908c3b9cd4ea06865673.tar.bz2 slixmpp-c892fba7785df9f3e1de908c3b9cd4ea06865673.tar.xz slixmpp-c892fba7785df9f3e1de908c3b9cd4ea06865673.zip |
Merge branch 'xep-0439-improved' into 'master'
XEP-0439: add events & tests
See merge request poezio/slixmpp!82
Diffstat (limited to 'slixmpp')
-rw-r--r-- | slixmpp/plugins/xep_0439/quickresponse.py | 42 |
1 files changed, 37 insertions, 5 deletions
diff --git a/slixmpp/plugins/xep_0439/quickresponse.py b/slixmpp/plugins/xep_0439/quickresponse.py index 74e299d6..f6836731 100644 --- a/slixmpp/plugins/xep_0439/quickresponse.py +++ b/slixmpp/plugins/xep_0439/quickresponse.py @@ -11,9 +11,12 @@ from typing import ( Tuple, ) -from slixmpp import JID +from slixmpp import JID, Message from slixmpp.plugins import BasePlugin from slixmpp.plugins.xep_0439 import stanza +from slixmpp.xmlstream.matcher import StanzaPath +from slixmpp.xmlstream.handler import Callback + class XEP_0439(BasePlugin): @@ -27,11 +30,40 @@ class XEP_0439(BasePlugin): def plugin_init(self) -> None: stanza.register_plugins() + self.xmpp.register_handler(Callback( + 'Action received', + StanzaPath('message/action'), + self._handle_action, + )) + self.xmpp.register_handler(Callback( + 'Response received', + StanzaPath('message/response'), + self._handle_response, + )) + self.xmpp.register_handler(Callback( + 'ActionSelected received', + StanzaPath('message/action_selected'), + self._handle_action_selected, + )) + + def plugin_send(self): + self.xmpp.remove_handler('Action received') + self.xmpp.remove_handler('Response received') + self.xmpp.remove_handler('ActionSelected received') + + def _handle_response(self, msg: Message): + self.xmpp.event('responses_received', msg) + + def _handle_action(self, msg: Message): + self.xmpp.event('action_received', msg) + + def _handle_action_selected(self, msg: Message): + self.xmpp.event('action_selected', msg) - def ask_for_responses(self, mto: JID, body: str, - responses: Iterable[Tuple[str, str]], - mtype: str = 'chat', lang: Optional[str] = None, *, - mfrom: Optional[JID] = None): + def ask_for_response(self, mto: JID, body: str, + responses: Iterable[Tuple[str, str]], + mtype: str = 'chat', lang: Optional[str] = None, *, + mfrom: Optional[JID] = None): """ Send a message with a set of responses. |