from lib import get_data


def part_1(data):
    bags = {}
    for line in data.splitlines():
        left, right = line.split(" contain ")
        lb = " ".join(left.split()[:2])
        rbs = right.replace(" bag", "").replace(" bags", "").replace(".", "").split(",")
        if "no others" in rbs:
            rbs = []
        else:
            rbs = [[int(v.split()[0]), " ".join(v.split()[1:])] for v in rbs]
        assert lb not in bags
        bags[lb] = rbs

    for lb, rbs in bags.items():
        for t in rbs:
            if t[0] > 1:
                t[1] = t[1][:-1]

    can_hold = set(["shiny gold"])
    can_hold_count = -1
    while can_hold_count != len(can_hold):
        can_hold_count = len(can_hold)
        for lb, rbs in bags.items():
            for _, bag_color in rbs:
                if bag_color in can_hold:
                    can_hold.add(lb)
    print(len(can_hold) - 1)

    def count(color):
        r = 1  # the bag itself
        for c, icolor in bags[color]:
            r += c * count(icolor)
        return r

    print(count("shiny gold") - 1)


def main():
    data = get_data(__file__)
    part_1(data)


if __name__ == "__main__":
    main()