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