2 files changed, 89 insertions, 0 deletions
diff --git a/wkhtmltopdf/__init__.py b/wkhtmltopdf/__init__.py
new file mode 100644
index 0000000..16281fe
--- /dev/null
+++ b/wkhtmltopdf/__init__.py
@@ -0,0 +1 @@
+from .utils import *
diff --git a/wkhtmltopdf/utils.py b/wkhtmltopdf/utils.py
new file mode 100644
index 0000000..7988371
--- /dev/null
+++ b/wkhtmltopdf/utils.py
@@ -0,0 +1,88 @@
+from os import fdopen, remove
+from subprocess import Popen, PIPE, CalledProcessError
+from tempfile import mkstemp
+
+from django.conf import settings
+from django.http import HttpResponse
+from django.template import loader
+
+WKHTMLTOPDF_CMD = getattr(settings, 'WKHTMLTOPDF_CMD', 'wkhtmltopdf')
+
+def wkhtmltopdf(pages, output=None, **kwargs):
+    """
+    Converts html to PDF using http://code.google.com/p/wkhtmltopdf/.
+
+    pages: List of file paths or URLs of the html to be converted.
+    output: Optionsal output file path.
+    **kwargs: Passed to wkhtmltopdf via _extra_args() (See
+              https://github.com/antialize/wkhtmltopdf/blob/master/README_WKHTMLTOPDF
+              for acceptable args.)
+              Kwargs is passed through as arguments. e.g.:
+                  {'footer_html': 'http://example.com/foot.html'}
+              becomes
+                  '--footer-html http://example.com/foot.html'
+              Where there is no value passed, use a blank string. e.g.:
+                  {'disable_javascript': ''}
+              becomes:
+                  '--disable-javascript '
+
+    example usage:
+        wkhtmltopdf(html_path="~/example.html",
+                    dpi=300,
+                    orientation="Landscape",
+                    disable_javascript="")
+    """
+
+    def _extra_args(**kwargs):
+        """Converts kwargs into a string of flags to be passed to wkhtmltopdf."""
+        flags = ''
+        for k, v in kwargs.items():
+            flags += ' --%s %s' % (k.replace('_', '-'), v)
+        return flags
+
+    if not isinstance(pages, list):
+        pages = [pages]
+
+    kwargs['quiet'] = ''
+    args = '%s %s %s %s' % (WKHTMLTOPDF_CMD, _extra_args(**kwargs), ' '.join(pages), output or '-')
+
+    process = Popen(args, stdout=PIPE, shell=True)
+    stdoutdata, stderrdata = process.communicate()
+
+    if process.returncode != 0:
+        raise CalledProcessError(returncode, args)
+    
+    if output is None:
+        output = stdoutdata
+
+    return output
+
+def render_to_pdf(template_name, dictionary=None, context_instance=None, header_template=None, footer_template=None, **kwargs):
+    """Renders a html template as a PDF response."""
+
+    page_path = template_to_temp_file(template_name, dictionary, context_instance)
+
+    tmp_files = []
+    if header_template is not None:
+        kwargs['header_html'] = render_to_temp_file(header_template, dictionary, context_instance)
+        tmp_files.append(kwargs['header_html'])
+    if footer_template is not None:
+        kwargs['footer_html'] = render_to_temp_file(footer_template, dictionary, context_instance)
+        tmp_files.append(kwargs['footer_html'])
+
+    content = wkhtmltopdf([page_path], **kwargs)
+
+    for path in tmp_files:
+        remove(path)
+
+    return HttpResponse(content, mimetype='application/pdf')
+
+def template_to_temp_file(*args, **kwargs):
+    """Renders a template to a temp file, and returns the path of the file."""
+    fd, tmppath = mkstemp(suffix='.html')
+    f = fdopen(fd, 'wt')
+    f.write(loader.render_to_string(*args, **kwargs))
+    f.close()
+    return tmppath
+
+