ans-command and scripts.py for the basic cli parser to be imported elsewhere

This commit is contained in:
Seth Vidal 2012-02-28 01:12:04 -05:00
parent ada866be46
commit 95706af074
2 changed files with 148 additions and 0 deletions

117
bin/ans-command Executable file
View file

@ -0,0 +1,117 @@
#!/usr/bin/python -tt
#skvidal
# (c) Red Hat, Inc 2012
#todo
# add 'execution time' option output to the output
import sys
import os
import getpass
import ansible.runner
import shlex
from ansible.scripts import base_ans_parser, errorprint
def main(args):
parser = base_ans_parser(outputpath=False)
parser.usage = "ans-command [options] command-to-run"
parser.add_option('--returncodes', action='store_true',
help="prefix each line with the commands returncode")
parser.add_option('-1', '--oneline', action='store_true',
help="output all things as one line - to make grepping easier, will \
not remove \\n's from output of commands, though")
parser.add_option('-o', '--output-to-dir', dest='output_dest', default=None,
help="output each hosts results to a file in a dir named for the host")
options, args = parser.parse_args(args)
sshpass = None
if options.askpass:
sshpass = getpass.getpass(prompt="SSH password: ")
pattern = '*'
if options.host:
pattern += ';'.join(options.host)
if options.output_dest:
if options.output_dest[0] != '/':
options.output_dest = os.path.realpath(os.path.expanduser(options.output_dest))
if not os.path.exists(options.output_dest):
try:
os.makedirs(options.output_dest)
except (IOError, OSError), e:
print >> sys.stderr, "Could not make dir %s: %s" % (options.output_dest, e)
sys.exit(1)
if not os.access(options.output_dest, os.W_OK):
print >> sys.stderr, "Cannot write to path %s" % options.output_dest
sys.exit(1)
mycmd = ' '.join(args)
runner = ansible.runner.Runner(
module_name='command',
module_path=options.module_path,
module_args=shlex.split(mycmd),
remote_user=options.remote_user,
remote_pass=sshpass,
host_list=options.host_list,
forks=options.forks,
pattern=pattern,
verbose=True,
)
print mycmd
results = runner.run()
for hn in sorted(results['contacted']):
d = results['contacted'][hn]
if d.get('rc', 0) != 0 or d.get('failed', 0):
msg = 'Error: %s: ' % hn
# too bad stdout/stderr is not interleaved :(
msg += d.get('stdout', '')
msg += d.get('stderr', '')
msg += d.get('traceback', '')
msg += d.get('error', '')
errorprint(msg)
continue
if options.output_dest:
fo = open(options.output_dest + '/' + hn +'.output', 'w')
fo.write(mycmd + '\n')
fo.write('%s:\n' % hn)
if options.returncodes:
fo.write('return code: %s\n' % d['rc'])
fo.write('%s\nErrors:\n%s\n' % (d['stdout'], d['stderr']))
fo.close()
continue
if options.oneline:
if options.returncodes:
print '%s:%s:%s:%s' % (hn, d['rc'], d['stdout'], d['stderr'])
else:
print '%s:%s:%s' % (hn, d['stdout'], d['stderr'])
else:
print '%s:' % hn
if options.returncodes:
print 'return code:%s\n' % d['rc']
print '%s' % d['stdout']
if d.get('stderr', None):
print '%s' % d['stderr']
if results['dark']:
errorprint('Hosts which could not be contacted or did not respond:')
for hn in sorted(results['dark']):
errorprint(hn)
print ''
if options.output_dest:
print "output written to %s" % options.output_dest
if __name__ == "__main__":
sys.exit(main(sys.argv[1:]))

31
lib/ansible/scripts.py Normal file
View file

@ -0,0 +1,31 @@
from optparse import OptionParser
import sys
import constants as C
def base_ans_parser(opthosts=True, outputpath=True, forkdef=C.DEFAULT_FORKS):
parser = OptionParser()
if opthosts:
parser.add_option('--host', default=[], action='append',
help="hosts to act on, defaults to ALL")
parser.add_option("-H", "--host-list", dest="host_list",
help="path to hosts list", default=C.DEFAULT_HOST_LIST)
parser.add_option("-L", "--library", dest="module_path",
help="path to module library", default=C.DEFAULT_MODULE_PATH)
parser.add_option('-u', '--user', default=C.DEFAULT_REMOTE_USER,
dest='remote_user', help='set the default username')
parser.add_option("-P", "--askpass", default=False, action="store_true",
help="ask the user to input the ssh password for connecting")
parser.add_option('-f','--forks', default=forkdef, type='int',
help='set the number of forks to start up')
if outputpath:
parser.add_option('--outputpath', default='/tmp/ansible', dest="outputpath",
help="basepath to store results/errors output.")
return parser
# other functions as needed for nicely handling output from json back
# to things people might be more inclined to deal with at a bash prompt
def errorprint(msg):
print >> sys.stderr, msg