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

yipeiwu_com6年前Python基础

本文实例为大家分享了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设计】。

相关文章

python基于event实现线程间通信控制

这篇文章主要介绍了python基于event实现线程间通信控制,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下 import thr...

Python基础练习之用户登录实现代码分享

python版本为python3.5 1.要求 1)输入用户名密码 2)认证成功后显示欢迎信息 3)输错三次后锁定 2.需求分析 1)用户信息存储在文件中(login/config...

Python+树莓派+YOLO打造一款人工智能照相机

Python+树莓派+YOLO打造一款人工智能照相机

不久之前,亚马逊刚刚推出了DeepLens 。这是一款专门面向开发人员的全球首个支持深度学习的摄像机,它所使用的机器学习算法不仅可以检测物体活动和面部表情,而且还可以检测类似弹吉他等复杂...

Python进阶-函数默认参数(详解)

一、默认参数 python为了简化函数的调用,提供了默认参数机制: def pow(x, n = 2): r = 1 while n > 0: r *= x n...

python实现linux下使用xcopy的方法

本文实例讲述了python实现linux下使用xcopy的方法。分享给大家供大家参考。具体如下: 这个python函数模仿windows下的xcopy命令编写,可以用在linux下...