aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--wkhtmltopdf/views.py37
1 files changed, 27 insertions, 10 deletions
diff --git a/wkhtmltopdf/views.py b/wkhtmltopdf/views.py
index ac406a3..9ff5395 100644
--- a/wkhtmltopdf/views.py
+++ b/wkhtmltopdf/views.py
@@ -71,16 +71,7 @@ class PDFTemplateResponse(TemplateResponse, PDFResponse):
context = self.resolve_context(self.context_data)
content = smart_str(template.render(context))
- # mattl:
- # convert all MEDIA_ROOT files into a file://URL paths
- # in order to correctly get it displayed in PDFs
- # disclaimer: I know it sucks, but I haz no time for better solution now
- media_root = settings.MEDIA_ROOT
- if not media_root.endswith('/'):
- media_root += '/'
- relative_path = settings.MEDIA_URL
- for x in re.findall('''["|']({0}.*?)["|']'''.format(relative_path), content):
- content = content.replace(x, pathname2fileurl(media_root) + x[len(relative_path):])
+ content = self.make_absolute_paths(content)
tempfile = NamedTemporaryFile(mode=mode, bufsize=bufsize,
suffix=suffix, prefix=prefix,
@@ -152,6 +143,32 @@ class PDFTemplateResponse(TemplateResponse, PDFResponse):
for f in filter(None, (input_file, header_file, footer_file)):
f.close()
+ def make_absolute_paths(self, content):
+ """Convert all MEDIA files into a file://URL paths in order to correctly get it displayed in PDFs
+
+ mattl's disclaimer: I know it sucks, but it works and I haz no time for better solution now
+ """
+
+ overrides = [
+ {
+ 'root': settings.MEDIA_ROOT,
+ 'url': settings.MEDIA_URL,
+ },
+ {
+ 'root': settings.STATIC_ROOT,
+ 'url': settings.STATIC_URL,
+ }
+ ]
+ has_scheme = re.compile(r'^[^:/]+://')
+
+ for x in overrides:
+ if not has_scheme.match(x['url']):
+ if not x['root'].endswith('/'):
+ x['root'] += '/'
+ for occur in re.findall('''["|']({0}.*?)["|']'''.format(x['url']), content):
+ content = content.replace(occur, pathname2fileurl(x['root']) + occur[len(x['url']):])
+ return content
+
class PDFTemplateView(TemplateView):
"""Class-based view for HTML templates rendered to PDF."""