python某一目录下指定格式的文件路径保存到指定格式的文件中

import sys
import glob
import os
if len(sys.argv) < 3 :
    print 'usage: filenmae.py dir suffix'
    exit()
dir = sys.argv[1]
suffix = sys.argv[2]
f = glob.glob(dir + '/*.' + suffix)
fileout = open(dir + '/' + suffix + '.txt','wt')
for file in f :
    filename = os.path.basename(file)
    print dir +'/'+filename
    fileout.write(dir +'/'+filename)
    fileout.write('\n')
fileout.close()