Python 遍历指定路径下的文件

def walk_dir(dir,fileinfo,topdown=True):
    for root, dirs, files in os.walk(dir, topdown):
        for name in files:
            print(os.path.join(name))
            fileinfo.write(os.path.join(root,name)+'\n')
            for name in dirs:
                print(os.path.join(name))
                fileinfo.write(' '+ os.path.join(root,name)+'\n')


if __name__ == '__main__':
    dir = raw_input('please input the path:')
#     dir = r'G:\codes\python\file_info'
    fileinfo = open('list2.txt','w')
    walk_dir(dir,fileinfo)
    fileinfo.close()