add 2023/day1-2

2023/01/main.py

@@ -0,0 +1,58 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8; py-indent-offset: 4 -*-
+
+# Author: Sebastian Mark
+# CC-BY-SA (https://creativecommons.org/licenses/by-sa/4.0/deed.de)
+
+# pylint: disable=missing-module-docstring,missing-function-docstring,consider-using-f-string
+
+import re
+
+
+def readinput():
+    with open("input", "r", encoding="utf-8") as file:
+        lines = file.readlines()
+    return lines
+
+
+def get_number(line: str) -> int:
+    regex = r"\d"
+    matches = re.findall(regex, line)
+    new_number = matches[0] + matches[-1]
+    return int(new_number)
+
+
+def get_spelled_number(line: str) -> int:
+    number_map = {
+        "one": "o1e",
+        "two": "t2o",
+        "three": "t3e",
+        "four": "f4r",
+        "five": "f5e",
+        "six": "s6x",
+        "seven": "s7n",
+        "eight": "e8t",
+        "nine": "n9e",
+    }
+
+    for k, v in number_map.items():
+        if k in line:
+            line = line.replace(k, v)
+
+    return get_number(line)
+
+
+def main():
+    lines = readinput()
+
+    # part1
+    count = sum(get_number(line) for line in lines)
+    print("Sum: %d" % count)
+
+    # part1
+    count = sum(get_spelled_number(line) for line in lines)
+    print("Sum (literals): %d" % count)
+
+
+if __name__ == "__main__":
+    main()