X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Frspecs%2Felements%2Felement.py;h=7dcfa007a994370cc0076eb76ac0a047ab80a4d3;hb=4a9e6751f9f396f463932133b9d62fc925a99ef6;hp=abfb7fcc7234ade621592d23e77c7b5b27c765db;hpb=4385014c109d01ed15240e4557fbaae3c97aa618;p=sfa.git diff --git a/sfa/rspecs/elements/element.py b/sfa/rspecs/elements/element.py index abfb7fcc..7dcfa007 100644 --- a/sfa/rspecs/elements/element.py +++ b/sfa/rspecs/elements/element.py @@ -2,58 +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, self.fields) + 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 objectes instanced by the specfied element_class. - """ - if not element_class: - element_class = Element - if not fields: - fields = element_class.fields.keys() - 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, 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))