如何通过python实现遍历子文件和所有子文件夹

Question

如何通过python实现遍历子文件和所有子文件夹

Answer 1

# -*- coding: cp936 -*-  

import os  

def getfilelist(filepath, tabnum=1):  

    simplepath = os.path.split(filepath)[1]  

    returnstr = simplepath+"目录<>"+"\n"  

    returndirstr = ""  

    returnfilestr = ""  

    filelist = os.listdir(filepath)  

    for num in range(len(filelist)):  

        filename=filelist[num]  

        if os.path.isdir(filepath+"/"+filename):  

            returndirstr += "\t"*tabnum+getfilelist(filepath+"/"+filename, tabnum+1)  

        else:  

            returnfilestr += "\t"*tabnum+filename+"\n"  

    returnstr += returnfilestr+returndirstr  

    return returnstr+"\t"*tabnum+"</>\n"  

path = raw_input("请输入文件路径:")  

usefulpath = path.replace('\\', '/')  

if usefulpath.endswith("/"):  

    usefulpath = usefulpath[:-1]  

if not os.path.exists(usefulpath):  

    print "路径错误!"  

elif not os.path.isdir(usefulpath):  

    print "输入的不是目录!"  

else:  

    filelist = os.listdir(usefulpath)  

    o=open("test.xml","w+")  

    o.writelines(getfilelist(usefulpath))  

    o.close()  

    print "成功！请查看test.xml文件"