Day 7, with given examples

Author: Chris Poch

7-1.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.

7-2.txt

@@ -0,0 +1,7 @@
+shiny gold bags contain 2 dark red bags.
+dark red bags contain 2 dark orange bags.
+dark orange bags contain 2 dark yellow bags.
+dark yellow bags contain 2 dark green bags.
+dark green bags contain 2 dark blue bags.
+dark blue bags contain 2 dark violet bags.
+dark violet bags contain no other bags.

7.py

@@ -0,0 +1,91 @@
+import os
+
+fileName = "7.txt"
+with open(fileName) as file:
+    rules = {}
+    total = 0
+    text = file.readlines()
+    for rule in text:
+        total = total + 1
+        myColor = rule[0:rule.find(" bag")]
+        i = rule.find("contain") + 7
+        others = rule[i:-2]
+        others = others.replace("bags","")
+        others = others.replace("bag","")
+        others = others.split(",")
+        for i in range(len(others)):
+            others[i] = others[i][3:-1]
+        rules[myColor] = others
+        #print(myColor,others)
+    #print(total, len(rules)) #proves rules are for unique colors
+    containsGold = {}
+    containsGold["shiny gold"] = True
+    #for rule in rules:
+    #    if "shiny gold" in rules[rule]:
+    #        containsGold[rule] = True
+    size = 0
+    while size != len(containsGold):
+        size = len(containsGold)
+        for foundColor in list(containsGold):
+            for newColor in rules:
+                if foundColor in rules[newColor]:
+                    containsGold[newColor] = True
+        #print(size,"----------------",len(containsGold))
+        #print(containsGold.keys())
+    print(len(containsGold) - 1) #remove shiny gold
+
+#part 2
+rules = {}
+total = 0
+
+def insidePackages(color):
+    if len(rules[color]) == 0:
+        return 0
+    count = 0
+    for sub in rules[color].keys():
+        #print ("---",sub,rules[color][sub],insidePackages(sub))
+        count = count + (rules[color][sub] * (insidePackages(sub) + 1))
+    #print(color,count)
+    return count
+
+
+with open(fileName) as file:
+#    text = file.readlines()
+    for rule in text:
+        total = total + 1
+        myColor = rule[0:rule.find(" bag")]
+        i = rule.find("contain") + 7
+        others = rule[i:-2]
+        others = others.replace("bags","")
+        others = others.replace("bag","")
+        others = others.split(",")
+        contents = {}
+        for other in others:
+            try:
+                contents[other[3:-1]] = int(other[1])
+            except:
+                pass
+        rules[myColor] = contents
+        #print(myColor,others)
+    #print(total, len(rules)) #proves rules are for unique colors
+
+    #print("------------")
+    #print(rules)
+    #print("------------")
+    insideGold = {}
+    for color in rules["shiny gold"]:
+        insideGold[color] = rules["shiny gold"][color]
+
+    #size = 0
+    #while size != len(insideGold):
+    #    size = len(insideGold)
+    #    for foundColor in list(insideGold):
+    #        for newColor in rules[foundColor].keys():
+    #            insideGold[newColor] = rules[foundColor][newColor]
+
+        #print(size,"----------------",len(containsGold))
+        #print(containsGold.keys())
+    packages = insidePackages("shiny gold")
+    #for item in insideGold.keys():
+    #    packages = packages + insideGold[item]
+    print(packages)