sleekxmpp/features/feature_mechanisms/stanza/mechanisms.py

   1 """
   2     SleekXMPP: The Sleek XMPP Library
   3     Copyright (C) 2011  Nathanael C. Fritz
   4     This file is part of SleekXMPP.
   5
   6     See the file LICENSE for copying permission.
   7 """
   8
   9 from sleekxmpp.xmlstream import ElementBase, ET
  10
  11
  12 class Mechanisms(ElementBase):
  13
  14     """
  15     """
  16
  17     name = 'mechanisms'
  18     namespace = 'urn:ietf:params:xml:ns:xmpp-sasl'
  19     interfaces = set(('mechanisms', 'required'))
  20     plugin_attrib = name
  21     is_extension = True
  22
  23     def get_required(self):
  24         """
  25         """
  26         return True
  27
  28     def get_mechanisms(self):
  29         """
  30         """
  31         results = []
  32         mechs = self.findall('{%s}mechanism' % self.namespace)
  33         if mechs:
  34             for mech in mechs:
  35                 results.append(mech.text)
  36         return results
  37
  38     def set_mechanisms(self, values):
  39         """
  40         """
  41         self.del_mechanisms()
  42         for val in values:
  43             mech = ET.Element('{%s}mechanism' % self.namespace)
  44             mech.text = val
  45             self.append(mech)
  46
  47     def del_mechanisms(self):
  48         """
  49         """
  50         mechs = self.findall('{%s}mechanism' % self.namespace)
  51         if mechs:
  52             for mech in mechs:
  53                 self.xml.remove(mech)