Refactoring test-module to be more like ansible.

hacking/test-module

@@ -30,6 +30,7 @@ import sys
 import os
 import subprocess
 import traceback
+import optparse
 import ansible.utils as utils
 import ansible.module_common as module_common
 
@@ -38,67 +39,93 @@ try:
 except ImportError:
     import simplejson as json
 
-modfile = None
+def parse():
+    """parse command line
 
-if len(sys.argv) == 1:
-    print >>sys.stderr, "usage: test-module ./library/command [key=value ...]"
+    :return : (options, args)"""
+    parser = optparse.OptionParser()
+
+    parser.usage = "%prog [options] (-h for help)"
+
+    parser.add_option('-m', '--module-name', dest='module_name',
+        help="module name to execute")
+    parser.add_option('-a', '--args', dest='module_args', default="",
+        help="module arguments")
+    parser.add_option('-D', '--debugger', dest='debugger', 
+        help="path to python debugger (e.g. /usr/bin/pdb)")
+    options, args = parser.parse_args()
+    if not options.module_name:
+        parser.print_help()
         sys.exit(1)
+    else:
+        return options, args
 
-modfile = sys.argv[1]
-if len(sys.argv) > 1:
-    args = " ".join(sys.argv[2:])
-else:
-    args = ""
+def write_argsfile( argstring):
+    """Write args to a file for the module's use.
     
-argspath = os.path.expanduser("~/.ansible_test_module_arguments")
-argsfile = open(argspath, 'w')
-argsfile.write(args)
-argsfile.close()
+    :return: full path to args file"""
+    argspath = os.path.expanduser("~/.ansible_test_module_arguments")
+    argsfile = open(argspath, 'w')
+    argsfile.write(argstring)
+    argsfile.close()
+    return argspath
 
-module_fh = open(modfile)
-module_data = module_fh.read()
-included_boilerplate = module_data.find(module_common.REPLACER) != -1
-module_fh.close()
+def boilerplate_module( modfile):
+    module_fh = open(modfile)
+    module_data = module_fh.read()
+    included_boilerplate = module_data.find(module_common.REPLACER) != -1
+    module_fh.close()
 
-if included_boilerplate:
+    if included_boilerplate:
         module_data = module_data.replace(module_common.REPLACER, module_common.MODULE_COMMON)
         modfile2_path = os.path.expanduser("~/.ansible_module_generated")
         print "* including generated source, if any, saving to: %s" % modfile2_path
-    print "* this will offset any line numbers in tracebacks!"
+        print "* this will offset any line numbers in tracebacks/debuggers!"
         modfile2 = open(modfile2_path, 'w')
         modfile2.write(module_data)
         modfile2.close()
         modfile = modfile2_path
-else:
-    print "* module boilerplate substitution not requested in module, tracebacks will be unaltered"
+        return modfile2_path
+    else:
+        print "* module boilerplate substitution not requested in module, line numbers will be unaltered"
+        return modfile
 
-os.system("chmod +x %s" % modfile)
-cmd = subprocess.Popen("%s %s" % (modfile, argspath), 
+def main(): 
+    options, args = parse()
+
+    argspath = write_argsfile( options.module_args)
+    modfile = boilerplate_module( options.module_name)
+    print argspath, modfile
+    sys.exit(0)
+    #===== run or debug the module
+    os.system("chmod +x %s" % modfile)
+    cmd = subprocess.Popen("%s %s" % (modfile, argspath), 
         shell=True,
         stdout=subprocess.PIPE, 
         stderr=subprocess.PIPE)
-(out, err) = cmd.communicate()
+    (out, err) = cmd.communicate()
 
-try:
+    try:
         print "***********************************"
         print "RAW OUTPUT"
         print out
         print err
         results = utils.parse_json(out)
         
-except:
+    except:
         print "***********************************"
         print "INVALID OUTPUT FORMAT"
         print out
         traceback.print_exc()
         sys.exit(1)
 
-print "***********************************"
-print "PARSED OUTPUT"
+    print "***********************************"
+    print "PARSED OUTPUT"
 
-print utils.jsonify(results,format=True)
-
-sys.exit(0)
+    print utils.jsonify(results,format=True)
 
 	
+if __name__ == "__main__":
+    main()
+