Day 06: Solution of part 2

unstko · unstko · commit 2d3000b0c780 · 2016-12-17T17:21:14.000+01:00
diff --git a/06/log.txt b/06/log.txt
@@ -28,8 +28,6 @@ Given the recording in your puzzle input, what is the error-corrected version of
 
 Your puzzle answer was nabgqlcw.
 
-The first half of this puzzle is complete! It provides one gold star: *
-
 --- Part Two ---
 
 Of course, that would be the message - if you hadn't agreed to use a modified repetition code instead.
@@ -38,4 +36,8 @@ In this modified code, the sender instead transmits what looks like random data,
 
 In the above example, the least common character in the first column is a; in the second, d, and so on. Repeating this process for the remaining characters produces the original message, advent.
 
-Given the recording in your puzzle input and this new decoding methodology, what is the original message that Santa is trying to send?
+Given the recording in your puzzle input and this new decoding methodology, what is the original message that Santa is trying to send?
+
+Your puzzle answer was ovtrjcjh.
+
+Both parts of this puzzle are complete! They provide two gold stars: **
diff --git a/06/solution.py b/06/solution.py
@@ -7,7 +7,8 @@ def __init__(self, nr):
         super().__init__(nr)
         self.length = 1
         self.dicts = []
-        self.message = ""
+        self.message1 = ""
+        self.message2 = ""
 
     def calculate(self, test=False):
         self.read_instructions()
@@ -28,9 +29,11 @@ def calc_message(self):
         for line in self.input:
             i = 0
             for char in line:
-                if i < self.length:
+                if char.isalpha() and i < self.length:
                     self.dicts[i].add(char)
                     i += 1
         for i in range(0, self.length):
-            self.message += self.dicts[i].get_sorted_string_by_value_and_key(1)
-        self.set_solution(1, self.message)
+            self.message1 += self.dicts[i].get_sorted_string_by_value_and_key(1)
+            self.message2 += self.dicts[i].get_sorted_string_by_value_and_key(1, True)
+        self.set_solution(1, self.message1)
+        self.set_solution(2, self.message2)
diff --git a/06/test.py b/06/test.py
@@ -25,3 +25,4 @@ def test_1(self):
 enarar"""
         self.execute_test(test_input)
         self.assertEqual("easter", self.solution.get_solution(1))
+        self.assertEqual("advent", self.solution.get_solution(2))
