from dataclasses import dataclass @dataclass class Race: time: int record: int def length_for_button_press(self, duration: int) -> int: length = 0 for i in range(duration, self.time): length += duration return length