sfa/rspecs/elements/element.py

   1 class Element(dict):
   2
   3     fields = {}
   4
   5     def __init__(self, fields={}, element=None, keys=None):
   6         self.element = element
   7         dict.__init__(self, dict.fromkeys(self.fields))
   8         if not keys:
   9             keys = fields.keys()
  10         for key in keys:
  11             if key in fields:
  12                 self[key] = fields[key]
  13
  14     @staticmethod
  15     def get_elements(xml, xpath, element_class=None, fields=None):
  16         """
  17         Search the specifed xml node for elements that match the
  18         specified xpath query.
  19         Returns a list of objects instanced by the specified element_class.
  20         """
  21         if not element_class:
  22             element_class = Element
  23 # xxx this obviously is wrong...
  24 #        if not fields and hasattr(element_class, fields):
  25 #           fields = element_class.fields
  26         elems = xml.xpath(xpath)
  27         objs = []
  28         for elem in elems:
  29             if not fields:
  30                 obj = element_class(elem.attrib, elem)
  31             else:
  32                 obj = element_class({}, elem)
  33                 for field in fields:
  34                     if field in elem.attrib:
  35                         obj[field] = elem.attrib[field]
  36             objs.append(obj)
  37         generic_elems = [element_class(elem.attrib, elem) for elem in elems]
  38         return objs
  39
  40
  41     @staticmethod
  42     def add_elements(xml, name, objs, fields=None):
  43         """
  44         Adds a child node to the specified xml node based on
  45         the specified name , element class and object.
  46         """
  47         if not isinstance(objs, list):
  48             objs = [objs]
  49         elems = []
  50         for obj in objs:
  51             if not obj:
  52                 continue
  53             if not fields:
  54                 fields = obj.keys()
  55             elem = xml.add_element(name)
  56             for field in fields:
  57                 if field in obj and obj[field]:
  58                     elem.set(field, unicode(obj[field]))
  59             elems.append(elem)
  60         return elems