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 get_network(self):
48 raise SfaNotImplemented()
50 def get_nodes(self, nodes_with_slivers=False):
51 raise SfaNotImplemented()
53 def add_nodes(self, nodes, check_for_dupes=False):
54 raise SfaNotImplemented()
56 def add_slivers(self, slivers, check_for_dupes=False):
57 raise SfaNotImplemented()
59 def add_links(self, links, check_for_dupes=False):
60 raise SfaNotImplemented()
62 def add_attribute(self, elem, name, value):
63 opt = etree.SubElement(elem, name)
66 def remove_attribute(self, elem, name, value):
68 opts = elem.iterfind(name)
78 return self.header + etree.tostring(self.xml, pretty_print=True)
80 def save(self, filename):
81 f = open(filename, 'w')
85 if __name__ == '__main__':