Python 批量读取文件名并写入txt中

最新推荐文章于 2024-07-05 17:15:30 发布

CVAIDL

最新推荐文章于 2024-07-05 17:15:30 发布

阅读量2.7k

点赞数

本文介绍了一个使用Python实现的宽度优先搜索(BFS)算法来遍历指定目录及其子目录的方法。该脚本可以输出所有文件名，并忽略.meta文件。通过定义不同的回调函数，可以实现对文件和目录的不同处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

[python]view plain copy
#coding=utf-8  
import sys  
import os, glob  
  
reload(sys)  
sys.setdefaultencoding('utf-8')  
  
#输出路径，自行修改  
TxtPath="C://Users//yupu//Desktop//f.txt"  
  
def BFS_Dir(dirPath, dirCallback = None, fileCallback = None):  
    queue = []  
    ret = []  
    f=open(TxtPath,'w')    # r只读，w可写，a追加  
    queue.append(dirPath);  
    while len(queue) > 0:  
        tmp = queue.pop(0)  
        if(os.path.isdir(tmp)):  
            ret.append(tmp)  
            for item in os.listdir(tmp):  
                queue.append(os.path.join(tmp, item))  
            if dirCallback:  
                dirCallback(tmp)  
        elif(os.path.isfile(tmp)):  
            ret.append(tmp)  
            if fileCallback:  
                mPath , ext = os.path.splitext(tmp)  
                names = os.path.split(mPath)  
                if(ext==".meta"):  
                    continue  
                else:  
                   print names[1]  
                   f.write(names[1])  
                   f.write('\n')  
                   fileCallback(tmp)  
    f.close()  
    return ret  
  
def printDir(dirPath):  
    print "dir: " + dirPath  
  
def printFile(dirPath):  
    print "file: " + dirPath  
  
if __name__ == '__main__':  
    while True:  
        path = raw_input("Path:")  
        try:  
            b = BFS_Dir(path , printDir, printFile)  
            print ("\r\n          *******\r\n"+"*********Done*********"+"\r\n          **********\r\n")  
        except:  
            print "Unexpected error:", sys.exc_info()  
        raw_input('press enter key to rehandle')  