python_samples/redirect_argparse.py

19 lines
509 B
Python

import sys
from argparse import ArgumentParser, FileType
if __name__ == "__main__":
parser = ArgumentParser()
parser.add_argument(
"-i",
"--input-file",
type=FileType("r"),
default=(None if sys.stdin.isatty() else sys.stdin),
)
ns = parser.parse_args()
if ns.input_file:
with (
open(ns.input_file, "r") if ns.input_file != sys.stdin else ns.input_file
) as f:
for line in f.readlines():
print(line)