operator sucks

aoc_wim/aoc2015/q16.py

@@ -2,11 +2,6 @@
 --- Day 16: Aunt Sue ---
 https://adventofcode.com/2015/day/16
 """
-from collections import defaultdict
-from operator import eq
-from operator import gt
-from operator import lt
-
 from aocd import data
 
 
@@ -34,13 +29,19 @@
 
 
 def find_sue(**kwargs):
-    opmap = defaultdict(lambda: eq)
-    opmap.update(kwargs)
     for sue, stats in sues.items():
         common_keys = d_message.keys() & stats.keys()
-        if all(opmap[k](stats[k], d_message[k]) for k in common_keys):
+        if all(kwargs.get(k, int.__eq__)(stats[k], d_message[k]) for k in common_keys):
             return int(sue.split()[1])
 
 
-print("answer_a:", find_sue())
-print("answer_b:", find_sue(cats=gt, trees=gt, pomeranians=lt, goldfish=lt))
+a = find_sue()
+b = find_sue(
+    cats=int.__gt__,
+    trees=int.__gt__,
+    pomeranians=int.__lt__,
+    goldfish=int.__lt__,
+)
+
+print("answer_a:", a)
+print("answer_b:", b)

aoc_wim/aoc2022/q21.py

@@ -2,15 +2,14 @@
 --- Day 21: Monkey Math ---
 https://adventofcode.com/2022/day/21
 """
-import operator as op
-
 from aocd import data
 
+
 ops = {
-    "+": op.add,
-    "*": op.mul,
-    "-": op.sub,
-    "/": op.truediv,
+    "+": complex.__add__,
+    "*": complex.__mul__,
+    "-": complex.__sub__,
+    "/": complex.__truediv__,
 }
 
 

aoc_wim/aoc2023/q22.py

@@ -2,8 +2,8 @@
 --- Day 22: Sand Slabs ---
 https://adventofcode.com/2023/day/22
 """
-import operator as op
 from dataclasses import dataclass
+from operator import attrgetter
 
 from aocd import data
 
@@ -57,7 +57,7 @@ def fall(self):
 # data = tdata
 
 bricks = [Brick.fromline(x) for x in data.splitlines()]
-bricks.sort(key=op.attrgetter("z0"))
+bricks.sort(key=attrgetter("z0"))
 space = {}
 for i, brick in enumerate(bricks):
     brick.id = i

aoc_wim/aoc2024/q07.py

@@ -3,7 +3,6 @@
 https://adventofcode.com/2024/day/7
 """
 import itertools as it
-import operator as op
 
 from aocd import data
 
@@ -15,7 +14,7 @@ def concat(n1, n2):
 a = b = 0
 for line in data.replace(":", "").splitlines():
     result, *ns = map(int, line.split())
-    for ops in it.product((op.add, op.mul), repeat=len(ns)-1):
+    for ops in it.product((int.__add__, int.__mul__), repeat=len(ns)-1):
         val = ns[0]
         for i, f in enumerate(ops, 1):
             val = f(val, ns[i])
@@ -24,7 +23,7 @@ def concat(n1, n2):
             b += result
             break
     else:
-        for ops in it.product((op.add, op.mul, concat), repeat=len(ns) - 1):
+        for ops in it.product((int.__add__, int.__mul__, concat), repeat=len(ns) - 1):
             val = ns[0]
             for i, f in enumerate(ops, 1):
                 val = f(val, ns[i])