day1.py

import time
intstrs = {'one': '1', 'two': '2', 'three': '3', 'four': '4', 'five': '5', 'six': '6', 'seven': '7', 'eight': '8', 'nine': '9'}

def parse_input(filename: str) -> list[str]:
    return open(filename, "r").read().splitlines()

def part_one(strings: list[str]) -> None:
    nums = []
    for string in strings:
        num = ''
        for char in string:
            if char.isdigit():
                num += char
                break
        for char in string[::-1]:
            if char.isdigit():
                num += char
                break
        nums.append(int(num))
        
    print(sum(nums))
    
def string_contains_digit(string: str) -> str:
    if string[-1].isdigit():
        return string[-1]
    for intstr, num in intstrs.items():
        if intstr in string:
            return num
    return '-1'

def part_two(strings: list[str]) -> None:
    nums: list[int] = []
    for string in strings:
        num = ''
        build = ''
        for char in string:
            build += char
            append = string_contains_digit(build)
            if append == '-1': continue
            num += append
            break
        build = ''
        for char in string[::-1]:
            if char.isdigit():
                num += char
                break
            build += char
            append = string_contains_digit(build[::-1])
            if append == '-1': continue
            num += append
            break
        nums.append(int(num))
    print(sum(nums))

def main(input_filename: str) -> None:
    inp = parse_input(input_filename)
    start_part_one = time.time()
    part_one(inp)
    start_part_two = time.time()
    part_two(inp)
    end_time = time.time()
    print(f"Part one took {start_part_two - start_part_one} seconds")
    print(f"Part two took {end_time - start_part_two} seconds")

if __name__ == "__main__":
    main("input.txt")