10 from config import config
11 from optparse import OptionParser
12 from automate import *
14 def build_vx_args(shell_cmd):
15 ssh_options="-q -o UserKnownHostsFile=junkssh -o StrictHostKeyChecking=no"
16 cmd="""ssh %s root@{} """ % ssh_options
18 args.append(shell_cmd)
21 def vx_start(filelist,outdir,cmd, timeout=0):
22 args = build_vx_args(cmd)
23 vxargs.start(None, 20, filelist, outdir, False, args, timeout)
25 if __name__ == "__main__":
26 parser = OptionParser()
27 parser.set_defaults(nodelist=None,
36 parser.add_option("", "--nodelist", dest="nodelist", metavar="filename",
37 help="Read list of nodes from specified file")
38 parser.add_option("", "--node", dest="node", metavar="hostname",
39 help="specify a single node name.")
40 parser.add_option("", "--timeout", dest="timeout", metavar="seconds",
41 help="Number of seconds to wait before timing out on host.")
42 parser.add_option("", "--outdir", dest="outdir", metavar="dirname",
43 help="Name of directory to place output")
44 parser.add_option("", "--cmd", dest="cmdfile", metavar="filename",
45 help="Name of file that contains a unix-to-csv command " + \
46 "to run on the hosts.")
48 config = config(parser)
51 if config.outdir == None:
56 if not os.path.exists(outdir):
57 os.system('mkdir -p %s' % outdir)
59 if config.nodelist == None and config.node == None:
60 filelist="nocomon.txt"
61 filelist = vxargs.getListFromFile(open(filelist,'r'))
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')
79 vx_start(filelist, outdir, cmd, int(config.timeout))