Advent of Code helpers

Advent of Code helper functions

from aoc.helpers import output, read_input_from_file, input_lines
from aoc import template

Setup Guide

Install with pip

pip install advent-of-code-helpers

Helper Usage

read_input_from_file reads the data as a single line from a file

from aoc.helpers import read_input_from_file
read_input_from_file('path/to/input_data')

input_lines returns a list of strings from the string input

from aoc.helpers import input_lines
input_lines('single\nstring\ninput')

output prints the result to console and writes to an output file if an output directory is provided

from aoc.helpers import output
output('result', part(int), day(int), year(int), output_dir(str), file_prefix(str))

Template Usage

You can specify data from a file using the data(input) function.

You can specify an output directory for output using the output(output) function. If left empty, it will still print to screen, but will not write the result to a file. If given an output directory, the results will be appended to the file so you can easily go back and look at previous results.

Examples

from aoc import template


class Part1(template.Part1):
    def solve(self):
        # Read input
        lines = input_lines(self.input())
        # Do some work here

        # Sample output
        result = ','.join(lines)
        return result


def main():
    output_dir = '../out'
    test_data = os.path.join(os.path.dirname(__file__),
                             'resources/test_input.txt')
    Part1(1, 2018).data(test_data).output(output_dir)

    data = os.path.join(os.path.dirname(__file__), 'resources/input.txt')
    Part1(1, 2018).data(data).output(output_dir)


if __name__ == "__main__":
    main()

More usage in the example.

Template Usage with Other Libraries

If you want to use your own input reader or a library like advent-of-code-data, you can override the input method.

Examples

from aoc import template
from aoc.helpers import input_lines
from aocd import get_data


class Part1(template.Part1):
    def input(self):
        if self.input_file:
            return super().input()
        else:
            return get_data(day=self.day, year=self.year)

    def solve(self):
        # Read input
        lines = input_lines(self.input())
        # Do some work here

        # Sample output
        result = ','.join(lines)
        return result


def main():
    Part1(1, 2018).output('../out')


if __name__ == "__main__":
    main()

mvaal / advent-of-code-helpers Goto Github PK