Completed the HTML exporter

[getaclue.git] / crosswordpuzzle.py
diff --git a/crosswordpuzzle.py b/crosswordpuzzle.py

index e660800..7d3b270 100644 (file)
--- a/crosswordpuzzle.py
+++ b/crosswordpuzzle.py
@@ -4,6 +4,7 @@
  
  # Class for the puzzle data representation
  
+# for export to PNG image
  import cairo
  
  class GridItem:
@@ -76,7 +77,8 @@ class CrosswordPuzzle:
                                 self.data[i].append (GridItem ())
  
         # export to an image
-       def export_image (self, filename, solution=False):
+       def export_image (self, pngfile, htmlfile=None, puztitle="Crossword Puzzle",
+               solution=True):
                 # don't export if grid is not frozen
                 if self.frozen_grid is False:
                         raise FrozenGridException
@@ -90,6 +92,11 @@ class CrosswordPuzzle:
                 ctx.rectangle (0, 0, self.cols*px, self.rows*px)
                 ctx.fill ()
  
+               # get the clues across and down
+               clues_across = self.get_clues_across ()
+               clues_down = self.get_clues_down ()
+
+
                 # traverse through the grid
                 for row in range (self.rows):
                         for col in range (self.cols):
@@ -121,7 +128,43 @@ class CrosswordPuzzle:
                                                 ctx.move_to (col*px+10, row*px+20)
                                                 ctx.show_text (self.data[row][col].char)
  
-               surf.write_to_png (open (filename, "wb"))
+               surf.write_to_png (open (pngfile, "wb"))
+
+               # if solution is false, publish the clues and the image in a HTML file
+               if htmlfile and solution is False:
+                       html_contents = ["<html>", "<head>", "<title>"]
+                       html_contents.append (puztitle)
+                       html_contents.append ("</title>")
+                       html_contents.append ("</head>")
+                       html_contents.append ("<body>")
+                       html_contents.append ("<h1>" + puztitle + "</h1>")
+                       html_contents.append ('<img src="' + pngfile + '" alt="puzzle" />')
+
+                       html_contents.append ("<h2>Across clues</h2>")
+                       html_contents.append ("<p>")
+                       for word, clue in clues_across:
+                               clue_str = str (self.data[word[1]][word[2]].numbered) + " - " \
+                                                       + clue
+                               html_contents.append (clue_str)
+                               html_contents.append ("<br />")
+                       html_contents.append ("</p>")
+
+                       html_contents.append ("<h2>Down clues</h2>")
+                       html_contents.append ("<p>")
+                       for word, clue in clues_down:
+                               clue_str = str (self.data[word[1]][word[2]].numbered) + " - " \
+                                                       + clue
+                               html_contents.append (clue_str)
+                               html_contents.append ("<br />")
+                       html_contents.append ("</p>")
+                       html_contents.append ("</body>")
+                       html_contents.append ("</html>")
+
+                       html_str = "\r\n".join (html_contents)
+
+                       fhtml = open (htmlfile, "wb")
+                       fhtml.write (html_str)
+                       fhtml.close ()
  
         # get the AcrossLite(TM) data for exporting
         def export_acrosslite (self, title, author, copyright):