#!/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()