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 objectes instanced by the specfied element_class.
  20         """
  21         if not element_class:
  22             element_class = Element
  23         if not fields:
  24            fields = element_class.fields.keys()
  25         elems = xml.xpath(xpath)
  26         objs = []
  27         for elem in elems:
  28             if not fields:
  29                 obj = element_class(elem.attrib, elem)
  30             else:
  31                 obj = element_class({}, elem)
  32                 for field in fields:
  33                     if field in elem.attrib:
  34                         obj[field] = elem.attrib[field]
  35             objs.append(obj)
  36         generic_elems = [element_class(elem.attrib, elem) for elem in elems]
  37         return objs
  38
  39
  40     @staticmethod
  41     def add_elements(xml, name, objs, fields=None):
  42         """
  43         Adds a child node to the specified xml node based on
  44         the specified name , element class and object.
  45         """
  46         if not isinstance(objs, list):
  47             objs = [objs]
  48         elems = []
  49         for obj in objs:
  50             if not obj:
  51                 continue
  52             if not fields:
  53                 fields = obj.keys()
  54             elem = xml.add_element(name)
  55             for field in fields:
  56                 if field in obj and obj[field]:
  57                     elem.set(field, unicode(obj[field]))
  58             elems.append(elem)
  59         return elems