sleekxmpp/plugins/xep_0128/static.py

   1 """
   2     SleekXMPP: The Sleek XMPP Library
   3     Copyright (C) 2010 Nathanael C. Fritz, Lance J.T. Stout
   4     This file is part of SleekXMPP.
   5
   6     See the file LICENSE for copying permission.
   7 """
   8
   9 import logging
  10
  11 import sleekxmpp
  12 from sleekxmpp.plugins.xep_0030 import StaticDisco
  13
  14
  15 log = logging.getLogger(__name__)
  16
  17
  18 class StaticExtendedDisco(object):
  19
  20     """
  21     Extend the default StaticDisco implementation to provide
  22     support for extended identity information.
  23     """
  24
  25     def __init__(self, static):
  26         """
  27         Augment the default XEP-0030 static handler object.
  28
  29         Arguments:
  30             static -- The default static XEP-0030 handler object.
  31         """
  32         self.static = static
  33
  34     def set_extended_info(self, jid, node, ifrom, data):
  35         """
  36         Replace the extended identity data for a JID/node combination.
  37
  38         The data parameter may provide:
  39             data -- Either a single data form, or a list of data forms.
  40         """
  41         with self.static.lock:
  42             self.del_extended_info(jid, node, ifrom, data)
  43             self.add_extended_info(jid, node, ifrom, data)
  44
  45     def add_extended_info(self, jid, node, ifrom, data):
  46         """
  47         Add additional extended identity data for a JID/node combination.
  48
  49         The data parameter may provide:
  50             data -- Either a single data form, or a list of data forms.
  51         """
  52         with self.static.lock:
  53             self.static.add_node(jid, node)
  54
  55             forms = data.get('data', [])
  56             if not isinstance(forms, list):
  57                 forms = [forms]
  58
  59             info = self.static.get_node(jid, node)['info']
  60             for form in forms:
  61                 info.append(form)
  62
  63     def del_extended_info(self, jid, node, ifrom, data):
  64         """
  65         Replace the extended identity data for a JID/node combination.
  66
  67         The data parameter is not used.
  68         """
  69         with self.static.lock:
  70             if self.static.node_exists(jid, node):
  71                 info = self.static.get_node(jid, node)['info']
  72                 for form in info['substanza']:
  73                     info.xml.remove(form.xml)