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