diff options
Diffstat (limited to 'slixmpp/features/feature_session/stanza.py')
-rw-r--r-- | slixmpp/features/feature_session/stanza.py | 18 |
1 files changed, 16 insertions, 2 deletions
diff --git a/slixmpp/features/feature_session/stanza.py b/slixmpp/features/feature_session/stanza.py index f68483d6..4ed8c4db 100644 --- a/slixmpp/features/feature_session/stanza.py +++ b/slixmpp/features/feature_session/stanza.py @@ -6,7 +6,7 @@ See the file LICENSE for copying permission. """ -from slixmpp.xmlstream import ElementBase +from slixmpp.xmlstream import ElementBase, ET class Session(ElementBase): @@ -16,5 +16,19 @@ class Session(ElementBase): name = 'session' namespace = 'urn:ietf:params:xml:ns:xmpp-session' - interfaces = set() + interfaces = {'optional'} plugin_attrib = 'session' + + def get_optional(self): + return self.xml.find('{%s}optional' % self.namespace) is not None + + def set_optional(self, value): + if value: + optional = ET.Element('{%s}optional' % self.namespace) + self.xml.append(optional) + else: + self.del_optional() + + def del_optional(self): + optional = self.xml.find('{%s}optional' % self.namespace) + self.xml.remove(optional) |