题目:编写一个程序,用户输入开始搜索的路径,查找该路径下(包含子文件夹内)所有的视频格式文件(要求查找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)