check in 14, 15 and timer decorator

Author: shruthilayaj

Diff for: 14/main.py

@@ -2,10 +2,12 @@
 import os
 import re
 
+from aoc2020.utils.decorators import timer
+
 
 def read_input():
     filename = os.path.join(os.path.dirname(__file__), "input.txt")
-    with open(filename) as f:    
+    with open(filename) as f:
         data = f.readlines()
 
     return data
@@ -19,39 +21,40 @@ def _parse_input(records):
         if match:
             mask = match.group(1)
             ones = int(mask.replace("X", "0"), 2)
-            zeros = int(mask.replace("1", "X").replace("0", "1").replace("X", "0"), 2)
+            zeros = int(mask.replace("X", "1"), 2)
             floats = mask.replace("1", "0").replace("X", "1")
 
             continue
 
-        match = re.match("mem\[(\d*)\] = (\d*)", line)
+        match = re.match(r"mem\[(\d*)\] = (\d*)", line)
         mem_address, value = match.group(1), match.group(2)
 
         data.append((int(mem_address), int(value), ones, zeros, floats))
-    
+
     return data
 
 
+@timer
 def docking_data(records):
     mem = {}
     for key, value, ones, zeros, _ in records:
         mem[key] = (value, ones, zeros)
 
     total = functools.reduce(
         lambda x, y: x + y,
-        list(map(lambda v: (v[0] | v[1]) - ((v[0] | v[1]) & v[2]), mem.values())),
+        list(map(lambda v: (v[0] | v[1]) & v[2], mem.values())),
     )
 
     return total
 
 
+@timer
 def docking_data_2(records):
     mem = {}
 
     for k, val, ones, zeros, floats in records:
         base = ((k | int(floats, 2)) ^ int(floats, 2)) | ones
         floats = list(floats)
-        # print(floats)
         a = []
         for i, binary in enumerate(floats):
             if binary == "1":
@@ -69,15 +72,16 @@ def docking_data_2(records):
                     a = ["0"]
                 else:
                     a = [x + "0" for x in a]
-        
+
         for v in a:
             k_ = base | int(v, 2)
             mem[k_] = val
-        
+
     return sum(mem.values())
 
 
 if __name__ == "__main__":
     records = _parse_input(read_input())
+
     print(f"Part 1: {docking_data(records)}")
     print(f"Part 2: {docking_data_2(records)}")

Diff for: 15/__init__.py

@@ -0,0 +1,33 @@
+from aoc2020.utils.decorators import timer
+
+
+@timer
+def rambunctious_recitation(records, count):
+    tracking_dict = {}
+    i = 1
+    while len(records) > 1:
+        last = records.pop(0)
+        tracking_dict[last] = i
+
+        i += 1
+
+    for j in range(i, count):
+        if records:
+            last = records.pop(0)
+
+        if last in tracking_dict:
+            temp = j - tracking_dict[last]
+            tracking_dict[last] = j
+            last = temp
+        else:
+            tracking_dict[last] = j
+            last = 0
+
+    return last
+
+
+if __name__ == "__main__":
+    input_ = [0, 13, 1, 16, 6, 17]
+
+    print(f"Part 1: {rambunctious_recitation(input_[:], 2020)}")
+    print(f"Part 1: {rambunctious_recitation(input_[:], 30000000)}")

Diff for: 15/main.py

@@ -1,61 +1,30 @@
-appnope==0.1.0
-argon2-cffi==20.1.0
-async-generator==1.10
-attrs==20.3.0
+appdirs==1.4.4
+appnope==0.1.2
 backcall==0.2.0
-bleach==3.2.1
-cffi==1.14.4
-cycler==0.10.0
 decorator==4.4.2
-defusedxml==0.6.0
-entrypoints==0.3
-importlib-metadata==3.1.0
-ipykernel==5.3.4
+distlib==0.3.1
+filelock==3.0.12
+flake8==3.8.4
+importlib-metadata==2.0.0
+ipdb==0.13.4
 ipython==7.19.0
 ipython-genutils==0.2.0
-ipywidgets==7.5.1
 jedi==0.17.2
-Jinja2==2.11.2
-jsonschema==3.2.0
-jupyter==1.0.0
-jupyter-client==6.1.7
-jupyter-console==6.2.0
-jupyter-core==4.7.0
-jupyterlab-pygments==0.1.2
-kiwisolver==1.3.1
-MarkupSafe==1.1.1
-matplotlib==3.3.3
-mistune==0.8.4
-nbclient==0.5.1
-nbconvert==6.0.7
-nbformat==5.0.8
-nest-asyncio==1.4.3
-notebook==6.1.5
-numpy==1.19.4
-packaging==20.7
-pandocfilters==1.4.3
+mccabe==0.6.1
 parso==0.7.1
+pbr==5.5.1
 pexpect==4.8.0
 pickleshare==0.7.5
-Pillow==8.0.1
-prometheus-client==0.9.0
 prompt-toolkit==3.0.8
 ptyprocess==0.6.0
-pycparser==2.20
-Pygments==2.7.2
-pyparsing==2.4.7
-pyrsistent==0.17.3
-python-dateutil==2.8.1
-pyzmq==20.0.0
-qtconsole==5.0.1
-QtPy==1.9.0
-Send2Trash==1.5.0
+pycodestyle==2.6.0
+pyflakes==2.2.0
+Pygments==2.7.3
 six==1.15.0
-terminado==0.9.1
-testpath==0.4.4
-tornado==6.1
+stevedore==3.2.2
 traitlets==5.0.5
+virtualenv==20.1.0
+virtualenv-clone==0.5.4
+virtualenvwrapper==4.8.4
 wcwidth==0.2.5
-webencodings==0.5.1
-widgetsnbextension==3.5.1
 zipp==3.4.0

Diff for: __init__py

@@ -0,0 +1,20 @@
+import functools
+import timeit
+from aoc2020.utils.log.base import get_logger
+
+logger = get_logger(__name__)
+
+
+def timer(func):
+    @functools.wraps(func)
+    def timer_wrapper(*args, **kwargs):
+        start = timeit.default_timer()
+
+        result = func(*args, **kwargs)
+
+        elapsed = timeit.default_timer() - start
+        logger.info(f"Completed {func.__name__} in {elapsed:4f} seconds.")
+
+        return result
+
+    return timer_wrapper

Diff for: requirements.

@@ -0,0 +1,10 @@
+import logging
+import logging.config
+import os
+
+
+def get_logger(name):
+    filename = os.path.join(os.path.dirname(__file__), "logging.conf")
+    logging.config.fileConfig(filename)
+
+    return logging.getLogger(name)

Diff for: utils/__init__.py

@@ -0,0 +1,30 @@
+[loggers]
+keys=root,dev
+
+[handlers]
+keys=consoleHandler
+
+[formatters]
+keys=extend,simple
+
+[logger_root]
+level=INFO
+handlers=consoleHandler
+
+[logger_dev]
+level=INFO
+handlers=consoleHandler
+qualname=dev
+propagate=0
+
+[handler_consoleHandler]
+class=StreamHandler
+level=INFO
+formatter=extend
+args=(sys.stdout,)
+
+[formatter_extend]
+format=%(asctime)s - %(name)s - %(levelname)s - %(message)s
+
+[formatter_simple]
+format=%(asctime)s - %(message)s