|
发表于 2020-2-10 21:40:39
|
显示全部楼层
本楼为最佳答案
 - import os
- import os.path
- def findmedia(file_path):
- global mediatype
- global filename
- file_path1 = file_path
- findtype(file_path1)
- print('%s已搜索完毕,共找到%d个文件' % (file_path, len(filename)))
- if len(filename):
- save = input('是否需要保存文件?【YES/NO】')
- if save in ['YES', 'yes', 'Yes']:
- mediafile = open('D:\\vedioList.txt', 'w', encoding="utf-8")
- mediafile.writelines(filename)
- mediafile.close()
- print('文件已保存在\'D:\\vedioList.txt\'')
- print('再见!')
- def findtype(file_path1):
- print('现在查找到%s' % file_path1)
- try:
- os.chdir(file_path1)
- files = os.listdir(file_path1)
- for eachfile in files:
- filetype = os.path.splitext(file_path1+'\\'+eachfile)
- if os.path.isdir(file_path1 + '\\' + eachfile):
- findtype((file_path1 + '\\' + eachfile))
- elif filetype[1] in mediatype:
- filename.append((file_path1 + '\\' + eachfile + "\n"))
- except:
- pass
- file_path = os.getcwd()
- filename = []
- mediatype = ['.avi', '.rmvb', '.avi']
- file_path = input(('''请确认是否使用当前路径:%s
- 【yes/请输入待查找的初始目录】''' % file_path))
- if os.path.isdir(file_path) and (file_path not in ['YES', 'yes', 'Yes']):
- input('按【回车】开始查找 --- %s --- ...' % (mediatype))
- findmedia((file_path) + '\\')
- elif file_path in ['YES', 'yes', 'Yes']:
- file_path = os.getcwd()
- input('按【回车】开始查找 --- %s --- ...' % (mediatype))
- findmedia((file_path+'\\'))
- else:
- print('输入的路径存在错误,程序结束,88!')
复制代码 |
|