diff --git a/slixmpp/plugins/xep_0045/stanza.py b/slixmpp/plugins/xep_0045/stanza.py index 71eb3fb5..428dbce7 100644 --- a/slixmpp/plugins/xep_0045/stanza.py +++ b/slixmpp/plugins/xep_0045/stanza.py @@ -179,6 +179,21 @@ class MUCInvite(ElementBase): interfaces = {'to', 'from', 'reason'} sub_interfaces = {'reason'} + def get_to(self) -> JID: + return JID(self._get_attr('to')) + + def set_to(self, value: Union[JID, str]): + if not isinstance(value, JID): + value = JID(value) + self._set_attr('to', value) + + def get_from(self) -> JID: + return JID(self._get_attr('from')) + + def set_from(self, value: Union[JID, str]): + if not isinstance(value, JID): + value = JID(value) + self._set_attr('from', value) class MUCDecline(ElementBase): name = 'decline' @@ -187,6 +202,22 @@ class MUCDecline(ElementBase): interfaces = {'to', 'from', 'reason'} sub_interfaces = {'reason'} + def get_to(self) -> JID: + return JID(self._get_attr('to')) + + def set_to(self, value: Union[JID, str]): + if not isinstance(value, JID): + value = JID(value) + self._set_attr('to', value) + + def get_from(self) -> JID: + return JID(self._get_attr('from')) + + def set_from(self, value: Union[JID, str]): + if not isinstance(value, JID): + value = JID(value) + self._set_attr('from', value) + class MUCHistory(ElementBase): name = 'history'