Allow shrinker to read traces from stdin

black-parrot · Feb 17, 2016 · 66853a9 · 66853a9
1 parent d70a473
commit 66853a9
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/axe-shrink.py b/src/axe-shrink.py
@@ -16,10 +16,13 @@
   sys.exit()
 
 # Open trace
-f = open(sys.argv[2], 'r')
-if f == None:
-  print "File not found: ", sys.argv[2]
-  sys.exit()
+if sys.argv[2] == "-":
+  f = sys.stdin
+else:
+  f = open(sys.argv[2], 'r')
+  if f == None:
+    print "File not found: ", sys.argv[2]
+    sys.exit()
 
 # Read trace
 lines = []