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

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

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

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

#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')

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

您可能感兴趣的文章:

• python+opencv 读取文件夹下的所有图像并批量保存ROI的方法
• Python从数据库读取大量数据批量写入文件的方法
• 使用python批量读取word文档并整理关键信息到excel表格的实例
• Python Pandas批量读取csv文件到dataframe的方法
• python批量读取txt文件为DataFrame的方法
• Python实现批量读取图片并存入mongodb数据库的方法示例
• Python实现批量读取word中表格信息的方法
• Python中将变量按行写入txt文本中的方法
• Python将列表数据写入文件（txt, csv，excel）
• Python打开文件,将list、numpy数组内容写入txt文件中的方法