-
-
Save kongkongruye/95492ca6762e433eda3490e686529450 to your computer and use it in GitHub Desktop.
Read from stdin or files in Python (combining argparse and fileinput)
This file contains hidden or 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