|
发表于 2019-10-26 11:18:47
|
显示全部楼层
- import os
- def search_video_start(start_dir, content): # 传入content列表
- os.chdir(start_dir)
-
- for each_file in os.listdir(os.curdir):
- if os.path.isfile(each_file):
- a =os.path.splitext(each_file)
- if a[1] in ['.mp4','.rmvb','.avi']:
- content.append(os.getcwd() + each_file + '\n')
- else:
- search_video_start(each_file, content)
- os.chdir(os.pardir)
- def search_video(start_dir):
- content = [] # 声明个空列表
- search_video_start(start_dir, content)
- dirname, filename = os.path.split(os.path.abspath(__file__))
- os.chdir(dirname) # 切换回脚本所在路径
- with open('videolist.txt','w') as f:
- f.writelines(content)
-
- start_dir = input('请输入待查找的初始目录:')
- search_video(start_dir)
-
复制代码 |
|