152 lines
4.5 KiB
Python
152 lines
4.5 KiB
Python
"""
|
|
SleekXMPP: The Sleek XMPP Library
|
|
Copyright (C) 2010 Nathanael C. Fritz, Lance J.T. Stout
|
|
This file is part of SleekXMPP.
|
|
|
|
See the file LICENSE for copying permission.
|
|
"""
|
|
|
|
from sleekxmpp.xmlstream import ElementBase, register_stanza_plugin
|
|
|
|
|
|
class DiscoItems(ElementBase):
|
|
|
|
"""
|
|
Example disco#items stanzas:
|
|
<iq type="get">
|
|
<query xmlns="http://jabber.org/protocol/disco#items" />
|
|
</iq>
|
|
|
|
<iq type="result">
|
|
<query xmlns="http://jabber.org/protocol/disco#items">
|
|
<item jid="chat.example.com"
|
|
node="xmppdev"
|
|
name="XMPP Dev" />
|
|
<item jid="chat.example.com"
|
|
node="sleekdev"
|
|
name="SleekXMPP Dev" />
|
|
</query>
|
|
</iq>
|
|
|
|
Stanza Interface:
|
|
node -- The name of the node to either
|
|
query or return info from.
|
|
items -- A list of 3-tuples, where each tuple contains
|
|
the JID, node, and name of an item.
|
|
|
|
Methods:
|
|
add_item -- Add a single new item.
|
|
del_item -- Remove a single item.
|
|
get_items -- Return all items.
|
|
set_items -- Set or replace all items.
|
|
del_items -- Remove all items.
|
|
"""
|
|
|
|
name = 'query'
|
|
namespace = 'http://jabber.org/protocol/disco#items'
|
|
plugin_attrib = 'disco_items'
|
|
interfaces = set(('node', 'items'))
|
|
|
|
# Cache items
|
|
_items = set()
|
|
|
|
def setup(self, xml=None):
|
|
"""
|
|
Populate the stanza object using an optional XML object.
|
|
|
|
Overrides ElementBase.setup
|
|
|
|
Caches item information.
|
|
|
|
Arguments:
|
|
xml -- Use an existing XML object for the stanza's values.
|
|
"""
|
|
ElementBase.setup(self, xml)
|
|
self._items = set([item[0:2] for item in self['items']])
|
|
|
|
def add_item(self, jid, node=None, name=None):
|
|
"""
|
|
Add a new item element. Each item is required to have a
|
|
JID, but may also specify a node value to reference
|
|
non-addressable entitities.
|
|
|
|
Arguments:
|
|
jid -- The JID for the item.
|
|
node -- Optional additional information to reference
|
|
non-addressable items.
|
|
name -- Optional human readable name for the item.
|
|
"""
|
|
if (jid, node) not in self._items:
|
|
self._items.add((jid, node))
|
|
item = DiscoItem(parent=self)
|
|
item['jid'] = jid
|
|
item['node'] = node
|
|
item['name'] = name
|
|
self.iterables.append(item)
|
|
return True
|
|
return False
|
|
|
|
def del_item(self, jid, node=None):
|
|
"""
|
|
Remove a single item.
|
|
|
|
Arguments:
|
|
jid -- JID of the item to remove.
|
|
node -- Optional extra identifying information.
|
|
"""
|
|
if (jid, node) in self._items:
|
|
for item_xml in self.findall('{%s}item' % self.namespace):
|
|
item = (item_xml.attrib['jid'],
|
|
item_xml.attrib.get('node', None))
|
|
if item == (jid, node):
|
|
self.xml.remove(item_xml)
|
|
return True
|
|
return False
|
|
|
|
def get_items(self):
|
|
"""Return all items."""
|
|
items = set()
|
|
for item in self['substanzas']:
|
|
if isinstance(item, DiscoItem):
|
|
items.add((item['jid'], item['node'], item['name']))
|
|
return items
|
|
|
|
def set_items(self, items):
|
|
"""
|
|
Set or replace all items. The given items must be in a
|
|
list or set where each item is a tuple of the form:
|
|
(jid, node, name)
|
|
|
|
Arguments:
|
|
items -- A series of items in tuple format.
|
|
"""
|
|
self.del_items()
|
|
for item in items:
|
|
jid, node, name = item
|
|
self.add_item(jid, node, name)
|
|
|
|
def del_items(self):
|
|
"""Remove all items."""
|
|
self._items = set()
|
|
items = [i for i in self.iterables if isinstance(i, DiscoItem)]
|
|
for item in items:
|
|
self.xml.remove(item.xml)
|
|
self.iterables.remove(item)
|
|
|
|
|
|
class DiscoItem(ElementBase):
|
|
name = 'item'
|
|
namespace = 'http://jabber.org/protocol/disco#items'
|
|
plugin_attrib = name
|
|
interfaces = set(('jid', 'node', 'name'))
|
|
|
|
def get_node(self):
|
|
"""Return the item's node name or ``None``."""
|
|
return self._get_attr('node', None)
|
|
|
|
def get_name(self):
|
|
"""Return the item's human readable name, or ``None``."""
|
|
return self._get_attr('name', None)
|
|
|
|
|
|
register_stanza_plugin(DiscoItems, DiscoItem, iterable=True)
|