[code.view]

[top] / python / PyMOTW / fileinput / fileinput_grep.py

     #!/usr/bin/env python
     # encoding: utf-8
     #
     # Copyright (c) 2010 Doug Hellmann.  All rights reserved.
     #
     """Simplistic grep implementation
     """
     #end_pymotw_header
     
     import fileinput
     import re
     import sys
     
     pattern = re.compile(sys.argv[1])
     
     for line in fileinput.input(sys.argv[2:]):
         if pattern.search(line):
             if fileinput.isstdin():
                 fmt = '{lineno}:{line}'
             else:
                 fmt = '{filename:<20}:{lineno}:{line}'
             print fmt.format(filename=fileinput.filename(),
                              lineno=fileinput.filelineno(),
                              line=line.rstrip())
     

[top] / python / PyMOTW / fileinput / fileinput_grep.py

contact | logmethods.com