5 from StringIO import StringIO
6 from optparse import OptionParser
9 def __init__(self, xml):
10 parser = etree.XMLParser(remove_blank_text=True)
11 tree = etree.parse(StringIO(xml), parser)
12 self.rspec = tree.getroot()
14 def get_node_element(self, hostname):
15 names = self.rspec.iterfind("./network/site/node/hostname")
17 if name.text == hostname:
18 return name.getparent()
21 def get_node_list(self):
22 result = self.rspec.xpath("./network/site/node/hostname/text()")
25 def get_sliver_list(self):
26 result = self.rspec.xpath("./network/site/node[sliver]/hostname/text()")
29 def add_sliver(self, hostname):
30 node = self.get_node_element(hostname)
31 etree.SubElement(node, "sliver")
33 def remove_sliver(self, hostname):
34 node = self.get_node_element(hostname)
35 node.remove(node.find("sliver"))
37 def attributes_list(self, elem):
41 opts.append((e.tag, e.text))
44 def get_default_sliver_attributes(self):
45 defaults = self.rspec.find(".//sliver_defaults")
46 return self.attributes_list(defaults)
48 def get_sliver_attributes(self, hostname):
49 node = self.get_node_element(hostname)
50 sliver = node.find("sliver")
51 return self.attributes_list(sliver)
53 def add_attribute(self, elem, name, value):
54 opt = etree.SubElement(elem, name)
57 def add_default_sliver_attribute(self, name, value):
58 defaults = self.rspec.find(".//sliver_defaults")
59 self.add_attribute(defaults, name, value)
61 def add_sliver_attribute(self, hostname, name, value):
62 node = self.get_node_element(hostname)
63 sliver = node.find("sliver")
64 self.add_attribute(sliver, name, value)
66 def remove_attribute(self, elem, name, value):
68 opts = elem.iterfind(name)
74 def remove_default_sliver_attribute(self, name, value):
75 defaults = self.rspec.find(".//sliver_defaults")
76 self.remove_attribute(defaults, name, value)
78 def remove_sliver_attribute(self, hostname, name, value):
79 node = self.get_node_element(hostname)
80 sliver = node.find("sliver")
81 self.remove_attribute(sliver, name, value)
83 def add_vlink(self, fromhost, tohost):
86 def remove_vlink(self, fromhost, tohost):
90 return etree.tostring(self.rspec, pretty_print=True,
96 def save(self, filename):
97 f = open(filename, "w")
103 def __init__(self, usage, description, epilog=None):
104 self.parser = OptionParser(usage=usage, description=description,
106 self.parser.add_option("-i", "", dest="infile", metavar="FILE",
107 help="read RSpec from FILE (default is stdin)")
108 self.parser.add_option("-o", "", dest="outfile", metavar="FILE",
109 help="write output to FILE (default is stdout)")
110 self.nodefile = False
113 def add_nodefile_option(self):
115 self.parser.add_option("-n", "", dest="nodefile",
117 help="read node list from FILE"),
119 def add_show_attributes_option(self):
120 self.parser.add_option("-s", "--show-attributes", action="store_true",
121 dest="showatt", default=False,
122 help="show sliver attributes")
124 def add_attribute_options(self):
125 self.parser.add_option("", "--capabilities",
126 metavar="<cap1,cap2,...>",
127 help="Vserver bcapabilities")
128 self.parser.add_option("", "--codemux", action="append",
129 metavar="<host,local-port>",
130 help="Demux HTTP between slices using " +
132 self.parser.add_option("", "--cpu-pct",
134 help="Reserved CPU percent (e.g., 25)")
135 self.parser.add_option("", "--cpu-share",
137 help="Number of CPU shares (e.g., 5)")
138 self.parser.add_option("", "--delegations",
139 metavar="<slice1,slice2,...>",
140 help="List of slices with delegation authority")
141 self.parser.add_option("", "--disk-max",
143 help="Disk quota (1k disk blocks)")
144 self.parser.add_option("", "--initscript",
146 help="Slice initialization script (e.g., stork)")
147 self.parser.add_option("", "--ip-addresses", action="append",
149 help="Add an IP address to a sliver")
150 self.parser.add_option("", "--net-i2-max-kbyte",
152 help="Maximum daily network Tx limit " +
154 self.parser.add_option("", "--net-i2-max_rate",
156 help="Maximum bandwidth over I2 routes")
157 self.parser.add_option("", "--net-i2-min-rate",
159 help="Minimum bandwidth over I2 routes")
160 self.parser.add_option("", "--net-i2-share",
162 help="Number of bandwidth shares over I2 routes")
163 self.parser.add_option("", "--net-i2-thresh-kbyte",
165 help="Limit sent to I2 hosts before warning, " +
167 self.parser.add_option("", "--net-max-kbyte",
169 help="Maximum daily network Tx limit " +
171 self.parser.add_option("", "--net-max_rate",
173 help="Maximum bandwidth over non-I2 routes")
174 self.parser.add_option("", "--net-min-rate",
176 help="Minimum bandwidth over non-I2 routes")
177 self.parser.add_option("", "--net-share",
179 help="Number of bandwidth shares over non-I2 " +
181 self.parser.add_option("", "--net-thresh-kbyte",
183 help="Limit sent to non-I2 hosts before " +
184 "warning, throttling")
185 self.parser.add_option("", "--vsys", action="append",
187 help="Vsys script (e.g., fd_fusemount)")
188 self.parser.add_option("", "--vsys-vnet",
189 metavar="<IP network>",
190 help="Allocate a virtual private network")
192 def get_attribute_dict(self):
193 attrlist = ['capabilities','codemux','cpu-pct','cpu-share',
194 'delegations','disk-max','initscript','ip-addresses',
195 'net-i2-max-kbyte','net-i2-max-rate','net-i2-min-rate',
196 'net-i2-share','net-i2-thresh-kbyte',
197 'net-max-kbyte','net-max-rate','net-min-rate',
198 'net-share','net-thresh-kbyte',
201 for attr in attrlist:
203 value = getattr(self.opts, attr, None)
204 if value is not None:
205 attrdict[name] = value
209 (self.opts, self.args) = self.parser.parse_args()
212 sys.stdin = open(self.opts.infile, "r")
213 xml = sys.stdin.read()
214 self.rspec = RSpec(xml)
217 if self.opts.nodefile:
218 f = open(self.opts.nodefile, "r")
219 self.nodes = f.read().split()
222 self.nodes = self.args
224 if self.opts.outfile:
225 sys.outfile = open(self.opts.outfile, "w")