cat_via_argf.py

"""ARGF from Ruby in Python.
"""

import sys, os


class _ARGF(object):

    def __init__(self):
        self.lineno = 1
        self.file = None

    def __iter__(self):
        return self.next()

    def next(self):
        files = filter(os.path.isfile, sys.argv)
        if files:
            for f in files:
                self.lineno = 0
                self.file = f
                for n in open(f).xreadlines():
                    self.lineno += 1
                    yield n
        else:
            for n in sys.stdin.readlines():
                self.file = 'STDIN'
                self.lineno += 1
                yield n

ARGF = _ARGF()


if __name__ == '__main__':
    for n in ARGF:
        print ARGF.file, ":", ARGF.lineno, ':', n