|  | 
 
 发表于 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)
                
 | 
 |