mvaal/advent-of-code-helpers

import os
import pathlib


def read_input_from_file(file_path) -> str:
    r""" Reads data from file path as a string.

    :param file_path: file path
    :type file_path: str
    :return: data
    :rtype: str
    """
    with open(file_path) as file:
        return file.read()


def input_lines(puzzle_input: str):
    return puzzle_input.split("\n")


def output(result: str,
           part: int,
           day: int,
           year: int,
           output_dir: str = None,
           file_prefix: str = None):
    print(result)
    if output_dir:
        output_file_dir = os.path.join(output_dir, str(year), str(day))
        pathlib.Path(output_file_dir).mkdir(parents=True, exist_ok=True)
        file_name = '{}-{}.txt'.format(file_prefix, str(
            part)) if file_prefix else '{}.txt'.format(str(part))
        output_file_path = os.path.join(output_file_dir, file_name)
        with open(output_file_path, "a+") as output_file:
            output_file.write('{}\n'.format(result))