9 from monitor import parser as parsermodule
10 from monitor import common
11 from automate import *
13 def build_vx_args(shell_cmd):
14 ssh_options="-q -o UserKnownHostsFile=junkssh -o StrictHostKeyChecking=no"
15 cmd="""ssh %s root@{} """ % ssh_options
17 args.append(shell_cmd)
20 def vx_start(filelist,outdir,cmd, timeout=0):
21 args = build_vx_args(cmd)
22 vxargs.start(None, 20, filelist, outdir, False, args, timeout)
24 if __name__ == "__main__":
26 parser = parsermodule.getParser(['nodesets'])
27 parser.set_defaults(outdir=None,
33 parser.add_option("", "--timeout", dest="timeout", metavar="seconds",
34 help="Number of seconds to wait before timing out on host.")
35 parser.add_option("", "--outdir", dest="outdir", metavar="dirname",
36 help="Name of directory to place output")
37 parser.add_option("", "--cmd", dest="cmdfile", metavar="filename",
38 help="Name of file that contains a unix-to-csv command " + \
39 "to run on the hosts.")
41 config = parsermodule.parse_args(parser)
43 if config.outdir == None:
48 if not os.path.exists(outdir):
49 os.system('mkdir -p %s' % outdir)
51 #if config.site is not None or \
52 # config.nodeselect is not None or \
53 # config.nodegroup is not None:
54 # print "TODO: implement support for nodeselect and site queries."
55 # print "%s %s %s" % (config.site, config.nodeselect, config.nodegroup)
58 nodelist = common.get_nodeset(config)
61 filelist = [ (x, '') for x in nodelist ]
62 elif os.path.exists(str(config.nodelist)) and os.path.isfile(config.nodelist):
63 filelist = vxargs.getListFromFile(open(config.nodelist,'r'))
64 elif os.path.exists(str(config.nodelist)) and os.path.isdir(config.nodelist):
65 filelist = get_hostlist_from_dir(config.nodelist)
66 elif config.node is not None:
67 filelist = [(config.node, '')]
69 # probably no such file.
70 raise Exception("No such file %s" % config.nodelist)
72 if config.cmdfile == None:
73 f = open("command.txt",'r')
76 f = open(config.cmdfile,'r')
81 vx_start(filelist, outdir, cmd, int(config.timeout))