diff options
| -rw-r--r-- | wkhtmltopdf/views.py | 37 |
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.""" |