Added the "reveal word" functionality

[getaclue.git] / crosswordpuzzle.py

diff --git a/crosswordpuzzle.py b/crosswordpuzzle.py
index e4a7f58..319a4fc 100644 (file)
--- a/crosswordpuzzle.py
+++ b/crosswordpuzzle.py
@@ -582,13 +582,13 @@ class CrosswordPuzzle:
        # remove an across word at position
        def remove_word_across (self, row, col):
                # if grid is frozen don't allow removal of word
-               assert_unfrozen_grid ()
+               self.assert_unfrozen_grid ()
 
                word, brow, bcol, l = self.get_word_across (row, col)
 
                # traverse from the beginning to end of the word and erase it
                c = bcol
-               while True:
+               while c < self.cols:
                        if self.data[brow][c].occupied_across is True:
                                self.data[brow][c].clear_across_data ()
                        else:
@@ -603,12 +603,38 @@ class CrosswordPuzzle:
                word, brow, bcol, l = self.get_word_down (row, col)
                # traverse from the beginn to end of the word and erase it
                r = brow
-               while True:
+               while r < self.rows:
                        if self.data[r][bcol].occupied_down is True:
                                self.data[r][bcol].clear_down_data ()
                        else:
                                break
                        r += 1
 
+       # reveal/unreveal a word at position
+       def reveal_word_across (self, row, col, revealed=True):
+               # set the revealed flag for the word at the position
+               word= self.get_word_across (row, col)
+
+               c = word[2]
+               while c < self.cols:
+                       if self.data[word[1]][c].occupied_across is True:
+                               self.data[word[1]][c].revealed = revealed
+                       else:
+                               break
+                       c += 1
+
+       # reveal/unreveal a word at position
+       def reveal_word_down (self, row, col, revealed=True):
+               # set the revealed flag for the word at the position
+               word = self.get_word_down (row, col)
+
+               r = word[1]
+               while r < self.rows:
+                       if self.data[r][word[2]].occupied_down is True:
+                               self.data[r][word[2]].revealed = revealed
+                       else:
+                               break
+                       r += 1
+