一,简介
在日常的调试过程中,经常会会遇到添加批量源文件的情况,本文介绍一种使用python脚本获取指定目录下的所有文件目录和指定文件名的方法,供参考。
二,脚本介绍
import os
def get_dir_and_file(path, file_set, dir_set):
file_list = os.listdir(path)
for file in file_list:
#print(file)
cur_path = os.path.join(path, file)
if os.path.isdir(cur_path):
dir_set.add(cur_path)
get_dir_and_file(cur_path, file_set, dir_set)
else:
if file.endswith(".c"):
file_set.add(cur_path)
if file.endswith(".txt"):
file_set.add(cur_path)
if __name__ == "__main__":
file_set = set([])
dir_set = set([])
get_dir_and_file("./files", file_set, dir_set)
with open('./file_name.txt', 'w', encoding='utf-8') as f: # txtname 根据具体所需的命名即可
for val_name in file_set:
f.write(val_name + '\n')
with open('./dir_name.txt', 'w', encoding='utf-8') as f: # txtname 根据具体所需的命名即可
for val_name in dir_set:
f.write(val_name + '\n')
print('-----------------')
print(file_set)
print('-----------------')
print(dir_set)
print('finish!\r\n')
三,总结
本文主要介绍如何查找指定路径下的目录和文件名,并写到对应的TXT文件中,供参考。欢迎一起讨论交流~