aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xinput_fail.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/input_fail.py b/input_fail.py
index 67f5e74..4b7e802 100755
--- a/input_fail.py
+++ b/input_fail.py
@@ -1,14 +1,13 @@
#!/usr/bin/env python
# Exit with 1 if any input is provided. Print the input to stdout, unless an
-# argument is specified. In that case, print the argument.
+# argument is specified. In that case, also print the argument.
import sys
input_string = sys.stdin.read()
if input_string != "":
+ sys.stdout.write(input_string)
if len(sys.argv) >= 2:
print(sys.argv[1])
- else:
- sys.stdout.write(input_string)
sys.exit(1)