X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=commands%2Fnodegroups.py;fp=commands%2Fnodegroups.py;h=ecb9c08125bdde6c0422793a093e493952c51816;hb=e637272100e8e03884188cb2118b21158e739bb0;hp=0000000000000000000000000000000000000000;hpb=328ee7b92f4e23570d8a33ad9244413ae3aee6bf;p=monitor.git

diff --git a/commands/nodegroups.py b/commands/nodegroups.py
new file mode 100755
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