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

Python实现读取目录所有文件的文件名并保存到txt文件代码
这篇文章主要介绍了Python实现读取目录所有文件的文件名并保存到txt文件代码,本文分别使用os.listdir和os.walk实现给出两段实现代码,需要的朋友可以参考下

代码: (使用os.listdir)

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 + "\n")
                    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()

代码:(使用os.walk) walk递归地对目录及子目录处理,每次返回的三项分别为:当前递归的目录,当前递归的目录下的所有子目录,当前递归的目录下的所有文件。

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 + "\n")
                     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()

你可能感兴趣的:(python)