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()