FarmerLi
4/13/2018 - 5:56 AM

List specific files under the specified folder.

List specific files under the specified folder.

from os import listdir
from os.path import splitext

def list_specific_suffix_files(directory, ispure=False, ext=".txt") :
    files = []
    f_list = listdir(directory)
    for f in f_list:
        name = splitext(f)
        if name[1] == ext:
            if ispure:
                files.append(name[0])
            else :
                files.append(f)
    return files