hope I got the merge wright...

[monitor.git] / tools / automate / fetch.py

diff --git a/tools/automate/fetch.py b/tools/automate/fetch.py
new file mode 100755 (executable)
index 0000000..35cb9da
--- /dev/null
+++ b/tools/automate/fetch.py
@@ -0,0 +1,81 @@
+#!/usr/bin/python
+
+import csv
+import sys
+import os
+from glob import glob
+
+import vxargs
+from monitor import parser as parsermodule
+from monitor import common
+from automate import *
+
+def build_vx_args(shell_cmd):
+       ssh_options="-q -o UserKnownHostsFile=junkssh -o StrictHostKeyChecking=no"
+       cmd="""ssh %s root@{} """  % ssh_options
+       args = cmd.split()
+       args.append(shell_cmd)
+       return args
+
+def vx_start(filelist,outdir,cmd, timeout=0):
+       args = build_vx_args(cmd)
+       vxargs.start(None, 20, filelist, outdir, False, args, timeout)
+
+if __name__ == "__main__":
+
+       parser = parsermodule.getParser(['nodesets'])
+       parser.set_defaults(outdir=None,
+                                               timeout=0,
+                                               simple=False,
+                                               run=False,
+                                               cmdfile=None,)
+
+       parser.add_option("", "--timeout", dest="timeout", metavar="seconds",
+                                               help="Number of seconds to wait before timing out on host.")
+       parser.add_option("", "--outdir", dest="outdir", metavar="dirname",
+                                               help="Name of directory to place output")
+       parser.add_option("", "--cmd", dest="cmdfile", metavar="filename",
+                                               help="Name of file that contains a unix-to-csv command " + \
+                                                        "to run on the hosts.")
+
+       config = parsermodule.parse_args(parser)
+
+       if config.outdir == None: 
+               outdir="checkhosts"
+       else: 
+               outdir=config.outdir
+
+       if not os.path.exists(outdir):
+               os.system('mkdir -p %s' % outdir)
+
+       #if config.site is not None or \
+       #   config.nodeselect is not None or \
+       #   config.nodegroup is not None:
+       #       print "TODO: implement support for nodeselect and site queries."
+       #       print "%s %s %s" % (config.site, config.nodeselect, config.nodegroup)
+       #       sys.exit(1)
+
+       nodelist = common.get_nodeset(config)
+
+       if len(nodelist) > 0:
+               filelist = [ (x, '') for x in nodelist ]
+       elif os.path.exists(str(config.nodelist)) and os.path.isfile(config.nodelist):
+               filelist = vxargs.getListFromFile(open(config.nodelist,'r'))
+       elif os.path.exists(str(config.nodelist)) and os.path.isdir(config.nodelist):
+               filelist = get_hostlist_from_dir(config.nodelist)
+       elif config.node is not None:
+               filelist = [(config.node, '')]
+       else:
+               # probably no such file.
+               raise Exception("No such file %s" % config.nodelist)
+
+       if config.cmdfile == None:
+               f = open("command.txt",'r')
+               cmd = f.read()
+       else:
+               f = open(config.cmdfile,'r')
+               cmd = f.read()
+
+       print filelist
+
+       vx_start(filelist, outdir, cmd, int(config.timeout))