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文件中,供参考。欢迎一起讨论交流~

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

音频金钟小赵

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值