|
|

楼主 |
发表于 2018-2-1 20:26:02
|
显示全部楼层
- import os
- def Find_Vedio(directory,save_path):
- flag=0
- save_data=[]
- os.chdir(directory)#改变工作目录
- container=os.walk((os.getcwd()))#获取当前工作目录#遍历该目录下所有子目录
-
- for sub_directory in container:
- for each_file_name in sub_directory[2]:
- splited_file_name=os.path.splitext(each_file_name)#返回[文件名,扩展名]
- if splited_file_name[1] in ['.avi','.mp4','.rmvb']:
- #vedio_name=(sub_directory[0]+'\\'+each_file_name+'\n')
- vedio_name=(sub_directory[0]+os.sep+each_file_name+os.linesep)#改成这句使程序更标准
- save_data.append(vedio_name)
- print(vedio_name)
- flag=1
- save_txt=open(save_path,'w')
- print('保存中.....')
- save_txt.writelines(save_data)
- save_txt.close()
- print('主人我已为你打包完毕~~')
-
- if flag==0:
- print('真是不看片的好小伙!居然找不到!!!')
-
- print('请输入待查找的初始目录:',end='')
- while 1:
- initial_directory = input()
- print('请输入保存路径(默认为【E:\\vedioList.txt】):',end='')
- Savepath=input()
- if Savepath=='':
- Savepath='E:/vedioList.txt'
- if ':\\' not in initial_directory and ':\\'not in Savepath:
- print("格式错误请重新输入:",end='')
- continue
- else:
- print('查找中....')
- Find_Vedio(initial_directory,Savepath)
- print('感谢使用')
- break
复制代码
这是我打的代码正常使用 |
|