10 from config import config
11 from optparse import OptionParser
12 from automate import *
14 parser = OptionParser()
15 parser.set_defaults(nodelist=None,
24 parser.add_option("", "--nodelist", dest="nodelist", metavar="filename",
25 help="Read list of nodes from specified file")
26 parser.add_option("", "--node", dest="node", metavar="hostname",
27 help="specify a single node name.")
28 parser.add_option("", "--timeout", dest="timeout", metavar="seconds",
29 help="Number of seconds to wait before timing out on host.")
30 parser.add_option("", "--outdir", dest="outdir", metavar="dirname",
31 help="Name of directory to place output")
32 parser.add_option("", "--cmd", dest="cmdfile", metavar="filename",
33 help="Name of file that contains a unix-to-csv command " + \
34 "to run on the hosts.")
36 config = config(parser)
39 def build_vx_args(shell_cmd):
40 ssh_options="-q -o UserKnownHostsFile=junkssh -o StrictHostKeyChecking=no"
41 cmd="""ssh %s root@{} """ % ssh_options
43 args.append(shell_cmd)
46 def vx_start(filelist,outdir,cmd):
47 args = build_vx_args(cmd)
48 #vxargs.start(None, 10, filelist, outdir, False, args, 120)
49 vxargs.start(None, 10, filelist, outdir, False, args, int(config.timeout))
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)