advent-of-code/2017-python/solutions/day_01.py

from solutions import BaseSolution


class Solution(BaseSolution):
    input_file = "01.txt"

    def __str__(self):
        return "Day 1: Inverse Captcha"

    def solve(self, puzzle_input, distance=1):
        pi_length = len(puzzle_input)
        return sum(
            int(puzzle_input[pos])
            for pos in range(pi_length)
            if puzzle_input[pos] == puzzle_input[(pos + distance) % pi_length]
        )

    def solve_again(self, puzzle_input):
        distance = len(puzzle_input) // 2
        return self.solve(puzzle_input, distance)


if __name__ == "__main__":
    solution = Solution()
    solution.show_results()