e8b2dd6698
get_roster() now returns the Iq result stanza instead of True (stanzas also evaluate to True).
127 lines
3.8 KiB
Python
127 lines
3.8 KiB
Python
"""
|
|
SleekXMPP: The Sleek XMPP Library
|
|
Copyright (C) 2010 Nathanael C. Fritz
|
|
This file is part of SleekXMPP.
|
|
|
|
See the file LICENSE for copying permission.
|
|
"""
|
|
|
|
from sleekxmpp.stanza import Iq
|
|
from sleekxmpp.xmlstream import JID
|
|
from sleekxmpp.xmlstream import ET, ElementBase, register_stanza_plugin
|
|
|
|
|
|
class Roster(ElementBase):
|
|
|
|
"""
|
|
Example roster stanzas:
|
|
<iq type="set">
|
|
<query xmlns="jabber:iq:roster">
|
|
<item jid="user@example.com" subscription="both" name="User">
|
|
<group>Friends</group>
|
|
</item>
|
|
</query>
|
|
</iq>
|
|
|
|
Stanza Inteface:
|
|
items -- A dictionary of roster entries contained
|
|
in the stanza.
|
|
|
|
Methods:
|
|
get_items -- Return a dictionary of roster entries.
|
|
set_items -- Add <item> elements.
|
|
del_items -- Remove all <item> elements.
|
|
"""
|
|
|
|
namespace = 'jabber:iq:roster'
|
|
name = 'query'
|
|
plugin_attrib = 'roster'
|
|
interfaces = set(('items',))
|
|
|
|
def set_items(self, items):
|
|
"""
|
|
Set the roster entries in the <roster> stanza.
|
|
|
|
Uses a dictionary using JIDs as keys, where each entry is itself
|
|
a dictionary that contains:
|
|
name -- An alias or nickname for the JID.
|
|
subscription -- The subscription type. Can be one of 'to',
|
|
'from', 'both', 'none', or 'remove'.
|
|
groups -- A list of group names to which the JID
|
|
has been assigned.
|
|
|
|
Arguments:
|
|
items -- A dictionary of roster entries.
|
|
"""
|
|
self.del_items()
|
|
for jid in items:
|
|
item = RosterItem()
|
|
item.values = items[jid]
|
|
item['jid'] = jid
|
|
self.append(item)
|
|
return self
|
|
|
|
def get_items(self):
|
|
"""
|
|
Return a dictionary of roster entries.
|
|
|
|
Each item is keyed using its JID, and contains:
|
|
name -- An assigned alias or nickname for the JID.
|
|
subscription -- The subscription type. Can be one of 'to',
|
|
'from', 'both', 'none', or 'remove'.
|
|
groups -- A list of group names to which the JID has
|
|
been assigned.
|
|
"""
|
|
items = {}
|
|
for item in self['substanzas']:
|
|
if isinstance(item, RosterItem):
|
|
items[item['jid']] = item.values
|
|
# Remove extra JID reference to keep everything
|
|
# backward compatible
|
|
del items[item['jid']]['jid']
|
|
return items
|
|
|
|
def del_items(self):
|
|
"""
|
|
Remove all <item> elements from the roster stanza.
|
|
"""
|
|
for item in self['substanzas']:
|
|
if isinstance(item, RosterItem):
|
|
self.xml.remove(item.xml)
|
|
|
|
|
|
class RosterItem(ElementBase):
|
|
namespace = 'jabber:iq:roster'
|
|
name = 'item'
|
|
plugin_attrib = 'item'
|
|
interfaces = set(('jid', 'name', 'subscription', 'ask',
|
|
'approved', 'groups'))
|
|
|
|
def get_groups(self):
|
|
groups = []
|
|
for group in self.xml.findall('{%s}group' % self.namespace):
|
|
groups.append(group.text)
|
|
return groups
|
|
|
|
def set_groups(self, values):
|
|
self.del_groups()
|
|
for group in values:
|
|
group_xml = ET.Element('{%s}group' % self.namespace)
|
|
group_xml.text = group
|
|
self.xml.append(group_xml)
|
|
|
|
def del_groups(self):
|
|
for group in self.xml.findall('{%s}group' % self.namespace):
|
|
self.xmp.remove(group)
|
|
|
|
|
|
|
|
|
|
register_stanza_plugin(Iq, Roster)
|
|
register_stanza_plugin(Roster, RosterItem, iterable=True)
|
|
|
|
# To comply with PEP8, method names now use underscores.
|
|
# Deprecated method names are re-mapped for backwards compatibility.
|
|
Roster.setItems = Roster.set_items
|
|
Roster.getItems = Roster.get_items
|
|
Roster.delItems = Roster.del_items
|