Python代码集pathlib应用之获取指定目录下的所有文件
作者：redrose2100
这篇文章主要介绍了Python代码集pathlib应用之获取指定目录下的所有文件,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
（1）如下代码，默认递归获取指定目录root_dir下的所有文件，当指定recursive参数为False时，则只获取root_dir目录下的所有文件，不会递归的查找，若指定suffix_tuple参数，则可以获取root_dir目录下的指定后缀文件
from pathlib import Path
def get_all_files(root_dir,recursive=True,suffix_tuple=()):
all_files=[]
if Path(root_dir).exists():
if Path(root_dir).is_dir():
if recursive:
for elem in Path(root_dir).glob("**/*"):
if Path(elem).is_file():
suffix=Path(elem).suffix
if not suffix_tuple:
all_files.append(elem)
else:
if suffix in suffix_tuple:
all_files.append(elem)
else:
for elem in Path(root_dir).iterdir():
if Path(elem).is_file():
suffix=Path(elem).suffix
if not suffix_tuple:
all_files.append(elem)
else:
if suffix in suffix_tuple:
all_files.append(elem)
else:
all_files.append(root_dir)
return all_files（2）具体使用方法如下，即测试代码，具体目录path指定为自己存在的目录
if __name__=="__main__":
path="D:/gitee/oepkgs/mugen/testcases/cli-test/acl/oe_test_acl_defaulr_rule.sh"
for elem in get_all_files(path):
print(elem)
print("-------------------------------------------------")
path = "D:/gitee/oepkgs/mugen/testcases/cli-test/acl"
for elem in get_all_files(path):
print(elem)
print("-------------------------------------------------")
path = "D:/gitee/oepkgs/mugen/testcases/cli-test/acl"
for elem in get_all_files(path,False):
print(elem)
print("-------------------------------------------------")
path = "D:/gitee/oepkgs/mugen/testcases/cli-test/acl"
for elem in get_all_files(path, True,(".sh",)):
print(elem)
print("-------------------------------------------------")
path = "D:/gitee/oepkgs/mugen/testcases/cli-test/acl"
for elem in get_all_files(path, True, (".json",)):
print(elem)执行结果如下，第一个当指定path为文件时，则直接将文件作为查询到的返回，最后一个指定.json后缀，因为调试机上没有json文件，所以打印为空
到此这篇关于Python代码集pathlib应用之获取指定目录下的所有文件的文章就介绍到这了,更多相关Python获取指定目录下的所有文件内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！