class_iterator.py

import os


class Iterator:
    def __init__(self, name_of_file : str) -> None: 
        self.name_of_file = name_of_file
        self.counter = 0
        self.list = []
        file = open(self.name_of_file, "r")
        for row in file:
            self.list.append(row)
        file.close()

    def __iter__(self):
        return self

    def __next__(self) -> str:
        if self.counter < len(self.list):
            tmp = self.list[self.counter]
            self.counter += 1
            return tmp
        else:
            raise StopIteration
        

if __name__ == "__main__":
    mark = 4
    path_to_dataset = os.path.join( 'application_programming_first_lab_and_dataset','dataset')
    folder_path = path_to_dataset + f'\\{mark}'
    num_of_files = sum(os.path.isfile(os.path.join(folder_path, f))
                       for f in os.listdir(folder_path)) + 1

    iter = Iterator(num_of_files, mark, folder_path)