|
马上注册,结交更多好友,享用更多功能^_^
您需要 登录 才可以下载或查看,没有账号?立即注册
x
import os
def search_video(start_dir):
os.chdir(start_dir)
content = []
for each_file in os.listdir(os.curdir):
type1 = os.path.splitext(each_file)[1]
if type1 in [".avi", ".mp4", ".rmvb"]:
content.append(os.getcwd() + os.sep + each_file + os.linesep)
if os.path.isdir(each_file):
search_video(each_file)
os.chdir(os.pardir)
return content
program_dir = os.getcwd()
start_dir = input("请输入待查找的初始目录:")
vediolist = search_video(start_dir)
file = open(program_dir + os.sep + "vedioList.txt","w")
file.writelines(vediolist)
file.close()
你要和小甲鱼的不一样,就传两个参数,递归时把列表一起传进去
- import os
- def search_video(content,start_dir):
- os.chdir(start_dir)
- for each_file in os.listdir(os.curdir):
- type1 = os.path.splitext(each_file)[1]
- if type1 in [".avi", ".mp4", ".rmvb",".ts"]:
- content.append(os.getcwd() + os.sep + each_file + os.linesep)
- if os.path.isdir(each_file):
- search_video(content,each_file)
- os.chdir(os.pardir)
- return content
- program_dir = os.getcwd()
- content = []
- #start_dir = input("请输入待查找的初始目录:")
- start_dir = r'D:\python\test'
- vediolist = search_video(content,start_dir)
- print(vediolist)
- file = open(program_dir + os.sep + "vedioList.txt","w")
- file.writelines(vediolist)
- file.close()
复制代码
|
|