3 from StringIO import StringIO
4 from datetime import datetime, timedelta
5 from sfa.util.xrn import *
6 from sfa.util.plxrn import hostname_to_urn
7 from sfa.util.config import Config
8 from sfa.util.faults import SfaNotImplemented, InvalidRSpec
12 header = '<?xml version="1.0"?>\n'
16 def __init__(self, rspec="", namespaces={}):
18 self.parse_rspec(rspec, namespaces)
22 def create(self, type="advertisement"):
23 # eg. 2011-03-23T19:53:28Z
24 date_format = '%Y-%m-%dT%H:%M:%SZ'
25 now = datetime.utcnow()
26 generated_ts = now.strftime(date_format)
27 expires_ts = (now + timedelta(minutes=30)).strftime(date_format)
28 self.xml = etree.Element("rspec", type = type,
29 valid_until=expires_ts,
30 generated=generated_ts)
32 def parse_rspec(self, rspec, namespaces={}):
33 parser = etree.XMLParser(remove_blank_text=True)
35 tree = etree.parse(rspec, parser)
37 # 'rspec' file doesnt exist. 'rspec' is proably an xml string
39 tree = etree.parse(StringIO(rspec), parser)
41 raise InvalidRSpec('Must specify a xml file or xml string. Received: ' + rspec )
43 self.xml = tree.getroot()
45 self.namespaces = namespaces
47 def add_attribute(self, elem, name, value):
48 opt = etree.SubElement(elem, name)
51 def remove_attribute(self, elem, name, value):
53 opts = elem.iterfind(name)
63 return self.header + etree.tostring(self.xml, pretty_print=True)
65 def save(self, filename):
66 f = open(filename, 'w')
70 if __name__ == '__main__':