def get_all_files()

in pyhanabi/common_utils/helper.py [0:0]

• 12 lines of code
• 8 McCabe index (conditional complexity)

def get_all_files(root, file_extension, contain=None):
    files = []
    for folder, _, fs in os.walk(root):
        for f in fs:
            if file_extension is not None:
                if f.endswith(file_extension):
                    if contain is None or contain in os.path.join(folder, f):
                        files.append(os.path.join(folder, f))
            else:
                if contain in f:
                    files.append(os.path.join(folder, f))
    return files