#-*-coding:cp936-*-
importos
defgetfilelist(filepath,tabnum=1):
simplepath=os.path.split(filepath)[1]
returnstr=simplepath+"目录"+"\n"
returndirstr=""
returnfilestr=""
filelist=os.listdir(filepath)
fornuminrange(len(filelist)):
filename=filelist[num]
ifos.path.isdir(filepath+"/"+filename):
returndirstr+="\t"*tabnum+getfilelist(filepath+"/"+filename,tabnum+1)
else:
returnfilestr+="\t"*tabnum+filename+"\n"
returnstr+=returnfilestr+returndirstr
returnreturnstr+"\t"*tabnum+"\n"
path=raw_input("请输入文件路径:")
usefulpath=path.replace('\\','/')
ifusefulpath.endswith("/"):
usefulpath=usefulpath[:-1]
ifnotos.path.exists(usefulpath):
print"路径错误!"
elifnotos.path.isdir(usefulpath):
print"输入的不是目录!"
else:
filelist=os.listdir(usefulpath)
o=open("test.xml","w+")
o.writelines(getfilelist(usefulpath))
o.close()
print"成功!请查看test.xml文件" |