sudoku/puzzles.py

#!env python3
# Eryn Wells <eryn@erynwells.me>
'''
Parser for puzzles in the ./puzzles directory.
'''

import argparse
import os.path
import sys

euler = []
norvig = []

def parse_puzzle_files(path, quiet=True):
    global euler, norvig

    if not quiet:
        print('Parsing Euler puzzles')
    euler.extend(_get_puzzles(os.path.join(path, 'euler.txt'), quiet))

    if not quiet:
        print('Parsing Norvig puzzles')
    norvig.extend(_get_puzzles(os.path.join(path, 'norvig.txt'), quiet))

def _get_puzzles(filename, quiet):
    with open(filename, 'r') as f:
        puzzles = f.readlines()
    return (_parse_puzzle(p, quiet) for p in puzzles if p)

def _parse_puzzle(puzzle, quiet):
    puzzle = puzzle.strip()
    if len(puzzle) == 81:
        if not quiet:
            print("Parsing  '{}'".format(puzzle))
        board = (int('0' if x == '.' else x) for x in puzzle)
        return Sudoku(board=board)
    else:
        if not quiet:
            print("Skipping '{}'".format(puzzle))
        return None

def parse_args(args):
    parser = argparse.ArgumentParser()
    parser.add_argument('--euler', '-e', dest='library', action='store_const', const=euler, default=None)
    parser.add_argument('--norvig', '-n', dest='library', action='store_const', const=norvig, default=None)
    parser.add_argument('--verbose', '-v', action='store_true', default=False)
    parser.add_argument('--library', '-l', dest='path', default='./puzzles')
    parser.add_argument('indexes', metavar='N', nargs='+', type=int)
    return parser.parse_args(args)

def main():
    args = parse_args(sys.argv[1:])
    parse_puzzle_files(args.path, quiet=not args.verbose)
    for i in args.indexes:
        print(args.library[i])
    return 0

if __name__ == '__main__':
    sys.exit(main())