# Licensed under the GNU GPL v3
# Cross puzzle creator class
+
import sys
+import cPickle
+import readline
import crosswordpuzzle
# Black background white text
BLACK_BG = '\033[40;1;37m'
# white background black text
- WHITE_BG = '\033[47;0;30m'
+ WHITE_BG = '\033[47;1;30m'
# white background red text
- REDWHITE_BG = '\033[47;0;31m'
+ REDWHITE_BG = '\033[47;1;31m'
# disable colors
ENDCOL = '\033[0m'
self.current_file = None
self.puzzle = None
+ # save the puzzle
+ def save_puzzle (self):
+ if self.current_file:
+ fpuzzle = open (self.current_file, "wb")
+ cPickle.dump (self.puzzle, fpuzzle, cPickle.HIGHEST_PROTOCOL)
+ sys.stdout.write (self.BLUE + "Puzzle saved to: " +
+ self.current_file + "\n" + self.ENDCOL)
+
+ # load the puzzle
+ def load_puzzle (self):
+ if self.current_file:
+ fpuzzle = open (self.current_file, "rb")
+ self.puzzle = cPickle.load (fpuzzle)
+
# display the grid with words
def print_puzzle (self, no_words=False):
# if self.puzzle is not none
sys.stdout.write (' ' + self.BLUE + "%2d" % row + self.ENDCOL + "\n")
raw_input (self.BRICKRED + "Press <return> to continue" + self.ENDCOL)
+ # display existing clues
+ def on_display_clues (self):
+ try:
+ aclues = self.puzzle.get_clues_across ()
+ sys.stdout.write (self.BOLD + "\n------------\n")
+ sys.stdout.write ("Across words\n")
+ sys.stdout.write ("------------\n" + self.ENDCOL)
+
+ for word, clue in aclues:
+ sys.stdout.write (self.BOLD + word[0] + ": " + self.ENDCOL)
+ if clue:
+ sys.stdout.write (self.BLUE + clue + "\n" + self.ENDCOL)
+ else:
+ sys.stdout.write (self.BLUE + "(No clue yet)\n" + self.ENDCOL)
+ except crosswordpuzzle.NoWordsException:
+ sys.stderr.write ("No words across\n")
+
+ try:
+ dclues = self.puzzle.get_clues_down ()
+ sys.stdout.write (self.BOLD + "\n----------\n")
+ sys.stdout.write ("Down words\n")
+ sys.stdout.write ("----------\n" + self.ENDCOL)
+
+ for word, clue in dclues:
+ sys.stdout.write (self.BOLD + word[0] + ": " + self.ENDCOL)
+ if clue:
+ sys.stdout.write (self.BLUE + clue + "\n" + self.ENDCOL)
+ else:
+ sys.stdout.write (self.BLUE + "(No clue yet)\n" + self.ENDCOL)
+ except crosswordpuzzle.NoWordsException:
+ sys.stderr.write ("No words down\n")
+
+ raw_input (self.BRICKRED + "Press <return> to continue" + self.ENDCOL)
+
+ # set a clue to a word
+ def on_set_clue (self):
+ self.print_puzzle ()
+ # get the row and column
+ srow = raw_input (self.BRICKRED + "At row: " + self.ENDCOL)
+ scol = raw_input (self.BRICKRED + "At col: " + self.ENDCOL)
+ # try converting it to number
+ try:
+ row = int (srow)
+ col = int (scol)
+ except ValueError:
+ sys.stderr.write ("Invalid row or column\n")
+ return
+
+ try:
+ # across word set the clue if found
+ aword, arow, acol, alen = self.puzzle.get_word_across (row, col)
+ sys.stdout.write (self.BLUE + "Across word at position: " + aword + "\n" + self.ENDCOL)
+ clue = raw_input (self.BRICKRED + "Clue for across word: " + self.ENDCOL)
+ self.puzzle.data[arow][acol].clue_across = clue
+ sys.stdout.write (self.BLUE + "Set the clue: \n" + self.puzzle.data[arow][acol].clue_across)
+ except crosswordpuzzle.NoWordException:
+ sys.stderr.write ("No across word found at that position\n")
+
+ try:
+ # down word set the clue if found
+ dword, drow, dcol, dlen = self.puzzle.get_word_down (row, col)
+ sys.stdout.write (self.BLUE + "Down word at position: " + dword + "\n" + self.ENDCOL)
+ clue = raw_input (self.BRICKRED + "Clue for down word: " + self.ENDCOL)
+ self.puzzle.data[drow][dcol].clue_down = clue
+ sys.stdout.write (self.BLUE + "Set the clue: \n" + self.puzzle.data[drow][dcol].clue_down)
+ except crosswordpuzzle.NoWordException:
+ sys.stderr.write ("No down word found at that position\n")
+
+ # remove a down word
+ def on_remove_down (self):
+ self.print_puzzle ()
+
+ srow = raw_input (self.BRICKRED + "At row: " + self.ENDCOL)
+ scol = raw_input (self.BRICKRED + "At col: " + self.ENDCOL)
+ try:
+ row = int (srow)
+ col = int (scol)
+ except ValueError:
+ sys.stderr.write ("Invalid row or column\n")
+ return
+
+ try:
+ self.puzzle.remove_word_down (row, col)
+ sys.stdout.write (self.BLUE + "Down word removed\n" + self.ENDCOL)
+ except crosswordpuzzle.FrozenGridException:
+ sys.stderr.write ("Word cannot be removed from a frozen puzzle\n")
+ except crosswordpuzzle.NoWordException:
+ sys.stderr.write ("No down word found at that position\n")
+
+ # remove an across word
+ def on_remove_across (self):
+ self.print_puzzle ()
+
+ srow = raw_input (self.BRICKRED + "At row: " + self.ENDCOL)
+ scol = raw_input (self.BRICKRED + "At col: " + self.ENDCOL)
+ try:
+ row = int (srow)
+ col = int (scol)
+ except ValueError:
+ sys.stderr.write ("Invalid row or column\n")
+ return
+
+ try:
+ self.puzzle.remove_word_across (row, col)
+ sys.stdout.write (self.BLUE + "Across word removed\n" + self.ENDCOL)
+ except crosswordpuzzle.FrozenGridException:
+ sys.stderr.write ("Word cannot be removed from a frozen puzzle\n")
+ except crosswordpuzzle.NoWordException:
+ sys.stderr.write ("No across word found at that position\n")
+
# add a word to the puzzle
- def add_word (self, across=True):
+ def on_add_word (self, across=True):
# first display the grid
self.print_puzzle ()
# get the row and column
except crosswordpuzzle.FrozenGridException:
sys.stderr.write ("Word cannot be added to a frozen puzzle.\n")
+ # Export to image/HTML
+ def on_export_image (self, solution=True):
+ try:
+ sys.stdout.write (self.BLUE + "Exporting puzzle to image/HTML\n")
+ pngfile = raw_input (self.BRICKRED + "Filename (PNG): " + self.ENDCOL)
+ if solution is False:
+ htmlfile = raw_input (self.BRICKRED + "Filename (HTML): " +
+ self.ENDCOL)
+ puztitle = raw_input (self.BRICKRED + "Title of puzzle: " +
+ self.ENDCOL)
+ self.puzzle.export_image (pngfile, htmlfile, puztitle, solution)
+ else:
+ self.puzzle.export_image (pngfile)
+
+ sys.stdout.write (self.BLUE + "Successfully exported!")
+ except crosswordpuzzle.FrozenGridException:
+ sys.stderr.write ("Cannot export as grid is not frozen/finalized\n")
+ except crosswordpuzzle.NoWordsException:
+ sys.stderr.write ("No words to export!\n")
+
+ # Export to across lite
+ def on_export_acrosslite (self):
+ try:
+ sys.stdout.write (self.BLUE + "Exporting to AcrossLite(tm) Format\n" +
+ self.ENDCOL)
+ title = raw_input (self.BRICKRED + "Puzzle title: " + self.ENDCOL)
+ name = raw_input (self.BRICKRED + "Author name: " + self.ENDCOL)
+ copyright = raw_input (self.BRICKRED + "Copyright: " + self.ENDCOL)
+ exportfile = raw_input (self.BRICKRED + "Export to file: " + self.ENDCOL)
+
+ acrosslite_str = self.puzzle.export_acrosslite (title, name, copyright)
+ fexport = open (exportfile, "w")
+ fexport.write (acrosslite_str)
+ fexport.close ()
+ sys.stdout.write (self.BLUE + "Exported AcrossLite(tm) File: " +
+ exportfile + "\n" + self.ENDCOL)
+ except crosswordpuzzle.FrozenGridException:
+ sys.stderr.write ("Cannot export as grid is not frozen/finalized\n")
+ except crosswordpuzzle.NoWordsException:
+ sys.stderr.write ("No words to export!\n")
+
# Puzzle loop
def do_puzzle_loop (self):
# there is a current file
- while True:
- if self.current_file and self.puzzle:
+ if self.current_file and self.puzzle:
+ while True:
sys.stdout.write (self.BOLD + "\n-----------------------------------\n")
sys.stdout.write ("Puzzle: " + self.current_file + "\n")
sys.stdout.write ("-----------------------------------" + self.ENDCOL + "\n")
sys.stdout.write ("5. Remove down word\n")
sys.stdout.write ("6. Freeze grid\n")
sys.stdout.write ("7. Unfreeze grid\n")
- sys.stdout.write ("8. Set a clue\n")
- sys.stdout.write ("9. Remove clue\n")
+ sys.stdout.write ("8. Set clue for word\n")
+ sys.stdout.write ("9. Display clues\n")
+ sys.stdout.write ("R. Reset grid\n")
sys.stdout.write ("S. Save puzzle\n")
+ sys.stdout.write ("E. Export to AcrossLite(TM) format\n")
+ sys.stdout.write ("H. Export puzzle as image/HTML\n")
+ sys.stdout.write ("I. Export solution as image\n")
sys.stdout.write ("X. Exit to main menu\n" + self.ENDCOL)
ch = raw_input (self.BRICKRED + "Your choice: " + self.ENDCOL)
if ch == "1":
self.print_puzzle ()
elif ch == "2":
- self.add_word ()
+ self.on_add_word ()
elif ch == "3":
- self.add_word (False)
+ self.on_add_word (False)
+ elif ch == "4":
+ self.on_remove_across ()
+ elif ch == "5":
+ self.on_remove_down ()
elif ch == "6":
self.puzzle.freeze_grid ()
elif ch == "7":
self.puzzle.unfreeze_grid ()
+ elif ch == "8":
+ self.on_set_clue ()
+ elif ch == "9":
+ self.on_display_clues ()
+ elif ch == "R" or ch == "r":
+ self.on_reset_grid ()
+ elif ch == "S" or ch == "s":
+ self.save_puzzle ()
+ elif ch == "E" or ch == "e":
+ self.on_export_acrosslite ()
+ elif ch == "H" or ch == "h":
+ self.on_export_image (False)
+ elif ch == "I" or ch == "i":
+ self.on_export_image ()
elif ch == "X" or ch == "x":
break
self.puzzle = crosswordpuzzle.CrosswordPuzzle (rows, cols)
self.do_puzzle_loop ()
+ # when user chooses to load puzzle
+ def on_load_puzzle (self):
+ self.current_file = raw_input (self.BRICKRED + "Puzzle to load: "
+ + self.ENDCOL)
+ self.load_puzzle ()
+ self.do_puzzle_loop ()
+
+ # when user chooses to reset grid
+ def on_reset_grid (self):
+ ans = raw_input (self.BRICKRED +
+ "This will clear the entire grid! Are you sure (Y/N)? " + self.ENDCOL)
+ if ans == "y" or ans == "Y":
+ self.puzzle.reset_grid ()
+ sys.stdout.write (self.BLUE + "Grid has been cleared of all data!"
+ + self.ENDCOL + "\n")
+
# Main application loop
def do_main_loop (self):
# display the menu
ch = raw_input (self.BRICKRED + "Your choice: " + self.ENDCOL)
if ch == '1':
self.on_new_puzzle ()
+ if ch == '2':
+ self.on_load_puzzle ()
if ch == 'x' or ch == 'X':
break
projects / getaclue.git / blobdiff