Make output file an option

Change-Id: I9f59cdf1ba9e6f62cccca72381cfb55f312cdabe

1 files changed, 3 insertions, 2 deletions

diff --git a/parser.py b/parser.py
index 8d3cc04..a351616 100755
--- a/parser.py
+++ b/parser.py
@@ -169,6 +169,7 @@ class CounterLabelsAction(argparse.Action):
 parser = argparse.ArgumentParser()
 parser.add_argument('--core', '-c', choices=counter_labels.keys(), action=CounterLabelsAction)
 parser.add_argument('input', type=argparse.FileType('r'))
+parser.add_argument('--output', '-o', type=argparse.FileType('w'))
 args = parser.parse_args()
 
 #Globals
@@ -220,6 +221,6 @@ for binary in binary_set:
 
 headers=['Binary', 'Args']
 headers.extend(counter_set)
-print '\t'.join(headers)
+print >> args.output, '\t'.join(headers)
 for row in table:
-  print '\t'.join(row)
+  print >> args.output, '\t'.join(row)