Python script to generate md5 and sha1 hashes for files of given extension in a directory

hash_generator.py

import hashlib, os

def hashfile(afile, hasher, blocksize=65536):
    buf = afile.read(blocksize)
    while len(buf) > 0:
        hasher.update(buf)
        buf = afile.read(blocksize)
    return hasher.hexdigest()

path = raw_input("Enter directory to scan: ")
extension = raw_input("Enter file extension: ")

print "File\tmd5\tsha1"

os.chdir(path)
for filename in os.listdir(path):
	if filename.endswith(extension):
		print filename + "\t" + hashfile(open(filename, 'rb'), hashlib.md5()) + "\t" + hashfile(open(filename, 'rb'), hashlib.sha1())