xep_0077_3.py

from .. xmlstream.stanzabase import ElementBase, ET, JID
from .. stanza.iq import Iq

class Registration(ElementBase):
    namespace = 'jabber:iq:register'
    name = 'query'
    plugin_attrib = 'register'
    interfaces = set(('username', 'password', 'registered', 'remove'))
    sub_interfaces = interfaces

    def getRegistered(self):
        present = self.xml.find('{%s}registered' % self.namespace)
        return True if present is not None else False

    def getRemove(self):
        present = self.xml.find('{%s}remove' % self.namespace)
        return True if present is not None else False

    def setRegistered(self, registered):
        if registered:
            self.addField('registered')
        else:
            del self['registered']

    def setRemove(self, remove):
        if remove:
            self.addField('remove')
        else:
            del self['remove']

    def addField(self, name):
        itemXML = ET.Element('{%s}%s' % (self.namespace, name))
        self.xml.append(itemXML)