Skip to content

Instantly share code, notes, and snippets.

@martinth
Created August 6, 2015 11:04
Show Gist options
  • Save martinth/ed991fb8cdcac3dfadf7 to your computer and use it in GitHub Desktop.
Save martinth/ed991fb8cdcac3dfadf7 to your computer and use it in GitHub Desktop.
Read from stdin or files in Python (combining argparse and fileinput)
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)
@fish8
Copy link

fish8 commented Aug 10, 2024

Thank you, both this and @drmull's comment were very helpful

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment