Dict: New module used for day 04

unstko · unstko · commit 511f16c4fb82 · 2016-12-17T16:12:06.000+01:00
- Sortable dictionary
- Get sorted strings with arbitrary length
diff --git a/04/solution.py b/04/solution.py
@@ -1,7 +1,6 @@
 from lib import solution
+from lib.dict import Dict
 import re
-import collections
-import operator
 
 
 class Solution(solution.Solution):
@@ -34,12 +33,11 @@ def process_room_list(self):
 
     @staticmethod
     def get_checksum(string):
-        d = collections.defaultdict(int)
+        dictionary = Dict()
         for c in string:
             if c.isalpha():
-                d[c] -= 1
-        s = collections.OrderedDict(sorted(d.items(), key=operator.itemgetter(1, 0)))
-        checksum = "".join(str(x) for x in s)[:5]
+                dictionary.add(c)
+        checksum = dictionary.get_sorted_string_by_value_and_key(5)
         return checksum
 
     @staticmethod
diff --git a/README.md b/README.md
@@ -36,11 +36,15 @@ My solutions for [adventofcode.com (2016)](http://adventofcode.com/2016) with Py
 - Set and get side lengths
 - Check if triangle is possible
 
+### Dict
+- Sortable dictionary
+- Get sorted strings with arbitrary length
+
 ## My solutions
 | Day | Task | Python modules | My modules |
 | --- | ---- | -------------- | ---------- |
 | 01 | Find shortest path and cross point | copy, math, PIL | Point2D, Map |
 | 02 | Find codes at keypads | os, json | Keypad |
 | 03 | Check triangles | re | Triangle |
-| 04 | Encrypt and check room names | re, collections, operator | - |
+| 04 | Encrypt and check room names | re, collections, operator | Dict |
 | 05 | Create password from MD5 hashes | hashlib | - |
diff --git a/lib/dict.py b/lib/dict.py
@@ -0,0 +1,20 @@
+import collections
+import operator
+
+
+class Dict:
+    def __init__(self):
+        self.dict = collections.defaultdict(int)
+
+    def reset(self):
+        self.dict.clear()
+
+    def add(self, key):
+        self.dict[key] -= 1
+
+    def get_sorted_string_by_value_and_key(self, length, reverse=False):
+        sorted_list = self.get_sorted_list_by_value_and_key(reverse)
+        return "".join(str(x) for x in sorted_list)[:length]
+
+    def get_sorted_list_by_value_and_key(self, reverse=False):
+        return collections.OrderedDict(sorted(self.dict.items(), key=operator.itemgetter(1, 0), reverse=reverse))
