|
|
马上注册,结交更多好友,享用更多功能^_^
您需要 登录 才可以下载或查看,没有账号?立即注册
x
import os
import os.path
def video_search(file_dir):
os.chdir(file_dir)
file_listdir = os.listdir(os.curdir)
videoList = []
ext = ('mp4','avi','rmvb')
for each in file_listdir:
if os.path.isfile(each):
if each.split('.')[1] in ext:
videoList.append(os.getcwd() + os.sep + each + '\n')
f = open('G:\\python\\练习\\课后练习\\030讲\\03\\videoList.txt','a')
f.writelines(videoList)
f.close()
else:
video_search(each)
os.chdir(os.pardir)
file_dir = input('请输入待查找的初始目录:')
video_search(file_dir)
为什么我测试的结果会是下图这样的:
请输入待查找的初始目录:G:\python\练习\课后练习\030讲\03\新建文件夹 (2)
因为代码中,f.writelines(videoList),表示每次循环你都会把videolist中所有的元素写入一遍,所以会重复,
可以 设置
file_name = videoList.append(os.getcwd() + os.sep + each + '\n')
然后 f.writelines(filename),这样会每次循环写入当前的一个元素
|
|