45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
|
from .. xmlstream.stanzabase import ElementBase, ET, JID
|
||
|
|
|
||
|
|
class Roster(ElementBase):
|
||
|
|
namespace = 'jabber:iq:roster'
|
||
|
|
name = 'query'
|
||
|
|
plugin_attrib = 'roster'
|
||
|
|
interfaces = set(('items',))
|
||
|
|
|
||
|
|
def setItems(self, items):
|
||
|
|
self.delItems()
|
||
|
|
for jid in items:
|
||
|
|
ijid = str(jid)
|
||
|
|
item = ET.Element('{jabber:iq:roster}item', {'jid': ijid})
|
||
|
|
if 'subscription' in items[jid]:
|
||
|
|
item.attrib['subscription'] = items[jid]['subscription']
|
||
|
|
if 'name' in items[jid]:
|
||
|
|
item.attrib['name'] = items[jid]['name']
|
||
|
|
if 'groups' in items[jid]:
|
||
|
|
for group in items[jid]['groups']:
|
||
|
|
groupxml = ET.Element('{jabber:iq:roster}group')
|
||
|
|
groupxml.text = group
|
||
|
|
item.append(groupxml)
|
||
|
|
self.xml.append(item)
|
||
|
|
return self
|
||
|
|
|
||
|
|
def getItems(self):
|
||
|
|
items = {}
|
||
|
|
itemsxml = self.xml.findall('{jabber:iq:roster}item')
|
||
|
|
if itemsxml is not None:
|
||
|
|
item = {}
|
||
|
|
for itemxml in itemsxml:
|
||
|
|
item['name'] = itemxml.get('name', '')
|
||
|
|
item['subscription'] = itemxml.get('subscription', '')
|
||
|
|
item['groups'] = []
|
||
|
|
groupsxml = itemxml.findall('{jabber:iq:roster}group')
|
||
|
|
if groupsxml is not None:
|
||
|
|
for groupxml in groupsxml:
|
||
|
|
item['groups'].append(groupxml.text)
|
||
|
|
items[JID(itemxml.get('jid'))] = item
|
||
|
|
return items
|
||
|
|
|
||
|
|
def delItems(self):
|
||
|
|
for child in self.xml.getchildren():
|
||
|
|
self.xml.remove(child)
|