Python练习003

题目：编写一个程序，用户输入开始搜索的路径，查找该路径下（包含子文件夹内）所有的视频格式文件（要求查找mp4,mp3,txt的格式即可），并把创建一个文件（list.txt）存放所有找到的文件的路径。
代码：

import os
target = ['.mp4', '.mp3', '.txt']
def search(start_dir):
    os.chdir(start_dir)
    print(os.listdir(os.curdir))
    for each_file in os.listdir(os.curdir):
        suffix = os.path.splitext(each_file)[1]
        if suffix in target:
            file.write(os.getcwd() + os.sep + each_file + '\n')
        if os.path.isdir(each_file):
            search(each_file)
            os.chdir(os.pardir)
    
start_dir = input("请输入搜索的起始路径：")
with open('list.txt','w') as file:
    search(start_dir)