pdf.py

import io

import weasyprint

from django.template.response import TemplateResponse


class PDFResponse(TemplateResponse):

    def __init__(self, *args, **kwargs):
        filename = kwargs.pop('filename', None)

        super().__init__(*args, **kwargs)

        self['Content-Disposition'] = 'attachment; filename={}'.format(filename)
        self['Content-Type'] = 'application/pdf'

    @property
    def rendered_content(self):
        template = self.resolve_template(self.template_name)
        context = self.resolve_context(self.context_data)

        rendered_html = template.render(context)

        pdf = self.content_processor(rendered_html)

        return pdf

    def content_processor(self, content):
        """ Define como o HTML deve ser transformado em PDF """
        output = io.StringIO()

        pdf = weasyprint.HTML(string=content)
        pdf.write_pdf(target=output)

        return output.getvalue()


class PDFView:
    filename = 'output.pdf'
    response_class = PDFResponse

    def get_filename(self):
        return self.filename

    def render_to_response(self, context, **response_kwargs):
        response_kwargs.update({
            'filename': self.get_filename()
        })

        return super().render_to_response(context, **response_kwargs)