runner for 1 day

Author: wimglenn

aoc_wim/aoc2016/q08.py

@@ -22,7 +22,7 @@ def part_a(data):
     return A.sum()
 
 
-def part_b(data, dump=False):
+def part_b(data, dump=True):
     A = parsed(data)
     if dump:
         print("\n")

aoc_wim/aoc2018/q07.py

@@ -61,8 +61,8 @@ def work(data, n_workers=4, delay=60):
                     w.t = 1
                 else:
                     w.t = ord(x) - ord('A') + 1 + delay
-        log.debug('%3d: %s   %s', t, ' '.join(w.has for w in workers), text)
         t += 1
+        log.info('%4d: %s   %s', t, ' '.join(w.has for w in workers), text)
     result = SimpleNamespace(text=text, n_iterations=t)
     return result
 

aoc_wim/aoc2018/q10.py

@@ -42,7 +42,7 @@ def varianceish(a):
     return a.std(axis=1)[:2].sum()
 
 
-def ocr(a, lettersize, draw=False):
+def ocr(a, lettersize, draw=True):
     ps = a[:2].copy()
     ps -= ps.min(axis=1).reshape(-1, 1)
     W, H = ps.max(axis=1) + 1
@@ -74,7 +74,7 @@ def ocr(a, lettersize, draw=False):
     return ''.join(letters)
 
 
-def part_ab(data, debug=False, lettersize=(10, 8)):
+def part_ab(data, debug=True, lettersize=(10, 8)):
     numbers = [int(x) for x in re.findall("-?\d+", data)]
     a = np.array(numbers).reshape(-1, 4).T
     t = 0

aoc_wim/aoc2018/q20.py

@@ -89,6 +89,7 @@ def part_ab(data):
     assert a == expected
 
 
+render(data)
 a, b = part_ab(data)
 print("part a:", a)
 print("part b:", b)

aoc_wim/aoc2018/q24.py

@@ -76,7 +76,7 @@ def attack(self):
         kills = self.damage(self.target) // self.target.hp
         kills = min(kills, self.target.n)
         self.target.n -= kills
-        log.debug(
+        log.info(
             "%s group %d attacks defending group %d, killing %d units",
             self.system.name, self.id, self.target.id, kills,
         )
@@ -94,7 +94,7 @@ def select_target_from(self, other_groups):
         for other_group in other_groups:
             damage = self.damage(other_group)
             if damage > 0:
-                log.debug(
+                log.info(
                     "%s group %d would deal defending group %d %d damage",
                     self.system.name, self.id, other_group.id, damage,
                 )
@@ -126,22 +126,22 @@ def attacking_phase(self):
 
     def tick(self):
         for system in self.immune, self.infection:
-            log.debug(system.name + ":")
+            log.info(system.name + ":")
             if system.alive:
                 for group in system.groups:
                     if group.alive:
-                        log.debug("Group %d contains %d units", group.id, group.n)
+                        log.info("Group %d contains %d units", group.id, group.n)
             else:
-                log.debug("No groups remain.")
-        log.debug("")
+                log.info("No groups remain.")
+        log.info("")
         self.target_selection_phase()
-        log.debug("")
+        log.info("")
         n_before = self.immune.n + self.infection.n
         self.attacking_phase()
         n_after = self.immune.n + self.infection.n
         if n_after == n_before:
             raise StaleMate
-        log.debug("")
+        log.info("")
         self.ticks += 1
 
     def run_until_complete(self):

aoc_wim/cli.py

@@ -0,0 +1,54 @@
+"""Wim's solutions for https://adventofcode.com/"""
+import logging
+import os
+import runpy
+import sys
+from argparse import ArgumentParser
+from datetime import datetime
+from shutil import copy
+
+from aocd.get import most_recent_year
+from aocd.models import Puzzle
+from aocd.utils import AOC_TZ
+
+
+def run_one():
+    parser = ArgumentParser(description=__doc__)
+    aoc_now = datetime.now(tz=AOC_TZ)
+    parser.add_argument(
+        "day",
+        nargs="?",
+        type=int,
+        choices=range(1, 26),
+        default=min(aoc_now.day, 25) if aoc_now.month == 12 else 1,
+        help="1-25 (default: %(default)s)",
+    )
+    parser.add_argument(
+        "year",
+        nargs="?",
+        type=int,
+        choices=range(2015, aoc_now.year + int(aoc_now.month == 12)),
+        default=most_recent_year(),
+        help=">= 2015 (default: %(default)s)",
+    )
+    parser.add_argument("-d", "--data")
+    log_levels = "DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"
+    parser.add_argument("--log-level", choices=log_levels)
+    args = parser.parse_args()
+    if args.log_level:
+        level_int = getattr(logging, args.log_level)
+        logging.basicConfig(format="%(message)s", level=level_int)
+    mod_name = f"aoc_wim.aoc{args.year}.q{args.day:02d}"
+    sys.modules.pop(mod_name, None)
+    if args.data is not None:
+        os.environ["AOC_SESSION"] = "aocw"
+        path = os.path.expanduser("~/.config/aocd/aocw/")
+        os.makedirs(path, exist_ok=True)
+        path += f"{args.year}_{args.day:02d}_input.txt"
+        if os.path.isfile(args.data):
+            copy(args.data, path)
+        else:
+            with open(path, "w") as f:
+                print(args.data, file=f)
+    print(f"--- {args.year} Day {args.day}: {Puzzle(args.year, args.day).title} ---")
+    runpy.run_module(mod_name, run_name="__main__")

setup.py

@@ -29,5 +29,8 @@
         "wimpy",
     ],
     packages=find_packages(),
-    entry_points={"adventofcode.user": ["wim = aoc_wim:plugin"]},
+    entry_points={
+        "adventofcode.user": ["wim = aoc_wim:plugin"],
+        "console_scripts": ["aocw = aoc_wim.cli:run_one"],
+    },
 )