import os def traversal(path, skip_num, ending): curr_num = 0 if os.path.isdir(path): for filename_1 in os.listdir(path): filename_1 = path + '/' + filename_1 filename_1 = filename_1.replace('//', '/') if os.path.isdir(filename_1): for filename_2 in os.listdir(filename_1): filename_2 = filename_1 + '/' + filename_2 filename_2 = filename_2.replace('//', '/') if (os.path.isfile(filename_2) and filename_2.endswith(ending)): curr_num += 1 if curr_num >= skip_num: print(curr_num, ': ', filename_2) yield filename_2 else: print(curr_num, ': ', filename_2, ' -- SKIPPED') elif os.path.isfile(filename_1) and filename_1.endswith(ending): curr_num += 1 if curr_num >= skip_num: print(curr_num, ': ', filename_1) yield filename_1 else: print(curr_num, ': ', filename_1, ' -- SKIPPED') elif os.path.isfile(path): if path.endswith(ending): print(path) yield path