A small python script to speed up input reading at programming competitions.

input_stage.py

class Reader:
    def __init__(self, path):
        self.contents = open(path, 'r').read().split("\n")
    
    def readList(self, line_index):
        return list(map(int, self.contents[line_index].split(" ")))

    def readInt(self, line_index):
        return int(self.contents[line_index])

    def readGroup(self, line_index, length_pattern):
        return_list = []

        for i in range(length_pattern):
            return_list.append(self.readList(line_index + i))

        return return_list

    def readRepeatingGroup(self, line_index, length_pattern, iterations):
        return_list = []

        for i in range(iterations):
            self.readGroup(line_index + length_pattern*i, length_pattern)

        return return_list