Python代码集pathlib应用之如何获取指定目录下的所有文件

from pathlib import Pathdef 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

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)