generate.py

import argparse
import time

from crossword import dictionary, grid, optimize

parser = argparse.ArgumentParser(description="Fill in missing crossword squares,")
parser.add_argument("filename", help="Path to the uncompleted crossword file")
filename = parser.parse_args().filename

dictionary = dictionary.Dictionary(seed=0)
grid = grid.load_grid(filename)
start = time.time()

grid, total_score = optimize.optimize(grid, dictionary, optimize_sections=True)

print("")
grid.print()
# for s, m in grid.get_all_words():
#     word = grid.get_word(s, m)
#     score = int(dictionary.search(word)[0][1])
#     print(f"{s} {m} {word}: {score}")
seconds = time.time() - start
nodes = 0  # generator.nodes_searched

print("")
print(f"Total score: {total_score}")
print(f"Average word score: {total_score/len(grid.get_all_words()):.2f}")
print(f"Generation took {seconds:.2f} seconds")
print(f"Searched {nodes} nodes, {nodes / seconds:.2f} nodes/sec")