hope I got the merge wright...

[monitor.git] / commands / nodegroups.py

diff --git a/commands/nodegroups.py b/commands/nodegroups.py
new file mode 100755 (executable)
index 0000000..ecb9c08
--- /dev/null
+++ b/commands/nodegroups.py
@@ -0,0 +1,134 @@
+#!/usr/bin/python
+
+# This script is used to manipulate the operational state of nodes in
+# different node groups.  These are basically set operations on nodes via the
+# PLC api.
+# 
+# Take the ng name as an argument....
+# optionally, 
+#  * restart them all.
+#  * Set some or all in the set to rins.
+#  * get a list of nodes in the Alpha nodegroup.
+# 
+# Given a nodelist, it could tag each one with a nodegroup name.
+#  * 
+
+from monitor import database
+from monitor.database.info.model import FindbadNodeRecord
+from monitor import util
+from monitor.wrapper import plc
+from monitor import parser as parsermodule
+
+api = plc.getAuthAPI()
+
+from monitor.common import *
+from sets import Set
+
+def main():
+
+       parser = parsermodule.getParser(['nodesets'])
+       parser.set_defaults( list=True,
+                                               add=False,
+                        nocolor=False,
+                                               notng=False,
+                                               delete=False,)
+
+       parser.add_option("", "--not", dest="notng", action="store_true", 
+                                               help="All nodes NOT in nodegroup.")
+       parser.add_option("", "--nocolor", dest="nocolor", action="store_true", 
+                                               help="Enable color")
+       parser.add_option("", "--list", dest="list", action="store_true", 
+                                               help="List all nodes in the given nodegroup")
+       parser.add_option("", "--add", dest="add", action="store_true", 
+                                               help="Add nodes to the given nodegroup")
+       parser.add_option("", "--delete", dest="delete", action="store_true", 
+                                               help="Delete nodes from the given nodegroup")
+
+       parser = parsermodule.getParser(['defaults'], parser)
+       config = parsermodule.parse_args(parser)
+
+       # COLLECT nodegroups, nodes and node lists
+       if config.node or config.nodelist:
+               if config.node: 
+                       hostlist = [ config.node ] 
+               else: 
+                       hostlist = util.file.getListFromFile(config.nodelist)
+
+               # NOTE: preserve order given in file.  Otherwise, return values are not in order
+               # given to GetNodes
+               nodelist = []
+               for h in hostlist:
+                       nodelist.append( plccache.GetNodeByName(h) )
+
+               group_str = "Given"
+
+       elif config.site:
+               site = plccache.GetSitesByName([config.site])
+               if len (site) > 0:
+                       site = site[0]
+                       nodelist = plccache.GetNodesByIds(site['node_ids'])
+               else:
+                       nodelist = []
+
+               group_str = config.site
+
+       elif config.nodeselect:
+               hostlist = query.node_select(config.nodeselect)
+               nodelist = [ plccache.GetNodeByName(h) for h in hostlist ]
+
+               group_str = "selection"
+               
+       else:
+               ng = api.GetNodeGroups({'name' : config.nodegroup})
+               nodelist = plccache.GetNodesByIds(ng[0]['node_ids'])
+
+               group_str = config.nodegroup
+
+       if config.notng:
+               # Get nodegroup nodes
+               ng_nodes = nodelist
+
+               # Get all nodes
+               all_nodes = plccache.l_nodes
+               
+               # remove ngnodes from all node list
+               ng_list = [ x['hostname'] for x in ng_nodes ]
+               all_list = [ x['hostname'] for x in all_nodes ]
+               not_ng_nodes = Set(all_list) - Set(ng_list)
+
+               # keep each node that *is* in the not_ng_nodes set
+               nodelist = filter(lambda x : x['hostname'] in not_ng_nodes, all_nodes)
+
+       hostnames = [ n['hostname'] for n in nodelist ]
+
+       # commands:
+
+       if config.add and config.nodegroup:
+               for node in hostnames:
+                       print "Adding %s to %s nodegroup" % (node, config.nodegroup)
+                       api.AddNodeToNodeGroup(node, config.nodegroup)
+
+       elif config.delete:
+               for node in hostnames:
+                       print "Deleting %s from %s nodegroup" % (node, config.nodegroup)
+                       api.DeleteNodeFromNodeGroup(node, config.nodegroup)
+
+       elif config.list:
+               print " ---- Nodes in the %s Node Group ----" % group_str
+               print "   Hostname                                   plc  obs     pcu     key         kernel                        last_contact, last change, comon uptime"
+               i = 1
+               for node in nodelist:
+                       print "%-2d" % i, 
+                       fbrec = FindbadNodeRecord.get_latest_by(hostname=node['hostname'])
+                       fbdata = fbrec.to_dict()
+                       print nodegroup_display(node, fbdata, config)
+                       i += 1
+
+       else:
+               print "no other options supported."
+
+if __name__ == "__main__":
+       try:
+               main()
+       except IOError:
+               pass