X-Git-Url: https://harishankar.org/repos/?p=getaclue.git;a=blobdiff_plain;f=crosswordpuzzle.py;h=012450f07e320e1a989cf77e9f380f4baf0bf4ae;hp=9617897fddada51883187489cdb2153e268ab6f6;hb=2f775ee1c4d2c260221155494b2c7856bc38dafd;hpb=8e802b4a6d2d5af14810a2b6603fd53b2697c039 diff --git a/crosswordpuzzle.py b/crosswordpuzzle.py index 9617897..012450f 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 ("\n") + across_data.append ("\n") + across_data.append (title + "\n") + across_data.append ("<AUTHOR>\n") + across_data.append (author + "\n") + across_data.append ("<COPYRIGHT>\n") + across_data.append (copyright + "\n") + across_data.append ("<SIZE>\n") + str_size = str (self.cols) + "x" + str (self.rows) + across_data.append (str_size + "\n") + across_data.append ("<GRID>\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 ("\n") + + across_data.append ("<ACROSS>\n") + clues_across = self.get_clues_across () + for word, clue in clues_across: + if clue: + across_data.append (clue + "\n") + else: + across_data.append ("(No clue yet)\n") + + across_data.append ("<DOWN>\n") + clues_down = self.get_clues_down () + for word, clue in clues_down: + if clue: + across_data.append (clue + "\n") + else: + across_data.append ("(No clue yet\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):