Python实现读取目录所有文件的文件名并保存到txt文件代码

import os
def ListFilesToTxt(dir,file,wildcard,recursion):
 exts = wildcard.split(" ")
 files = os.listdir(dir)
 for name in files:
 fullname=os.path.join(dir,name)
 if(os.path.isdir(fullname) & recursion):
 ListFilesToTxt(fullname,file,wildcard,recursion)
 else:
 for ext in exts:
 if(name.endswith(ext)):
 file.write(name + "
")
 break
def Test():
 dir="J:\1"
 outfile="binaries.txt"
 wildcard = ".txt .exe .dll .lib"
 
 file = open(outfile,"w")
 if not file:
 print ("cannot open the file %s for writing" % outfile)
 ListFilesToTxt(dir,file,wildcard, 1)
 
 file.close()
Test()

import os
def ListFilesToTxt(dir,file,wildcard,recursion):
 exts = wildcard.split(" ")
 for root, subdirs, files in os.walk(dir):
 for name in files:
 for ext in exts:
 if(name.endswith(ext)):
 file.write(name + "
")
 break
 if(not recursion):
 break
def Test():
 dir="J:\1"
 outfile="binaries.txt"
 wildcard = ".txt .exe .dll .lib"
 
 file = open(outfile,"w")
 if not file:
 print ("cannot open the file %s for writing" % outfile)
 ListFilesToTxt(dir,file,wildcard, 0)
 
 file.close()
Test()