python脚本获取指定目录下文件目录和文件名

一，简介

在日常的调试过程中，经常会会遇到添加批量源文件的情况，本文介绍一种使用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文件中，供参考。欢迎一起讨论交流~