Make a Python script behave like a Linux command (take a filename or stdin)

cmd.py

import sys

if __name__ == "__main__":
    f = None
    try:
        filename = sys.argv[1]
        f = open(filename, 'r')
    except IndexError:
        f = sys.stdin.readlines()

    for line in f:
        print line,