12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273 |
- """
- 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.
- """
- import logging
- import sleekxmpp
- from sleekxmpp.plugins.xep_0030 import StaticDisco
- log = logging.getLogger(__name__)
- class StaticExtendedDisco(object):
- """
- Extend the default StaticDisco implementation to provide
- support for extended identity information.
- """
- def __init__(self, static):
- """
- Augment the default XEP-0030 static handler object.
- Arguments:
- static -- The default static XEP-0030 handler object.
- """
- self.static = static
- def set_extended_info(self, jid, node, ifrom, data):
- """
- Replace the extended identity data for a JID/node combination.
- The data parameter may provide:
- data -- Either a single data form, or a list of data forms.
- """
- with self.static.lock:
- self.del_extended_info(jid, node, ifrom, data)
- self.add_extended_info(jid, node, ifrom, data)
- def add_extended_info(self, jid, node, ifrom, data):
- """
- Add additional extended identity data for a JID/node combination.
- The data parameter may provide:
- data -- Either a single data form, or a list of data forms.
- """
- with self.static.lock:
- self.static.add_node(jid, node)
- forms = data.get('data', [])
- if not isinstance(forms, list):
- forms = [forms]
- info = self.static.get_node(jid, node)['info']
- for form in forms:
- info.append(form)
- def del_extended_info(self, jid, node, ifrom, data):
- """
- Replace the extended identity data for a JID/node combination.
- The data parameter is not used.
- """
- with self.static.lock:
- if self.static.node_exists(jid, node):
- info = self.static.get_node(jid, node)['info']
- for form in info['substanza']:
- info.xml.remove(form.xml)
|