import os
File_list = []
def search_file(start_dir):
os.chdir(start_dir) # 改变当前工作目录 # 获取到当前文件夹下面所有的文件 os.curdir 表示当前目录
for each_file in os.listdir(os.curdir): # 返回os.curdir指定的文件夹包含的文件或文件夹的名字的列表。 # d://data中所有的文件
# 如果是文件
if os.path.isfile(each_file):
file_ext = os.path.splitext(each_file)[1] # 获取到文件的后缀
if file_ext in ['.pdf', '.csv', '.word', '.txt']:
# d:\data \ 1彼得兔影评.csv 结束标志
File_list.append(os.getcwd()+os.sep+each_file+os.linesep)
# 如果each_file是文件夹
if os.path.isdir(each_file):
search_file(each_file) # 递归调用
os.chdir(os.pardir) # 递归调用后切记返回上一层目录
return File_list
if __name__ == '__main__':
start_dir = input("请输入待查找的初始目录:")
File_list = search_file(start_dir)
print(File_list)
f = open(os.getcwd() + os.sep + 'File_list.txt', 'w+') # 存储
f.writelines(File_list)
f.seek(0, 0)
print(f.read())
f.close()