File Search

#!/usr/bin/python

"""pygrep.py -- 
searches for word and returns count and line number"""

import sys
import string
import fileinput
filename = raw_input('Enter filename to search. ')
word = raw_input('Enter search term. ')

def filefind(word, filename):

    count = 0
    try:
        fh = open(filename, 'r')

    except:
        print filename, ":", sys.exc_value[1]
        sys.exit(1)

    allLines = fh.readlines()
    fh.close()
    for eachLine in allLines:
        if string.find(eachLine, word) > -1:
            count = count + 1
            print 'Found', eachLine,

    print 'Total =', count

def getline(word, filename):
    for line in fileinput.input(filename):
        if line.find(word) >= 0:
            print '%s found on line %d' % (word, (fileinput.filelineno()))



def main():
    filefind(word, filename)
    getline(word, filename)

if __name__ == '__main__':
    main()