您的位置:首页 > 编程语言 > Python开发

关于os的一系列问题:寻找目录中的视频文件,并存储路径

2017-01-03 15:07 459 查看
需求:编写一个程序,用户输入开始搜索的路径,查找该路径下(包含子文件夹内)所有的视频格式文件(要求查找MP4、rmvb、avi的格式即可),并创建一个文件(videoList.txt)存放所有找到的文件的路径。

# -*- coding: utf-8 -*-

import os

def search_file(start_dir):
os.chdir(start_dir) #os.chdir() 方法用于改变当前工作目录到指定的路径。
for each_file in os.listdir(os.curdir):
# print (each_file)
if os.path.splitext(each_file)[1] in [".mp4",".rmvb",".avi"]:
# print (os.getcwd()+os.sep+each_file+os.linesep)
txt_write(os.getcwd()+os.sep+each_file+os.linesep)
#os.linesep字符串给出当前平台使用的行终止符。例如,Windows使用'\r',Linux使用''而Mac使用'\r'。
if os.path.isdir(each_file):
search_file(each_file) #递归调用
os.chdir(os.pardir)  # 切换到上级目录 递归调用后,切记返回上一层目录

def txt_write(content):
file_write=open("E:\\Python Program\\videoList.txt","a")
print (content,file=file_write)
file_write.close()

start_dir=input(r"请输入待查找的初始目录:")
search_file(start_dir)


参考:http://bbs.fishc.com/forum.php?mod=viewthread&tid=45649&extra=page%3D1%26filter%3Dtypeid%26typeid%3D398
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  python