X-Git-Url: https://harishankar.org/repos/?p=getaclue.git;a=blobdiff_plain;f=crosswordpuzzlecreator.py;h=ed7e1d86268b20e26c7666ab127006b566fbb534;hp=42d4fddd0d476b52f1929eaff7e9424fa652c965;hb=8e802b4a6d2d5af14810a2b6603fd53b2697c039;hpb=cd153df772f936f9aaed0f75d83f15c05a6ee9b6

diff --git a/crosswordpuzzlecreator.py b/crosswordpuzzlecreator.py
index 42d4fdd..ed7e1d8 100644
--- a/crosswordpuzzlecreator.py
+++ b/crosswordpuzzlecreator.py
@@ -19,9 +19,9 @@ class CrosswordPuzzleCreator:
 	# 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'
 
@@ -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