python - args from file(s) or STDIN

args.py

#! /usr/bin/python

import fileinput, argparse
from glob import glob

"""
# read from <file(s)> or <STDIN>
# Examples:
args.py *.txt
args.py 1.txt 2.txt 3.txt
args.py *.txt -n
args.py < *.txt
dir | args.py -n
df -hP | python3.5 ./args.py -n
"""

args = argparse.ArgumentParser(description='args from file(s) or STDIN')

args.add_argument('--version', action='version', version='%(prog)s ')
args.add_argument('files', metavar='FILE', nargs='*', help='file(s) to read')
args.add_argument('-n', '--number', action='store_true', help='print line numbers')

args = args.parse_args()
 
with fileinput.input([f for files in args.files for f in glob(files)]) as f:
 for line in f:
  if args.number:
   print ("%s : %d : %s" % (f.filename(), f.filelineno(), line.rstrip()))
  else:
   print(line.rstrip())