Added day 7

3 files changed, 661 insertions, 0 deletions

diff --git a/day7/__init__.py b/day7/__init__.py
new file mode 100644
index 0000000..375c67a
--- /dev/null
+++ b/day7/__init__.py
@@ -0,0 +1,58 @@
+import re
+from functools import lru_cache
+from typing import Iterator, Any, Generator, Dict
+
+from aoc import BaseAssignment
+
+rule_matcher = re.compile(
+    r'(?P<bag>\w+\ \w+) bags contain (?P<contains>no other bags|(((, )?\d \w+\ \w+ bags?)+))\.')
+contains_matcher = re.compile(r'(?P<count>\d+) (?P<bag>\w+\ \w+) bags?')
+
+
+class Assignment(BaseAssignment):
+    def get_bag_contents(self, contents: str) -> dict:
+        content_dict = {}
+        if contents != 'no other bags':
+            for content in contents.split(', '):
+                count = contains_matcher.match(content).groupdict()
+                content_dict[count['bag']] = int(count['count'])
+
+        return content_dict
+
+    def read_input(self, example=False) -> Dict:
+        self.mapping = {}
+
+        for line in super().read_input(example):
+            match = rule_matcher.match(line)
+            rule = match.groupdict()
+            self.mapping[rule['bag']] = self.get_bag_contents(rule['contains'])
+
+
+class AssignmentOne(Assignment):
+    @lru_cache
+    def contains_shiny_gold_bag(self, bag):
+        if 'shiny gold' in self.mapping[bag]:
+            return True
+
+        return any([
+            self.contains_shiny_gold_bag(containing_bag)
+            for containing_bag in self.mapping[bag].keys()
+        ])
+
+    def run(self, input) -> Any:
+        nr_of_possibilities = 0
+        for bag, contains in self.mapping.items():
+            if self.contains_shiny_gold_bag(bag):
+                nr_of_possibilities += 1
+        return nr_of_possibilities
+
+
+class AssignmentTwo(Assignment):
+    def nr_of_bags_inside(self, bag):
+        return sum([
+            count + (count * self.nr_of_bags_inside(containing_bag))
+            for containing_bag, count in self.mapping[bag].items()
+        ])
+
+    def run(self, input) -> Any:
+        return self.nr_of_bags_inside('shiny gold')
diff --git a/day7/example.txt b/day7/example.txt
new file mode 100644
index 0000000..e7ba02e
--- /dev/null
+++ b/day7/example.txt
@@ -0,0 +1,9 @@
+light red bags contain 1 bright white bag, 2 muted yellow bags.
+dark orange bags contain 3 bright white bags, 4 muted yellow bags.
+bright white bags contain 1 shiny gold bag.
+muted yellow bags contain 2 shiny gold bags, 9 faded blue bags.
+shiny gold bags contain 1 dark olive bag, 2 vibrant plum bags.
+dark olive bags contain 3 faded blue bags, 4 dotted black bags.
+vibrant plum bags contain 5 faded blue bags, 6 dotted black bags.
+faded blue bags contain no other bags.
+dotted black bags contain no other bags.
\ No newline at end of file
diff --git a/day7/input.txt b/day7/input.txt
new file mode 100644
index 0000000..555513a
--- /dev/null
+++ b/day7/input.txt
@@ -0,0 +1,594 @@
+light salmon bags contain 5 dark brown bags, 2 dotted coral bags, 5 mirrored turquoise bags.
+drab magenta bags contain 1 vibrant purple bag, 5 dark lime bags, 2 clear silver bags.
+striped coral bags contain 2 dim lime bags.
+drab cyan bags contain 1 dark lime bag.
+bright purple bags contain 5 posh salmon bags, 1 posh blue bag, 1 shiny lavender bag.
+dull tomato bags contain 2 shiny maroon bags, 1 posh salmon bag, 4 posh teal bags, 5 dark orange bags.
+faded aqua bags contain 3 striped salmon bags, 5 dim purple bags, 3 shiny gold bags, 4 drab indigo bags.
+drab maroon bags contain 2 dark gray bags, 3 mirrored tan bags, 4 pale blue bags, 4 dim black bags.
+posh blue bags contain 1 mirrored maroon bag, 4 striped lavender bags.
+dull orange bags contain 2 light aqua bags, 2 striped chartreuse bags, 4 dim bronze bags.
+vibrant fuchsia bags contain 1 clear orange bag, 1 plaid turquoise bag, 3 posh maroon bags, 4 light silver bags.
+posh violet bags contain 5 plaid violet bags, 5 vibrant turquoise bags, 1 pale orange bag.
+light gray bags contain 5 vibrant tan bags, 4 shiny tomato bags, 5 muted olive bags.
+striped teal bags contain 2 drab cyan bags, 3 dull tomato bags, 1 light gold bag, 1 dark beige bag.
+muted orange bags contain 3 pale blue bags, 1 muted coral bag.
+wavy magenta bags contain 1 dotted crimson bag.
+dark fuchsia bags contain 3 plaid yellow bags, 4 dim green bags.
+drab beige bags contain 4 dull brown bags.
+light olive bags contain 2 posh magenta bags, 4 dim crimson bags.
+wavy chartreuse bags contain 3 plaid magenta bags.
+dim cyan bags contain 4 light lime bags, 3 dim crimson bags, 1 striped green bag.
+vibrant violet bags contain 2 dark teal bags, 5 vibrant turquoise bags.
+drab tomato bags contain 2 vibrant salmon bags, 5 bright salmon bags.
+mirrored lime bags contain 5 striped teal bags, 3 dim lavender bags, 4 dim cyan bags, 2 dotted green bags.
+wavy plum bags contain 4 plaid magenta bags.
+dim brown bags contain 1 dull yellow bag, 2 plaid beige bags.
+posh magenta bags contain 4 vibrant turquoise bags.
+posh maroon bags contain 3 dark teal bags, 2 striped plum bags.
+pale magenta bags contain 4 dim chartreuse bags, 3 mirrored salmon bags.
+shiny blue bags contain 2 shiny violet bags, 3 pale red bags, 5 pale salmon bags.
+clear tan bags contain 3 wavy lavender bags.
+pale bronze bags contain 3 muted chartreuse bags.
+pale tomato bags contain 1 dotted brown bag, 1 posh lavender bag, 2 striped violet bags.
+dotted lime bags contain 3 light magenta bags, 2 clear yellow bags, 3 faded olive bags.
+shiny tan bags contain 2 dotted fuchsia bags, 5 dull crimson bags, 5 clear orange bags.
+light lavender bags contain 3 shiny tomato bags, 5 drab beige bags, 3 mirrored yellow bags.
+wavy bronze bags contain 2 muted green bags, 3 dotted bronze bags.
+faded brown bags contain 3 drab gold bags, 3 dotted red bags, 1 dim white bag.
+muted cyan bags contain 4 striped turquoise bags.
+faded tan bags contain 1 muted red bag, 1 mirrored tan bag, 4 shiny purple bags.
+dark violet bags contain 4 posh tomato bags.
+posh purple bags contain 3 shiny violet bags, 4 clear indigo bags.
+vibrant orange bags contain 1 dim salmon bag, 1 vibrant olive bag, 4 drab fuchsia bags.
+dull blue bags contain 5 pale lavender bags, 3 dark lavender bags, 1 mirrored yellow bag.
+muted white bags contain 3 dark salmon bags, 4 posh bronze bags, 5 pale plum bags.
+striped lavender bags contain 4 faded plum bags, 2 dark tomato bags, 2 plaid teal bags.
+dark gold bags contain 4 dark lime bags.
+dull magenta bags contain 2 muted chartreuse bags.
+striped cyan bags contain 4 dotted red bags, 3 drab lime bags.
+dotted bronze bags contain 1 clear magenta bag, 2 bright salmon bags.
+clear lavender bags contain 4 dim lime bags.
+plaid salmon bags contain 2 dark orange bags.
+posh silver bags contain 5 muted fuchsia bags, 4 muted crimson bags.
+bright plum bags contain 3 vibrant salmon bags, 3 clear tan bags.
+posh beige bags contain 3 mirrored lavender bags, 4 pale olive bags, 2 dotted black bags.
+bright yellow bags contain 3 mirrored violet bags.
+plaid green bags contain 5 striped black bags, 5 posh bronze bags, 5 mirrored teal bags.
+plaid coral bags contain 5 plaid beige bags, 5 drab chartreuse bags.
+dotted maroon bags contain 4 dotted indigo bags, 4 dotted coral bags.
+dotted chartreuse bags contain 2 wavy turquoise bags, 2 posh indigo bags, 3 plaid silver bags, 2 mirrored salmon bags.
+dark yellow bags contain 1 posh brown bag, 4 light tan bags, 2 clear fuchsia bags, 5 dotted black bags.
+dull lavender bags contain 4 plaid green bags, 1 striped plum bag, 3 drab teal bags.
+bright maroon bags contain 1 dotted tan bag, 4 light olive bags, 5 posh cyan bags.
+vibrant gold bags contain 2 faded gold bags.
+dim lavender bags contain 3 dark silver bags, 1 striped violet bag, 2 dotted bronze bags, 1 striped black bag.
+light tomato bags contain 2 striped chartreuse bags.
+dull yellow bags contain 5 drab tomato bags.
+dotted coral bags contain 1 dark blue bag, 1 mirrored teal bag, 1 drab tomato bag.
+pale lime bags contain 3 clear cyan bags, 5 pale silver bags, 4 wavy olive bags.
+wavy red bags contain 3 vibrant gray bags, 4 shiny teal bags, 4 clear maroon bags.
+pale beige bags contain 2 shiny teal bags, 2 posh fuchsia bags, 4 dark aqua bags, 4 posh orange bags.
+striped tomato bags contain 2 light coral bags.
+shiny violet bags contain 4 vibrant chartreuse bags, 3 wavy lavender bags.
+plaid teal bags contain 3 mirrored teal bags.
+clear black bags contain 2 faded beige bags, 4 faded tan bags, 1 light salmon bag, 5 wavy white bags.
+clear yellow bags contain 2 dull tan bags, 5 plaid gold bags.
+faded green bags contain 5 light brown bags.
+bright coral bags contain 5 dotted teal bags, 2 mirrored gray bags, 3 wavy lime bags.
+muted bronze bags contain 3 dim maroon bags, 3 shiny beige bags, 4 light bronze bags.
+vibrant crimson bags contain 4 dim brown bags, 3 dotted maroon bags, 4 vibrant gold bags, 5 dull tomato bags.
+pale green bags contain 4 plaid beige bags.
+dull green bags contain 2 clear salmon bags, 2 dark bronze bags.
+dotted tan bags contain 1 dim tan bag.
+muted plum bags contain 3 faded crimson bags, 5 muted violet bags, 5 striped violet bags, 2 drab silver bags.
+faded coral bags contain 2 dark plum bags, 1 muted cyan bag, 3 vibrant beige bags.
+vibrant brown bags contain 4 striped maroon bags, 4 bright cyan bags.
+drab coral bags contain 3 vibrant coral bags, 3 dark aqua bags.
+striped chartreuse bags contain 2 vibrant salmon bags, 2 posh bronze bags, 3 drab teal bags.
+bright indigo bags contain 4 faded plum bags.
+clear olive bags contain 4 muted turquoise bags, 1 dotted olive bag.
+drab yellow bags contain 1 dim brown bag, 4 plaid green bags, 2 posh red bags.
+striped purple bags contain 2 mirrored orange bags, 4 vibrant chartreuse bags, 1 pale orange bag, 2 shiny green bags.
+mirrored olive bags contain 1 faded gold bag, 1 vibrant chartreuse bag, 5 faded brown bags, 3 wavy indigo bags.
+dull aqua bags contain 3 muted black bags, 2 muted orange bags, 2 pale lime bags, 4 dark plum bags.
+light lime bags contain 5 plaid beige bags, 5 shiny gold bags, 3 plaid plum bags.
+vibrant beige bags contain 2 posh maroon bags, 3 dotted plum bags.
+mirrored yellow bags contain 4 dotted crimson bags, 3 drab olive bags, 1 faded bronze bag, 2 striped violet bags.
+mirrored plum bags contain 4 dotted gray bags, 1 striped fuchsia bag, 3 dim yellow bags.
+faded beige bags contain 5 striped chartreuse bags, 3 faded bronze bags, 3 dark lime bags.
+drab crimson bags contain 2 dull green bags, 1 muted teal bag.
+shiny salmon bags contain 3 wavy olive bags, 1 faded crimson bag, 1 clear chartreuse bag.
+vibrant turquoise bags contain 4 drab black bags.
+dim yellow bags contain 5 shiny blue bags.
+wavy maroon bags contain 5 pale brown bags.
+dark coral bags contain 2 bright teal bags, 3 bright green bags, 3 drab coral bags.
+bright beige bags contain 3 faded salmon bags, 5 plaid coral bags.
+posh gray bags contain 5 bright silver bags, 1 pale brown bag.
+plaid orange bags contain 3 faded olive bags, 4 muted fuchsia bags, 5 vibrant turquoise bags.
+faded indigo bags contain 1 muted silver bag, 3 dim green bags, 3 faded purple bags, 1 plaid beige bag.
+drab lavender bags contain 5 light blue bags.
+light coral bags contain 3 pale orange bags, 2 plaid teal bags.
+posh indigo bags contain 2 drab olive bags, 2 dim cyan bags, 2 light teal bags.
+muted fuchsia bags contain 2 wavy white bags, 1 dark tan bag.
+muted brown bags contain 5 drab coral bags, 4 dull brown bags, 4 mirrored black bags.
+bright aqua bags contain 3 clear orange bags.
+faded magenta bags contain 1 striped violet bag, 1 dull lavender bag.
+clear brown bags contain 4 mirrored green bags, 1 clear fuchsia bag, 2 vibrant violet bags, 1 shiny aqua bag.
+clear green bags contain 3 dotted brown bags, 2 faded plum bags, 3 dotted beige bags, 2 striped brown bags.
+dull violet bags contain 2 vibrant violet bags, 4 light coral bags, 3 clear black bags.
+light gold bags contain no other bags.
+striped silver bags contain 4 clear bronze bags, 1 striped plum bag, 3 dull yellow bags.
+dull indigo bags contain 5 wavy olive bags, 3 shiny tomato bags, 3 vibrant lime bags, 4 dark beige bags.
+dark teal bags contain 2 dark blue bags, 4 mirrored teal bags, 4 posh bronze bags, 1 muted gold bag.
+light cyan bags contain 3 drab olive bags, 2 plaid teal bags, 5 vibrant indigo bags.
+clear turquoise bags contain 3 bright teal bags.
+clear white bags contain 4 faded black bags.
+plaid olive bags contain 3 vibrant blue bags, 5 posh gold bags.
+drab turquoise bags contain 3 striped black bags, 1 dull brown bag, 5 dark lime bags.
+plaid tomato bags contain 1 dim orange bag.
+wavy lime bags contain 3 muted crimson bags, 2 shiny green bags.
+dim salmon bags contain 4 faded black bags, 4 mirrored teal bags, 4 drab tomato bags, 2 vibrant green bags.
+pale lavender bags contain 2 striped violet bags.
+pale violet bags contain 5 dim magenta bags, 1 muted cyan bag, 5 clear red bags, 2 pale fuchsia bags.
+dark purple bags contain 2 light red bags.
+drab blue bags contain 2 dark blue bags, 2 dotted bronze bags, 4 bright teal bags.
+wavy white bags contain 3 dim red bags, 1 vibrant chartreuse bag, 3 muted gray bags.
+muted beige bags contain 1 shiny purple bag.
+dim aqua bags contain 1 muted yellow bag, 3 drab green bags, 2 plaid magenta bags.
+pale orange bags contain 5 mirrored teal bags, 4 dark aqua bags, 5 shiny gold bags.
+clear coral bags contain 5 dark white bags, 5 light blue b