Day 07: Solution of part 2

Author: unstko

07/log.txt

@@ -14,8 +14,6 @@ How many IPs in your puzzle input support TLS?
 
 Your puzzle answer was 115.
 
-The first half of this puzzle is complete! It provides one gold star: *
-
 --- Part Two ---
 
 You would also like to know which IPs support SSL (super-secret listening).
@@ -28,4 +26,8 @@ aba[bab]xyz supports SSL (aba outside square brackets with corresponding bab wit
 xyx[xyx]xyx does not support SSL (xyx, but no corresponding yxy).
 aaa[kek]eke supports SSL (eke in supernet with corresponding kek in hypernet; the aaa sequence is not related, because the interior character must be different).
 zazbz[bzb]cdb supports SSL (zaz has no corresponding aza, but zbz has a corresponding bzb, even though zaz and zbz overlap).
-How many IPs in your puzzle input support SSL?
+How many IPs in your puzzle input support SSL?
+
+Your puzzle answer was 231.
+
+Both parts of this puzzle are complete! They provide two gold stars: **

07/solution.py

@@ -8,7 +8,9 @@ def __init__(self, nr):
         self.match_outside = re.compile(r'([^[\]]+)(?:$|\[)')
         self.match_inside = re.compile(r'\[(.*?)\]')
         self.match_abba = re.compile(r'.*(.)(?!\1)(.)\2\1.*')
+        self.match_aba = re.compile(r'(?=((.)(?!\2)(.)\2))\w')
         self.count1 = 0
+        self.count2 = 0
 
     def calculate(self, test=False):
         self.read_instructions()
@@ -23,7 +25,12 @@ def check_ips(self):
             words_inside = self.match_inside.findall(line)
             if self.contains_abba(words_outside) and not self.contains_abba(words_inside):
                 self.count1 += 1
+            aba_ouside = self.findall_aba(words_outside)
+            aba_inside = self.findall_aba(words_inside)
+            if self.exists_corresponding_aba(aba_ouside, aba_inside):
+                self.count2 += 1
         self.set_solution(1, self.count1)
+        self.set_solution(2, self.count2)
 
     def contains_abba(self, words):
         contains = False
@@ -33,3 +40,23 @@ def contains_abba(self, words):
                     contains = True
                     break
         return contains
+
+    def findall_aba(self, words):
+        list_aba = []
+        if words:
+            for word in words:
+                match = self.match_aba.findall(word)
+                if match:
+                    for m in match:
+                        list_aba.append(m[0])
+        return list_aba
+
+    @staticmethod
+    def exists_corresponding_aba(words1, words2):
+        exists = False
+        for word in words1:
+            word_correspond = ""+word[1]+word[0]+word[1]
+            if word_correspond in words2:
+                exists = True
+                break
+        return exists

07/test.py

@@ -1,7 +1,7 @@
 from lib import test
 
 
-class Test(test.Test):
+class Test07(test.Test):
     @classmethod
     def setUpClass(cls):
         cls.nr = '07'