Bagaimana untuk mendapatkan semua fail dalam direktori tertentu menggunakan aplikasi pathlib set kod Python

(1) Kod berikut, secara lalai, memperoleh semua fail dalam direktori root_dir yang ditentukan Apabila parameter rekursif ditentukan sebagai Palsu, hanya semua fail dalam direktori root_dir diperoleh, dan tiada carian rekursif dilakukan parameter suffix_tuple ditentukan, kemudian Anda boleh mendapatkan fail suffix yang ditentukan dalam direktori 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) Kaedah penggunaan khusus adalah seperti berikut, iaitu kod ujian, laluan direktori khusus ditentukan sebagai direktori di mana ia wujud

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)

hasil pelaksanaan Seperti yang ditunjukkan di bawah, apabila laluan pertama ditentukan sebagai fail, fail dikembalikan terus sebagai hasil pertanyaan dan yang terakhir menentukan akhiran .json . Kerana tiada fail json pada mesin penyahpepijatan, cetakan kosong

Atas ialah kandungan terperinci Bagaimana untuk mendapatkan semua fail dalam direktori tertentu menggunakan aplikasi pathlib set kod Python. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!