获取目录下的所有文件

原创于 2020-02-28 02:24:09 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

工具模块专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一个Python函数，用于递归地遍历指定的文件夹路径，收集并返回所有以.py结尾的文件的绝对路径。该函数通过判断文件类型，排除了非Python文件和特定初始化文件，适用于需要对大量Python源代码进行管理和处理的场景。

def allfile(all_file_path, li=[]):
	"""
	获取目录中所有文件，
	:param all_file_path:  文件夹路径
	:li：是一个空列表， 
	:return: 返回值是文件abspath组成的列表
	"""
	li_path = os.listdir(all_file_path)
	if not li_path:
		return
	for i in li_path:
		file_path = os.path.join(all_file_path, i)
		if os.path.isdir(file_path):

			allfile(file_path, li)
		elif os.path.isfile(file_path):
			if not i.endswith('.py') or i == '__init__.py':
				continue
			# the_file_path = os.path.dirname(file_path).split('aaa/')[1]
			the_file_path = os.path.abspath(file_path)
			li.append(the_file_path)

	return li