fetch.py

   1 #!/usr/bin/python
   2
   3 import csv
   4 import sys
   5 import os
   6 import config
   7 from glob import glob
   8
   9 import vxargs
  10 from config import config
  11 from optparse import OptionParser
  12 from automate import *
  13
  14 def build_vx_args(shell_cmd):
  15         ssh_options="-q -o UserKnownHostsFile=junkssh -o StrictHostKeyChecking=no"
  16         cmd="""ssh %s root@{} """  % ssh_options
  17         args = cmd.split()
  18         args.append(shell_cmd)
  19         return args
  20
  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)
  24
  25 if __name__ == "__main__":
  26         parser = OptionParser()
  27         parser.set_defaults(nodelist=None,
  28                                                 node=None,
  29                                                 outdir=None,
  30                                                 querystr=None,
  31                                                 timeout=0,
  32                                                 simple=False,
  33                                                 run=False,
  34                                                 cmdfile=None,)
  35
  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.")
  47
  48         config = config(parser)
  49         config.parse_args()
  50
  51         if config.outdir == None:
  52                 outdir="checkhosts"
  53         else:
  54                 outdir=config.outdir
  55
  56         if not os.path.exists(outdir):
  57                 os.system('mkdir -p %s' % outdir)
  58
  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, '')]
  68         else:
  69                 # probably no such file.
  70                 raise Exception("No such file %s" % config.nodelist)
  71
  72         if config.cmdfile == None:
  73                 f = open("command.txt",'r')
  74                 cmd = f.read()
  75         else:
  76                 f = open(config.cmdfile,'r')
  77                 cmd = f.read()
  78
  79         vx_start(filelist, outdir, cmd, int(config.timeout))