From 95706af0742d120d65b6c6d63a46c26f2f46dcc6 Mon Sep 17 00:00:00 2001
From: Seth Vidal <skvidal@fedoraproject.org>
Date: Tue, 28 Feb 2012 01:12:04 -0500
Subject: [PATCH] ans-command and scripts.py for the basic cli parser to be
 imported elsewhere

---
 bin/ans-command        | 117 +++++++++++++++++++++++++++++++++++++++++
 lib/ansible/scripts.py |  31 +++++++++++
 2 files changed, 148 insertions(+)
 create mode 100755 bin/ans-command
 create mode 100644 lib/ansible/scripts.py

diff --git a/bin/ans-command b/bin/ans-command
new file mode 100755
index 0000000000..6b885a89e0
--- /dev/null
+++ b/bin/ans-command
@@ -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:]))
+
+
+
+
+
diff --git a/lib/ansible/scripts.py b/lib/ansible/scripts.py
new file mode 100644
index 0000000000..740ead6119
--- /dev/null
+++ b/lib/ansible/scripts.py
@@ -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
+