start zabbix_agent after installing it... assumes it's in a prod env.

[monitor.git] / nodegroups.py

diff --git a/nodegroups.py b/nodegroups.py
index 725d0e0..9e14e2f 100755 (executable)
--- a/nodegroups.py
+++ b/nodegroups.py
@@ -13,53 +13,47 @@
 # Given a nodelist, it could tag each one with a nodegroup name.
 #  * 
 
-import plc
-import auth
-api = plc.PLC(auth.auth, auth.plc)
+from monitor import database
+from monitor.database import FindbadNodeRecord
+from monitor import util
+from monitor.wrapper import plc
+from monitor import parser as parsermodule
 
-from optparse import OptionParser
-from sets import Set
+api = plc.getAuthAPI()
 
+from nodequery import verify,query_to_dict,node_select
 from nodecommon import *
-import soltesz
+from sets import Set
 
 def main():
-       from config import config
-       fb = soltesz.dbLoad("findbad")
-
-       parser = OptionParser()
-       parser.set_defaults(nodegroup="Alpha",
-                                               node=None,
-                                               nodelist=None,
-                                               list=False,
+
+       parser = parsermodule.getParser(['nodesets'])
+       parser.set_defaults( list=True,
                                                add=False,
+                        nocolor=False,
                                                notng=False,
-                                               delete=False,
-                                               )
+                                               delete=False,)
+
        parser.add_option("", "--not", dest="notng", action="store_true", 
                                                help="All nodes NOT in nodegroup.")
-       parser.add_option("", "--nodegroup", dest="nodegroup", metavar="NodegroupName",
-                                               help="Specify a nodegroup to perform actions on")
-
+       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.add_option("", "--node", dest="node", metavar="nodename.edu", 
-                                               help="A single node name to add to the nodegroup")
-       parser.add_option("", "--nodelist", dest="nodelist", metavar="list.txt", 
-                                               help="Use all nodes in the given file for operation.")
-       config = config(parser)
-       config.parse_args()
+
+       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 = config.getListFromFile(config.nodelist)
+                       hostlist = util.file.getListFromFile(config.nodelist)
 
                # NOTE: preserve order given in file.  Otherwise, return values are not in order
                # given to GetNodes
@@ -70,6 +64,22 @@ def main():
                #nodelist = api.GetNodes(hostlist)
                group_str = "Given"
 
+       elif config.site:
+               site = api.GetSites(config.site)
+               if len (site) > 0:
+                       site = site[0]
+                       nodelist = api.GetNodes(site['node_ids'])
+               else:
+                       nodelist = []
+
+               group_str = config.site
+
+       elif config.nodeselect:
+               hostlist = node_select(config.nodeselect)
+               nodelist = api.GetNodes(hostlist)
+
+               group_str = "selection"
+               
        else:
                ng = api.GetNodeGroups({'name' : config.nodegroup})
                nodelist = api.GetNodes(ng[0]['node_ids'])
@@ -94,15 +104,8 @@ def main():
        hostnames = [ n['hostname'] for n in nodelist ]
 
        # commands:
-       if config.list:
-               print " ---- Nodes in the %s Node Group ----" % group_str
-               i = 1
-               for node in nodelist:
-                       print "%-2d" % i, 
-                       print nodegroup_display(node, fb)
-                       i += 1
 
-       elif config.add and config.nodegroup:
+       if config.add and config.nodegroup:
                for node in hostnames:
                        print "Adding %s to %s nodegroup" % (node, config.nodegroup)
                        api.AddNodeToNodeGroup(node, config.nodegroup)
@@ -112,6 +115,17 @@ def main():
                        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.query.filter(FindbadNodeRecord.hostname==node['hostname']).order_by(FindbadNodeRecord.date_checked.desc()).first()
+                       fbdata = fbrec.to_dict()
+                       print nodegroup_display(node, fbdata, config)
+                       i += 1
+
        else:
                print "no other options supported."