Clues can now be entered for words

[getaclue.git] / crosswordpuzzlecreator.py

diff --git a/crosswordpuzzlecreator.py b/crosswordpuzzlecreator.py
index 4098cd3..ed7e1d8 100644 (file)
--- a/crosswordpuzzlecreator.py
+++ b/crosswordpuzzlecreator.py
@@ -77,6 +77,30 @@ class CrosswordPuzzleCreator:
                                sys.stdout.write (' ' + self.BLUE + "%2d" % row + self.ENDCOL + "\n")
                        raw_input (self.BRICKRED + "Press <return> to continue" + self.ENDCOL)
 
+       # set a clue to a word
+       def 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:
+                       aword, arow, acol, alen = self.puzzle.get_word_across (row, col)
+                       sys.stdout.write (self.BLUE + "Word at position: " + aword + "\n" + self.ENDCOL)
+                       clue = raw_input (self.BRICKRED + "Clue for word: " + self.ENDCOL)
+                       if clue:
+                               self.puzzle.data[arow][acol].clue_across = clue
+                               sys.stdout.write (self.BLUE + "Set clue: \n" + self.puzzle.data[arow][acol].clue_across)
+               except crosswordpuzzle.NoWordException:
+                       sys.stderr.write ("No across word found at that position")
+
        # add a word to the puzzle
        def add_word (self, across=True):
                # first display the grid
@@ -122,8 +146,7 @@ class CrosswordPuzzleCreator:
                                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 ("S. Save puzzle\n")
                                sys.stdout.write ("X. Exit to main menu\n" + self.ENDCOL)
                                ch = raw_input (self.BRICKRED + "Your choice: " + self.ENDCOL)
@@ -137,6 +160,8 @@ class CrosswordPuzzleCreator:
                                        self.puzzle.freeze_grid ()
                                elif ch == "7":
                                        self.puzzle.unfreeze_grid ()
+                               elif ch == "8":
+                                       self.set_clue ()
                                elif ch == "X" or ch == "x":
                                        break