sfa/util/rspecHelper.py

   1 #! /usr/bin/env python
   2
   3 import sys
   4 from lxml import etree
   5 from StringIO import StringIO
   6 from optparse import OptionParser
   7
   8 class RSpec:
   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()
  13
  14     def get_node_element(self, hostname):
  15         names = self.rspec.iterfind("./network/site/node/hostname")
  16         for name in names:
  17             if name.text == hostname:
  18                 return name.getparent()
  19         return None
  20
  21     def get_node_list(self):
  22         result = self.rspec.xpath("./network/site/node/hostname/text()")
  23         return result
  24
  25     def get_sliver_list(self):
  26         result = self.rspec.xpath("./network/site/node[sliver]/hostname/text()")
  27         return result
  28
  29     def add_sliver(self, hostname):
  30         node = self.get_node_element(hostname)
  31         etree.SubElement(node, "sliver")
  32
  33     def remove_sliver(self, hostname):
  34         node = self.get_node_element(hostname)
  35         node.remove(node.find("sliver"))
  36
  37     def attributes_list(self, elem):
  38         opts = []
  39         if elem is not None:
  40             for e in elem:
  41                 opts.append((e.tag, e.text))
  42         return opts
  43
  44     def get_default_sliver_attributes(self):
  45         defaults = self.rspec.find(".//sliver_defaults")
  46         return self.attributes_list(defaults)
  47
  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)
  52
  53     def add_attribute(self, elem, name, value):
  54         opt = etree.SubElement(elem, name)
  55         opt.text = value
  56
  57     def add_default_sliver_attribute(self, name, value):
  58         defaults = self.rspec.find(".//sliver_defaults")
  59         self.add_attribute(defaults, name, value)
  60
  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)
  65
  66     def remove_attribute(self, elem, name, value):
  67         if elem is not None:
  68             opts = elem.iterfind(name)
  69             if opts is not None:
  70                 for opt in opts:
  71                     if opt.text == value:
  72                         elem.remove(opt)
  73
  74     def remove_default_sliver_attribute(self, name, value):
  75         defaults = self.rspec.find(".//sliver_defaults")
  76         self.remove_attribute(defaults, name, value)
  77
  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)
  82
  83     def add_vlink(self, fromhost, tohost):
  84         pass
  85
  86     def remove_vlink(self, fromhost, tohost):
  87         pass
  88
  89     def toxml(self):
  90         return etree.tostring(self.rspec, pretty_print=True,
  91                               xml_declaration=True)
  92
  93     def __str__(self):
  94         return self.toxml()
  95
  96     def save(self, filename):
  97         f = open(filename, "w")
  98         f.write(self.toxml())
  99         f.close()
 100
 101
 102 class Commands:
 103     def __init__(self, usage, description, epilog=None):
 104         self.parser = OptionParser(usage=usage, description=description,
 105                                    epilog=epilog)
 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
 111         self.attributes = {}
 112
 113     def add_nodefile_option(self):
 114         self.nodefile = True
 115         self.parser.add_option("-n", "", dest="nodefile",
 116                                metavar="FILE",
 117                                help="read node list from FILE"),
 118
 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")
 123
 124     def add_attribute_options(self):
 125         self.parser.add_option("", "--capabilities", action="append",
 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 " +
 131                                "localhost ports")
 132         self.parser.add_option("", "--cpu-pct", action="append",
 133                                metavar="<num>",
 134                                help="Reserved CPU percent (e.g., 25)")
 135         self.parser.add_option("", "--cpu-share", action="append",
 136                                metavar="<num>",
 137                                help="Number of CPU shares (e.g., 5)")
 138         self.parser.add_option("", "--delegations",
 139                                metavar="<slice1,slice2,...>", action="append",
 140                                help="List of slices with delegation authority")
 141         self.parser.add_option("", "--disk-max",
 142                                metavar="<num>", action="append",
 143                                help="Disk quota (1k disk blocks)")
 144         self.parser.add_option("", "--initscript",
 145                                metavar="<name>", action="append",
 146                                help="Slice initialization script (e.g., stork)")
 147         self.parser.add_option("", "--ip-addresses", action="append",
 148                                metavar="<IP addr>",
 149                                help="Add an IP address to a sliver")
 150         self.parser.add_option("", "--net-i2-max-kbyte",
 151                                metavar="<KBytes>", action="append",
 152                                help="Maximum daily network Tx limit " +
 153                                "to I2 hosts.")
 154         self.parser.add_option("", "--net-i2-max-rate",
 155                                metavar="<Kbps>", action="append",
 156                                help="Maximum bandwidth over I2 routes")
 157         self.parser.add_option("", "--net-i2-min-rate",
 158                                metavar="<Kbps>", action="append",
 159                                help="Minimum bandwidth over I2 routes")
 160         self.parser.add_option("", "--net-i2-share",
 161                                metavar="<num>", action="append",
 162                                help="Number of bandwidth shares over I2 routes")
 163         self.parser.add_option("", "--net-i2-thresh-kbyte",
 164                                metavar="<KBytes>", action="append",
 165                                help="Limit sent to I2 hosts before warning, " +
 166                                "throttling")
 167         self.parser.add_option("", "--net-max-kbyte",
 168                                metavar="<KBytes>", action="append",
 169                                help="Maximum daily network Tx limit " +
 170                                "to non-I2 hosts.")
 171         self.parser.add_option("", "--net-max-rate",
 172                                metavar="<Kbps>", action="append",
 173                                help="Maximum bandwidth over non-I2 routes")
 174         self.parser.add_option("", "--net-min-rate",
 175                                metavar="<Kbps>", action="append",
 176                                help="Minimum bandwidth over non-I2 routes")
 177         self.parser.add_option("", "--net-share",
 178                                metavar="<num>", action="append",
 179                                help="Number of bandwidth shares over non-I2 " +
 180                                "routes")
 181         self.parser.add_option("", "--net-thresh-kbyte",
 182                                metavar="<KBytes>", action="append",
 183                                help="Limit sent to non-I2 hosts before " +
 184                                "warning, throttling")
 185         self.parser.add_option("", "--vsys",
 186                                metavar="<name>", action="append",
 187                                help="Vsys script (e.g., fd_fusemount)")
 188         self.parser.add_option("", "--vsys-vnet",
 189                                metavar="<IP network>", action="append",
 190                                help="Allocate a virtual private network")
 191
 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',
 199                     'vsys','vsys_vnet']
 200         attrdict = {}
 201         for attr in attrlist:
 202             value = getattr(self.opts, attr, None)
 203             if value is not None:
 204                 attrdict[attr] = value
 205         return attrdict
 206
 207     def prep(self):
 208         (self.opts, self.args) = self.parser.parse_args()
 209
 210         if self.opts.infile:
 211             sys.stdin = open(self.opts.infile, "r")
 212         xml = sys.stdin.read()
 213         self.rspec = RSpec(xml)
 214
 215         if self.nodefile:
 216             if self.opts.nodefile:
 217                 f = open(self.opts.nodefile, "r")
 218                 self.nodes = f.read().split()
 219                 f.close()
 220             else:
 221                 self.nodes = self.args
 222
 223         if self.opts.outfile:
 224             sys.stdout = open(self.opts.outfile, "w")
 225
 226
 227
 228
 229
 230
 231