X-Git-Url: https://harishankar.org/repos/?p=getaclue.git;a=blobdiff_plain;f=crosswordpuzzle.py;h=694a7660b19884e377ed5ff0e8435a1ba9c272b0;hp=9617897fddada51883187489cdb2153e268ab6f6;hb=f9926c428d81940df4d22a6caeffa9865057b12c;hpb=8e802b4a6d2d5af14810a2b6603fd53b2697c039 diff --git a/crosswordpuzzle.py b/crosswordpuzzle.py index 9617897..694a766 100644 --- a/crosswordpuzzle.py +++ b/crosswordpuzzle.py @@ -50,6 +50,11 @@ class NoWordException (Exception): def __init__ (self, row, col): self.pos = (row, col) +# exception when no words are present in the grid +class NoWordsException (Exception): + def __init__ (self): + self.msg = "No words in grid" + class CrosswordPuzzle: def __init__ (self, rows, cols): # define number of rows and columns @@ -68,6 +73,84 @@ class CrosswordPuzzle: for j in range (cols): self.data[i].append (GridItem ()) + # get the AcrossLite(TM) data for exporting + def export_acrosslite (self, title, author, copyright): + # don't export if grid is frozen + if self.frozen_grid is False: + raise FrozenGridException + + across_data = [] + across_data.append ("\r\n") + across_data.append ("\r\n") + across_data.append (title + "\r\n") + across_data.append ("<AUTHOR>\r\n") + across_data.append (author + "\r\n") + across_data.append ("<COPYRIGHT>\r\n") + across_data.append (copyright + "\r\n") + across_data.append ("<SIZE>\r\n") + str_size = str (self.cols) + "x" + str (self.rows) + across_data.append (str_size + "\r\n") + across_data.append ("<GRID>\r\n") + for row in range (self.rows): + for col in range (self.cols): + if (self.data[row][col].occupied_across is True or + self.data[row][col].occupied_down is True): + across_data.append (self.data[row][col].char) + else: + across_data.append (".") + across_data.append ("\r\n") + + across_data.append ("<ACROSS>\r\n") + clues_across = self.get_clues_across () + for word, clue in clues_across: + if clue: + across_data.append (clue + "\r\n") + else: + across_data.append ("(No clue yet)\r\n") + + across_data.append ("<DOWN>\r\n") + clues_down = self.get_clues_down () + for word, clue in clues_down: + if clue: + across_data.append (clue + "\r\n") + else: + across_data.append ("(No clue yet\r\n") + + acrosslite_str = "".join (across_data) + return acrosslite_str + + # get all the clues for across + def get_clues_across (self): + clues = [] + # traverse the grid + for row in range (self.rows): + for col in range (self.cols): + if (self.data[row][col].occupied_across is True and + self.data[row][col].across_start is True): + word_across = self.get_word_across (row, col) + clues.append ((word_across, self.data[row][col].clue_across)) + # if no across words are found at all + if not clues: + raise NoWordsException + + return clues + + # get all the clues for down + def get_clues_down (self): + clues = [] + # traverse the grid + for row in range (self.rows): + for col in range (self.cols): + if (self.data[row][col].occupied_down is True and + self.data[row][col].down_start is True): + word_down = self.get_word_down (row, col) + clues.append ((word_down, self.data[row][col].clue_down)) + # if no down words are found at all + if not clues: + raise NoWordsException + + return clues + # getting a down word at a position def get_word_down (self, row, col): # if index is out of bounds @@ -136,7 +219,7 @@ class CrosswordPuzzle: word = "".join (word_chars) # return the word, starting column, row and length as a tuple - return (word, start_col, row, len(word)) + return (word, row, start_col, len(word)) # setting a down word def set_word_down (self, row, col, word):