当前位置： 移动技术网 > IT编程>脚本编程>Python > Python实现读取目录所有文件的文件名并保存到txt文件代码

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

2019年03月24日  | 移动技术网IT编程  | 我要评论

都是爱情惹的祸dj,宋希濂,淘气的粉红女王

代码： (使用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()

您可能感兴趣的文章:

如对本文有疑问，请在下面进行留言讨论，广大热心网友会与你互动！！ 点击进行留言回复

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网