paul@0 | 1 | # -*- coding: iso-8859-1 -*- |
paul@0 | 2 | """ |
paul@0 | 3 | MoinMoin - Export as PDF action |
paul@0 | 4 | |
paul@0 | 5 | @copyright: 2013 by Paul Boddie <paul@boddie.org.uk> |
paul@0 | 6 | @license: GNU GPL (v2 or later), see COPYING.txt for details. |
paul@0 | 7 | """ |
paul@0 | 8 | |
paul@3 | 9 | from MoinMoin import caching |
paul@3 | 10 | from MoinMoin.action import ActionBase, cache |
paul@0 | 11 | from MoinMoin.wikiutil import escape |
paul@0 | 12 | from MoinSupport import ActionSupport, escattr, getFormatterClass, formatText, get_send_headers |
paul@0 | 13 | from os.path import join |
paul@1 | 14 | import subprocess, os, codecs |
paul@0 | 15 | |
paul@0 | 16 | __version__ = "0.1" |
paul@0 | 17 | |
paul@0 | 18 | Dependencies = ['page'] |
paul@0 | 19 | |
paul@0 | 20 | # Configuration settings. |
paul@0 | 21 | |
paul@1 | 22 | # Choose one value for the export mode. |
paul@1 | 23 | |
paul@1 | 24 | PDF_EXPORT_MODE = "docbook" |
paul@2 | 25 | #PDF_EXPORT_MODE = "wkhtmltopdf" |
paul@1 | 26 | #PDF_EXPORT_MODE = "htmldoc" |
paul@1 | 27 | |
paul@1 | 28 | # Settings for "docbook" mode. |
paul@1 | 29 | |
paul@0 | 30 | XSLT_PROCESSOR = "/usr/bin/xsltproc" |
paul@0 | 31 | FO_PROCESSOR = "/usr/bin/fop" |
paul@0 | 32 | DOCBOOK_STYLESHEET_BASE = "/usr/share/xml/docbook/stylesheet" |
paul@0 | 33 | |
paul@1 | 34 | # Tool settings for "docbook" mode. |
paul@0 | 35 | |
paul@0 | 36 | DOCBOOK_TO_FO_STYLESHEET = "docbook-xsl/fo/docbook.xsl" |
paul@0 | 37 | |
paul@2 | 38 | # Settings for "wkhtmltopdf" mode. |
paul@2 | 39 | |
paul@2 | 40 | XVFB_WRAPPER = "/usr/bin/xvfb-run" |
paul@2 | 41 | WKHTMLTOPDF_PROCESSOR = "/usr/bin/wkhtmltopdf" |
paul@2 | 42 | |
paul@1 | 43 | # Settings for "htmldoc" mode. |
paul@1 | 44 | |
paul@1 | 45 | HTMLDOC_PROCESSOR = "/usr/bin/htmldoc" |
paul@1 | 46 | |
paul@0 | 47 | # NOTE: From docbook-xsl/fo/param.xsl. |
paul@0 | 48 | |
paul@1 | 49 | docbook_paper_sizes = [ |
paul@0 | 50 | "A0", "A1", "A2", "A3", "A4", "A5", "A6", "A7", "A8", "A9", "A10", |
paul@0 | 51 | "B0", "B1", "B2", "B3", "B4", "B5", "B6", "B7", "B8", "B9", "B10", |
paul@0 | 52 | "C0", "C1", "C2", "C3", "C4", "C5", "C6", "C7", "C8", "C9", "C10", |
paul@0 | 53 | "A4landscape", "USletter", "USlandscape", "4A0", "2A0", |
paul@0 | 54 | ] |
paul@0 | 55 | |
paul@1 | 56 | docbook_paper_size_labels = { |
paul@0 | 57 | "A4landscape" : "A4 landscape", |
paul@0 | 58 | "USletter" : "US letter", |
paul@0 | 59 | "USlandscape" : "US landscape", |
paul@0 | 60 | "4A0" : "Quadruple A0", |
paul@0 | 61 | "2A0" : "Double A0" |
paul@0 | 62 | } |
paul@0 | 63 | |
paul@2 | 64 | wkhtmltopdf_paper_sizes = [ |
paul@2 | 65 | "A4", "Letter" |
paul@2 | 66 | ] |
paul@2 | 67 | |
paul@2 | 68 | wkhtmltopdf_paper_size_labels = {} |
paul@2 | 69 | |
paul@1 | 70 | # NOTE: From the htmldoc man page. |
paul@1 | 71 | |
paul@1 | 72 | htmldoc_paper_sizes = [ |
paul@1 | 73 | "a4", "legal", "letter", "universal" |
paul@1 | 74 | ] |
paul@1 | 75 | |
paul@1 | 76 | htmldoc_paper_size_labels = { |
paul@1 | 77 | "a4" : "A4", |
paul@1 | 78 | "legal" : "US legal", |
paul@1 | 79 | "letter" : "US letter", |
paul@1 | 80 | "universal" : "US universal" |
paul@1 | 81 | } |
paul@1 | 82 | |
paul@2 | 83 | paper_sizes = { |
paul@2 | 84 | "docbook" : docbook_paper_sizes, |
paul@2 | 85 | "wkhtmltopdf" : wkhtmltopdf_paper_sizes, |
paul@2 | 86 | "htmldoc" : htmldoc_paper_sizes |
paul@2 | 87 | } |
paul@2 | 88 | |
paul@2 | 89 | paper_size_labels = { |
paul@2 | 90 | "docbook" : docbook_paper_size_labels, |
paul@2 | 91 | "wkhtmltopdf" : wkhtmltopdf_paper_size_labels, |
paul@2 | 92 | "htmldoc" : htmldoc_paper_size_labels |
paul@2 | 93 | } |
paul@2 | 94 | |
paul@0 | 95 | class ExportPDF(ActionBase, ActionSupport): |
paul@0 | 96 | |
paul@0 | 97 | "Export the current page as PDF." |
paul@0 | 98 | |
paul@1 | 99 | mode = PDF_EXPORT_MODE |
paul@1 | 100 | |
paul@1 | 101 | def _get_paper_sizes(self): |
paul@2 | 102 | return paper_sizes.get(self.mode) |
paul@1 | 103 | |
paul@1 | 104 | def _get_paper_size_labels(self): |
paul@2 | 105 | return paper_size_labels.get(self.mode) |
paul@1 | 106 | |
paul@0 | 107 | def get_form_html(self, buttons_html): |
paul@0 | 108 | |
paul@0 | 109 | "Return the action's form incorporating the 'buttons_html'." |
paul@0 | 110 | |
paul@0 | 111 | _ = self._ |
paul@0 | 112 | request = self.request |
paul@0 | 113 | form = self.get_form() |
paul@0 | 114 | |
paul@0 | 115 | paper_size = form.get("paper-size", ["A4"])[0] |
paul@0 | 116 | |
paul@0 | 117 | paper_size_options = [] |
paul@2 | 118 | paper_size_labels = self._get_paper_size_labels() or {} |
paul@0 | 119 | |
paul@2 | 120 | for size in self._get_paper_sizes() or []: |
paul@0 | 121 | paper_size_options.append('<option value="%s" %s>%s</option>' % ( |
paul@0 | 122 | escattr(size), self._get_selected(size, paper_size), |
paul@0 | 123 | escape(_(paper_size_labels.get(size) or size)) |
paul@0 | 124 | )) |
paul@0 | 125 | |
paul@0 | 126 | d = { |
paul@0 | 127 | "paper_size_label" : escape(_("Paper size")), |
paul@0 | 128 | "paper_size_options" : u"".join(paper_size_options), |
paul@0 | 129 | "buttons_html" : buttons_html, |
paul@0 | 130 | } |
paul@0 | 131 | |
paul@0 | 132 | return u"""\ |
paul@0 | 133 | <table> |
paul@0 | 134 | <tr> |
paul@0 | 135 | <td class="label"><label>%(paper_size_label)s</label></td> |
paul@0 | 136 | <td><select name="paper-size">%(paper_size_options)s</select></td> |
paul@0 | 137 | </tr> |
paul@0 | 138 | <tr> |
paul@0 | 139 | <td></td> |
paul@0 | 140 | <td class="buttons">%(buttons_html)s</td> |
paul@0 | 141 | </tr> |
paul@0 | 142 | </table> |
paul@0 | 143 | """ % d |
paul@0 | 144 | |
paul@0 | 145 | def do_action(self): |
paul@0 | 146 | |
paul@0 | 147 | "Attempt to post a comment." |
paul@0 | 148 | |
paul@0 | 149 | _ = self._ |
paul@0 | 150 | form = self.get_form() |
paul@3 | 151 | request = self.request |
paul@0 | 152 | |
paul@0 | 153 | paper_size = form.get("paper-size", [""])[0] |
paul@0 | 154 | |
paul@2 | 155 | if not paper_size in self._get_paper_sizes() or []: |
paul@0 | 156 | return 0, _("A paper size must be chosen.") |
paul@0 | 157 | |
paul@3 | 158 | # See if the revision is cached. |
paul@3 | 159 | |
paul@3 | 160 | cache_key = cache.key(request, content="%s-%s" % (self.page.rev, paper_size)) |
paul@3 | 161 | cache_entry = caching.CacheEntry(request, self.page, cache_key, scope="item") |
paul@3 | 162 | |
paul@3 | 163 | # Open any available cache entry and read it. |
paul@3 | 164 | |
paul@3 | 165 | if cache_entry.exists(): |
paul@3 | 166 | cache_entry.open() |
paul@3 | 167 | try: |
paul@3 | 168 | self._write_pdf(cache_entry.read()) |
paul@3 | 169 | return 1, None |
paul@3 | 170 | finally: |
paul@3 | 171 | cache_entry.close() |
paul@3 | 172 | |
paul@3 | 173 | # Otherwise, prepare the PDF. |
paul@3 | 174 | |
paul@1 | 175 | if self.mode == "docbook": |
paul@3 | 176 | return self._export_using_docbook(paper_size, cache_entry) |
paul@2 | 177 | elif self.mode == "wkhtmltopdf": |
paul@3 | 178 | return self._export_using_wkhtmltopdf(paper_size, cache_entry) |
paul@1 | 179 | elif self.mode == "htmldoc": |
paul@3 | 180 | return self._export_using_htmldoc(paper_size, cache_entry) |
paul@1 | 181 | else: |
paul@1 | 182 | return 0, _("The action must be configured to use a particular PDF generation tool.") |
paul@1 | 183 | |
paul@2 | 184 | def _get_page_as_html(self): |
paul@2 | 185 | |
paul@2 | 186 | "Get the page in HTML format." |
paul@1 | 187 | |
paul@1 | 188 | request = self.request |
paul@1 | 189 | page = self.page |
paul@1 | 190 | |
paul@1 | 191 | fmt = getFormatterClass(request, "text_html")(request) |
paul@1 | 192 | fmt.setPage(page) |
paul@1 | 193 | |
paul@1 | 194 | page_as_html = [] |
paul@1 | 195 | append = page_as_html.append |
paul@1 | 196 | |
paul@1 | 197 | append("""\ |
paul@1 | 198 | <html> |
paul@1 | 199 | <head> |
paul@1 | 200 | <meta http-equiv="Content-Type" content="text/html; charset=utf-8"> |
paul@1 | 201 | </head> |
paul@1 | 202 | <body> |
paul@1 | 203 | """) |
paul@1 | 204 | append(formatText(page.get_raw_body(), request, fmt, inhibit_p=False)) |
paul@1 | 205 | append("""\ |
paul@1 | 206 | </body> |
paul@1 | 207 | </html> |
paul@1 | 208 | """) |
paul@1 | 209 | |
paul@2 | 210 | return u"".join(page_as_html) |
paul@1 | 211 | |
paul@2 | 212 | def _get_page_as_docbook(self): |
paul@1 | 213 | |
paul@2 | 214 | "Get the page in DocBook format." |
paul@1 | 215 | |
paul@1 | 216 | request = self.request |
paul@1 | 217 | page = self.page |
paul@1 | 218 | |
paul@0 | 219 | fmt = getFormatterClass(request, "text_docbook")(request) |
paul@0 | 220 | fmt.setPage(page) |
paul@0 | 221 | |
paul@0 | 222 | # The DocBook formatter needs to pretend a full document is being made. |
paul@0 | 223 | |
paul@0 | 224 | page_as_docbook = [] |
paul@0 | 225 | append = page_as_docbook.append |
paul@0 | 226 | |
paul@0 | 227 | append(fmt.startDocument(page.page_name)) |
paul@0 | 228 | append(fmt.startContent()) |
paul@0 | 229 | append(formatText(page.get_raw_body(), request, fmt, inhibit_p=False).encode("utf-8")) |
paul@0 | 230 | append(fmt.endContent()) |
paul@0 | 231 | append(fmt.endDocument()) |
paul@0 | 232 | |
paul@2 | 233 | return "".join(page_as_docbook) |
paul@2 | 234 | |
paul@2 | 235 | def _write_pdf_for_html(self, p, page_as_html): |
paul@2 | 236 | |
paul@2 | 237 | """ |
paul@2 | 238 | Write to the process 'p', the HTML for the page, reading the PDF output |
paul@2 | 239 | from the process and writing it to the browser. |
paul@2 | 240 | """ |
paul@2 | 241 | |
paul@2 | 242 | writer = codecs.getwriter("utf-8")(p.stdin) |
paul@2 | 243 | writer.write(page_as_html) |
paul@2 | 244 | |
paul@2 | 245 | out, err = p.communicate() |
paul@2 | 246 | |
paul@2 | 247 | retcode = p.wait() |
paul@2 | 248 | |
paul@2 | 249 | if retcode != 0: |
paul@2 | 250 | return 0, err |
paul@2 | 251 | |
paul@2 | 252 | self._write_pdf(out) |
paul@2 | 253 | return 1, None |
paul@2 | 254 | |
paul@3 | 255 | def _export_using_wkhtmltopdf(self, paper_size, cache_entry): |
paul@2 | 256 | |
paul@2 | 257 | """ |
paul@2 | 258 | Send the page HTML to the processor, indicating the given 'paper_size'. |
paul@2 | 259 | """ |
paul@2 | 260 | |
paul@2 | 261 | p = subprocess.Popen([ |
paul@2 | 262 | XVFB_WRAPPER, "--", |
paul@2 | 263 | WKHTMLTOPDF_PROCESSOR, |
paul@2 | 264 | "--page-size", paper_size, |
paul@2 | 265 | "-", |
paul@2 | 266 | "-" |
paul@2 | 267 | ], |
paul@2 | 268 | shell=False, |
paul@2 | 269 | stdin=subprocess.PIPE, |
paul@2 | 270 | stdout=subprocess.PIPE, |
paul@2 | 271 | stderr=subprocess.PIPE) |
paul@2 | 272 | |
paul@3 | 273 | return self._write_pdf_for_html(p, self._get_page_as_html(), cache_entry) |
paul@2 | 274 | |
paul@3 | 275 | def _export_using_htmldoc(self, paper_size, cache_entry): |
paul@2 | 276 | |
paul@2 | 277 | """ |
paul@2 | 278 | Send the page HTML to the processor, indicating the given 'paper_size'. |
paul@2 | 279 | """ |
paul@2 | 280 | |
paul@2 | 281 | os.environ["HTMLDOC_NOCGI"] = "1" |
paul@2 | 282 | |
paul@2 | 283 | p = subprocess.Popen([ |
paul@2 | 284 | HTMLDOC_PROCESSOR, |
paul@2 | 285 | "-t", "pdf", "--quiet", "--webpage", |
paul@2 | 286 | "--size", paper_size, |
paul@2 | 287 | "-" |
paul@2 | 288 | ], |
paul@2 | 289 | shell=False, |
paul@2 | 290 | stdin=subprocess.PIPE, |
paul@2 | 291 | stdout=subprocess.PIPE, |
paul@2 | 292 | stderr=subprocess.PIPE) |
paul@2 | 293 | |
paul@3 | 294 | return self._write_pdf_for_html(p, self._get_page_as_html(), cache_entry) |
paul@2 | 295 | |
paul@3 | 296 | def _export_using_docbook(self, paper_size, cache_entry): |
paul@2 | 297 | |
paul@2 | 298 | """ |
paul@2 | 299 | Send the page DocBook XML to the processor, indicating the given |
paul@2 | 300 | 'paper_size'. |
paul@2 | 301 | """ |
paul@0 | 302 | |
paul@0 | 303 | p1 = subprocess.Popen([ |
paul@0 | 304 | XSLT_PROCESSOR, |
paul@0 | 305 | "-stringparam", "fop1.extensions", "1", |
paul@0 | 306 | "--stringparam", "paper.type", paper_size, |
paul@0 | 307 | join(DOCBOOK_STYLESHEET_BASE, DOCBOOK_TO_FO_STYLESHEET), |
paul@0 | 308 | "-" |
paul@0 | 309 | ], |
paul@0 | 310 | shell=False, |
paul@0 | 311 | stdin=subprocess.PIPE, |
paul@0 | 312 | stdout=subprocess.PIPE, |
paul@0 | 313 | stderr=subprocess.PIPE) |
paul@0 | 314 | |
paul@2 | 315 | p1.stdin.write(self._get_page_as_docbook()) |
paul@0 | 316 | p1.stdin.close() |
paul@0 | 317 | |
paul@0 | 318 | # Pipe the XML-FO output to the FO processor. |
paul@0 | 319 | |
paul@0 | 320 | p2 = subprocess.Popen([ |
paul@0 | 321 | FO_PROCESSOR, |
paul@0 | 322 | "-fo", "-", |
paul@0 | 323 | "-pdf", "-", |
paul@0 | 324 | ], |
paul@0 | 325 | shell=False, |
paul@0 | 326 | stdin=p1.stdout, |
paul@0 | 327 | stdout=subprocess.PIPE, |
paul@0 | 328 | stderr=subprocess.PIPE) |
paul@0 | 329 | |
paul@0 | 330 | out, err = p2.communicate() |
paul@0 | 331 | |
paul@0 | 332 | retcode = p1.wait() |
paul@0 | 333 | |
paul@0 | 334 | if retcode != 0: |
paul@0 | 335 | return 0, err |
paul@0 | 336 | |
paul@0 | 337 | retcode = p2.wait() |
paul@0 | 338 | |
paul@0 | 339 | if retcode != 0: |
paul@0 | 340 | return 0, err |
paul@0 | 341 | |
paul@3 | 342 | self._write_to_cache(out, cache_entry) |
paul@1 | 343 | self._write_pdf(out) |
paul@1 | 344 | return 1, None |
paul@1 | 345 | |
paul@3 | 346 | def _write_to_cache(self, out, cache_entry): |
paul@3 | 347 | |
paul@3 | 348 | "Write the output 'out' to the given 'cache_entry'." |
paul@3 | 349 | |
paul@3 | 350 | cache_entry.open(mode="w") |
paul@3 | 351 | try: |
paul@3 | 352 | try: |
paul@3 | 353 | cache_entry.write(out) |
paul@3 | 354 | finally: |
paul@3 | 355 | cache_entry.close() |
paul@3 | 356 | except IOError: |
paul@3 | 357 | if cache_entry.exists(): |
paul@3 | 358 | cache_entry.remove() |
paul@3 | 359 | |
paul@1 | 360 | def _write_pdf(self, out): |
paul@3 | 361 | |
paul@3 | 362 | "Write the output 'out' to the request/response." |
paul@3 | 363 | |
paul@1 | 364 | request = self.request |
paul@1 | 365 | |
paul@0 | 366 | send_headers = get_send_headers(request) |
paul@0 | 367 | headers = ["Content-Type: application/pdf"] |
paul@0 | 368 | send_headers(headers) |
paul@0 | 369 | request.write(out) |
paul@0 | 370 | |
paul@0 | 371 | def render_success(self, msg, msgtype=None): |
paul@0 | 372 | |
paul@0 | 373 | """ |
paul@0 | 374 | Render neither 'msg' nor 'msgtype' since a resource has already been |
paul@0 | 375 | produced. |
paul@0 | 376 | NOTE: msgtype is optional because MoinMoin 1.5.x does not support it. |
paul@0 | 377 | """ |
paul@0 | 378 | |
paul@0 | 379 | pass |
paul@0 | 380 | |
paul@0 | 381 | # Action invocation function. |
paul@0 | 382 | |
paul@0 | 383 | def execute(pagename, request): |
paul@0 | 384 | ExportPDF(pagename, request).render() |
paul@0 | 385 | |
paul@0 | 386 | # vim: tabstop=4 expandtab shiftwidth=4 |