diff refinery_test_tool.py @ 8:ee221004e3b4 draft

Add an option to produce empty files.
author hackdna
date Mon, 13 May 2013 17:59:26 -0400
parents a410fb6c43fa
children
line wrap: on
line diff
--- a/refinery_test_tool.py	Mon May 13 16:52:26 2013 -0400
+++ b/refinery_test_tool.py	Mon May 13 17:59:26 2013 -0400
@@ -29,8 +29,11 @@
     input = read_files(args.input_files)
     try:
         for out_file in args.output_files:
-            out_file.write("Output file name: " + out_file.name + "\n\n")
-            out_file.write(input)
+            if args.empty_outfile:
+                output = ''
+            else:
+                output = "Output file name: " + out_file.name + "\n\n" + input
+            out_file.write(output)
     except IOError as e:
         cleanup(args)
         parser.error(e)
@@ -104,6 +107,8 @@
                         help='write a message to stdout')
     parser.add_argument('--stderr', action='store_true',
                         help='write a message to stderr')
+    parser.add_argument('--empty_outfile', action='store_true',
+                        help='produce empty output file(s)')
     parser.add_argument('-p', '--p-fail', type=float, default=0.0,
                         help='probability of execution failure, default: %(default)s')
     parser.add_argument('-s', '--sleep', dest='seconds', type=int, default=0,