summaryrefslogtreecommitdiffstats
path: root/day2/__init__ (conflicted copy 2023-12-03 150713).py
blob: 1191e0f0a4e611d39748a8043ca121bb36b2d6b7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# -*- coding: utf-8 -*-
from abc import ABC
from enum import Enum
from typing import Iterator

from aoc import BaseAssignment, I, T


class Color(Enum):
    red = 'red'
    green = 'green'
    blue = 'blue'

ColorCount = tuple[int, Color]

class Assignment(BaseAssignment, ABC):
    def parse_item(self, item: str) -> list[tuple[ColorCount, ...]]:
        _, items = item.split(': ')

        return [
            tuple(
                tuple((
                    int(_.split(' ')[0]),
                    Color(_.split(' ')[1])
                ))
                for _ in pair.split(', ')
            )
            for pair in items.split('; ')
        ]


class AssignmentOne(Assignment):
    example_result = 8
    def run(self, input: Iterator[I]) -> T:
        for i in input:
            print(i)


class AssignmentTwo(Assignment):
    pass