python一个遍历文件目录的小脚本

#!/usr/bin/env python
#coding=utf-8
import os
'''遍历一个文件目录'''
def searchdir(dirname, flag = 1):
    if not os.path.isdir(dirname):
        print "the dir name you input does not exists"
    else:
        filelist = os.listdir(dirname)
        os.chdir(dirname)
        cwd = os.getcwd()
        path = ''
        for each_file in filelist:
            path = os.path.join(cwd, each_file)
            if os.path.isdir(path):
                searchdir(each_file,flag = flag + 1)
            else:
                print '_'*flag*3 ,'%s %s' % ( cwd, each_file)
dirname = raw_input("please input the dir you want to search:")
searchdir(dirname)