import os
def ListFileToTxt(dir,file,wildcard,recursion=False):
'''
dir : 文件路径
file: 找到的文件名写入路径
wildcard :查找的文件类型字符串
recursion : 查找方式: 是否递归查找
'''
exts = wildcard.split(' ')
files = os.listdir(dir)
for name in files:
fullname = os.path.join(dir,name)
#如果子文件是文件夹且递归查找为True
if (os.path.isdir(fullname)&recursion):
ListFileToTxt(fullname,file,wildcard,recursion=True)
else:
for ext in exts:
if name.endswith(ext):
file.write(name+'\n')
def Test():
dir="C:\\Users\\11488\\Desktop"
outfile="binaries.txt"
wildcard = ".txt"
file = open(outfile,"w")
if not file:
print ("cannot open the file %s for writing" % outfile)
ListFileToTxt(dir,file,wildcard, 1)
file.close()
Test()
dir ="C:\\Users\\11488\\Desktop"
print(os.listdir(dir))