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

Python  /  管理员 发布于 7年前   313

本文实例为大家分享了python批量读取文件名并写入txt中的具体代码，供大家参考，具体内容如下

先说下脚本使用的环境吧，在做项目的过程中需要动态加载图片，使用Unity的Resources.Load方法，但是百十张图片怎么能一 一写下他们的名字作为加载的路径呢？总不能一个一个编辑后存到数组中吧，（虽然我最初是这么做的）。所以必须有一个批量的工具，必须的。

于是乎激发了我的灵感，下面看代码。备注少，不动的给我留言，我会及时回复的。

#coding=utf-8import sysimport 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')

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。