Created
August 6, 2015 11:04
-
-
Save martinth/ed991fb8cdcac3dfadf7 to your computer and use it in GitHub Desktop.
Read from stdin or files in Python (combining argparse and fileinput)
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import argpase | |
import fileinput | |
if __name__ == '__main__': | |
parser = ArgumentParser() | |
parser.add_argument('--dummy', help='dummy argument') | |
parser.add_argument('files', metavar='FILE', nargs='*', help='files to read, if empty, stdin is used') | |
args = parser.parse_args() | |
# If you would call fileinput.input() without files it would try to process all arguments. | |
# We pass '-' as only file when argparse got no files which will cause fileinput to read from stdin | |
for line in fileinput.input(files=args.files if len(args.files) > 0 else ('-', )): | |
print(line) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Thank you, both this and @drmull's comment were very helpful