file_finder.py

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