X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Frspecs%2Felements%2Felement.py;h=7dcfa007a994370cc0076eb76ac0a047ab80a4d3;hb=4a9e6751f9f396f463932133b9d62fc925a99ef6;hp=a841323ce1d21620c5b24216f92b6ed27cd2bd16;hpb=4a8462f577f803f53c0b61840faaab5c803770fa;p=sfa.git diff --git a/sfa/rspecs/elements/element.py b/sfa/rspecs/elements/element.py index a841323c..7dcfa007 100644 --- a/sfa/rspecs/elements/element.py +++ b/sfa/rspecs/elements/element.py @@ -2,59 +2,22 @@ class Element(dict): fields = {} - def __init__(self, fields={}, element=None, keys=None): + def __init__(self, fields=None, element=None, keys=None): + if fields is None: + fields = {} self.element = element dict.__init__(self, dict.fromkeys(self.fields)) if not keys: - keys = fields.keys() + keys = list(fields.keys()) for key in keys: if key in fields: - self[key] = fields[key] + self[key] = fields[key] - @staticmethod - def get_elements(xml, xpath, element_class=None, fields=None): - """ - Search the specifed xml node for elements that match the - specified xpath query. - Returns a list of objects instanced by the specified element_class. - """ - if not element_class: - element_class = Element -# xxx this obviously is wrong... -# if not fields and hasattr(element_class, fields): -# fields = element_class.fields - elems = xml.xpath(xpath) - objs = [] - for elem in elems: - if not fields: - obj = element_class(elem.attrib, elem) - else: - obj = element_class({}, elem) - for field in fields: - if field in elem.attrib: - obj[field] = elem.attrib[field] - objs.append(obj) - generic_elems = [element_class(elem.attrib, elem) for elem in elems] - return objs - - - @staticmethod - def add_elements(xml, name, objs, fields=None): - """ - Adds a child node to the specified xml node based on - the specified name , element class and object. - """ - if not isinstance(objs, list): - objs = [objs] - elems = [] - for obj in objs: - if not obj: - continue - if not fields: - fields = obj.keys() - elem = xml.add_element(name) - for field in fields: - if field in obj and obj[field]: - elem.set(field, unicode(obj[field])) - elems.append(elem) - return elems + def __getattr__(self, name): + if hasattr(self.__dict__, name): + return getattr(self.__dict__, name) + elif hasattr(self.element, name): + return getattr(self.element, name) + else: + raise AttributeError("class Element of type {} has no attribute {}" + .format(self.__class__.__name__, name))