day4.py

# vi: set shiftwidth=4 tabstop=4 expandtab:
import datetime
import os
import itertools
import hashlib


top_dir = os.path.dirname(os.path.abspath(__file__)) + "/../../"


def get_key_from_file(file_path=top_dir + "resources/year2015_day4_input.txt"):
    with open(file_path) as f:
        for l in f:
            return l.strip()


def get_coin(key, nb_zeros=5):
    zeros = "0" * nb_zeros
    for i in itertools.count():
        s = key + str(i)
        h = hashlib.md5(s.encode("utf-8")).hexdigest()
        if h.startswith(zeros):
            return i


def run_tests():
    assert get_coin("abcdef") == 609043
    assert get_coin("pqrstuv") == 1048970


def get_solutions():
    key = get_key_from_file()
    print(get_coin(key) == 282749)
    print(get_coin(key, 6) == 9962624)


if __name__ == "__main__":
    begin = datetime.datetime.now()
    run_tests()
    get_solutions()
    end = datetime.datetime.now()
    print(end - begin)