[Python] Read command line arguments from file name or stdin

#!/usr/bin/env python3
"""
If the argument is a file, the contents of the file,
Otherwise print the arguments
"""
import sys
import fileinput
from pathlib import Path

if Path(sys.argv[1]).exists():  #If the first argument is a file
    for line in fileinput.input():  #Print the contents of the file line by line
        print('file input')
        print(line)
else:  #If the first argument is not a file
    for i in sys.argv[1:]:  #Print the argument string
        print('args input')
        print(i)

$ python stdin_test.py hoge.txt fuga.txt
file input
MYNAME!

file input
HERO!!


$ python stdin_test.py 1 2 3
args input
1
args input
2
args input
3

if Path(sys.argv[1]).exists():  #If the first argument is a file
    ARGV = [
        line.replace('\n', '') for line in fileinput.input()  #Delete line breaks
        if line != '\n'  #Delete blank lines
    ]
else:  #If the first argument is not a file
    ARGV = sys.argv[1:]

Recommended Posts