eryn/sudoku
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Allow specifying path to puzzles library

Browse source 
By default look in the current directory for a puzzles/ directory.
This commit is contained in:
Eryn Wells 2017-10-11 16:03:45 -07:00
parent ed41930092
commit 7809c38104
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
sudoku/puzzles.py
View file

@ -5,22 +5,22 @@ Parser for puzzles in the ./puzzles directory.
''' '''
import argparse import argparse
import sudoku import os.path
import sys import sys
euler = [] euler = []
norvig = [] norvig = []
def parse_puzzle_files(quiet=True): def parse_puzzle_files(path, quiet=True):
global euler, norvig global euler, norvig
if not quiet: if not quiet:
print('Parsing Euler puzzles') print('Parsing Euler puzzles')
euler.extend(_get_puzzles('puzzles/euler.txt', quiet)) euler.extend(_get_puzzles(os.path.join(path, 'euler.txt'), quiet))
if not quiet: if not quiet:
print('Parsing Norvig puzzles') print('Parsing Norvig puzzles')
norvig.extend(_get_puzzles('puzzles/norvig.txt', quiet)) norvig.extend(_get_puzzles(os.path.join(path, 'norvig.txt'), quiet))
def _get_puzzles(filename, quiet): def _get_puzzles(filename, quiet):
with open(filename, 'r') as f: with open(filename, 'r') as f:
@ -33,7 +33,7 @@ def _parse_puzzle(puzzle, quiet):
if not quiet: if not quiet:
print("Parsing '{}'".format(puzzle)) print("Parsing '{}'".format(puzzle))
board = (int('0' if x == '.' else x) for x in puzzle) board = (int('0' if x == '.' else x) for x in puzzle)
return sudoku.Sudoku(board=board) return Sudoku(board=board)
else: else:
if not quiet: if not quiet:
print("Skipping '{}'".format(puzzle)) print("Skipping '{}'".format(puzzle))
@ -44,12 +44,13 @@ def parse_args(args):
parser.add_argument('--euler', '-e', dest='library', action='store_const', const=euler, default=None) 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('--norvig', '-n', dest='library', action='store_const', const=norvig, default=None)
parser.add_argument('--verbose', '-v', action='store_true', default=False) 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) parser.add_argument('indexes', metavar='N', nargs='+', type=int)
return parser.parse_args(args) return parser.parse_args(args)
def main(): def main():
args = parse_args(sys.argv[1:]) args = parse_args(sys.argv[1:])
parse_puzzle_files(quiet=not args.verbose) parse_puzzle_files(args.path, quiet=not args.verbose)
for i in args.indexes: for i in args.indexes:
print(args.library[i]) print(args.library[i])
return 0 return 0