使用python遍历目录

要求：使用python遍历目录，并将目录中所有文件名保存到txt文件里。

#getFilesName
# -*- coding: cp936 -*-
import os
import re
import gl

gl.rootdir = os.getcwd()
outfilename = gl.rootdir + '\\allfilenames.txt'

'''
将文件名写入文件filenames.txt
'''
def outfile(fn):
    cout = open(outfilename,'a+')
    cout.write(fn+'\n')
    cout.close()
    return
'''
遍历函数..root/v/y/k.txt 
         root/z/t/l.txt
'''
def walk(path):
    os.chdir(path)
    dirs = os.listdir(os.getcwd())
    for child in dirs:
        subpath = os.path.join(path , child)
        if os.path.isdir(subpath):          
            walk(subpath)
        else:
            outfile(child)
    return
'''
test
'''
def test():
    walk(gl.rootdir)
    
test()

gl.py里存放全局变量rootdir

#gl.py
rootdir = ''

使用IDLE运行可以看到在当前目录输出的txt文件。