生成带有附件的受保护 Django 网页的 PDF

Question

所以我正在尝试生成我在 django Web 应用程序中拥有的视图的 PDF。此视图受保护，这意味着用户必须登录并具有特定权限才能查看该页面。我还有一些附件（以FileFields 的形式存储在数据库中），我想将它们附加到 PDF 的末尾。

我已经阅读了大部分关于如何使用 pdfkit 或 reportlab 从网页生成 PDF 的帖子，但由于某种原因，所有这些帖子对我来说都失败了。

目前，我得到的最接近的是使用pdfkit 成功生成页面的 PDF，但这需要我删除要求用户登录并拥有页面权限的限制，这实际上不是一个长期的选择。我发现了几篇讨论在受保护页面上打印 pdf 并提供登录信息的帖子，但我无法让其中任何一个工作。

我还没有找到任何关于如何包含附件的信息，并且真的不知道从哪里开始。

如果需要，我很乐意用更多信息或代码 sn-ps 来更新这个问题，但这里有很多活动部分，我不想让人们用无用的信息淹没。让我知道我是否应该提供任何其他信息，并提前感谢您的帮助。

Answer 1

您可以使用pdfkit 来执行此操作。您可以使用 url 检索页面，pdfkit 将处理其余部分：

pdfkit.from_url('http://website.com/somepage', 'somepage.pdf')

您必须使用适当的标题正确访问该页面，因为它当然受到保护：

options = {
    'cookie': [
        ('cookie-name1', 'cookie-value1'),
        ('cookie-name2', 'cookie-value2'),
    ]
}

pdfkit.from_url('http://website.com/somepage', 'somepage.pdf')
`

Answer 2

如果您只是想保护它，您可以编写一个自定义身份验证后端，让您的服务器欺骗用户。方式过度杀戮，但它会解决您的问题，至少您可以了解自定义身份验证后端！ （注意：您应该使用 HTTPS。）

https://docs.djangoproject.com/en/1.11/topics/auth/customizing/#writing-an-authentication-backend

1. 在app/auth_backends.py 中创建身份验证后端
2. 将app.auth_backends.SpoofAuthBackend 后端添加到settings.py，这需要shared_secret 和user_id。
3. 创建一个 URL 路由，如 url(r'^spoof-user/(?P<user_id>\d+)/$', 'app.views.spoof_user', name="spoof-user")
4. 添加视图spoof_user，该视图必须同时调用django.contrib.auth.authenticate（调用上面#1 中的后端），并且在从authenticate(...) 获取用户后，您使用用户django.contrib.auth.login(request, user) 填充请求。最后，如果共享密钥错误，此视图应返回 HttpResponseForbidden 或将 HttpResponseRedirect 返回到您实际想要的 PDF URL（在通过 authenticate 和 login 以编程方式登录到欺骗用户之后）。

您可能希望使用 cache.set('spoof-user-%s' % user_id, RANDOM_STRING, 30) 之类的东西为每个请求创建一个随机密钥，它将共享密钥保留 30 秒以留出请求时间。然后执行pdf_response = requests.get("%s?shared_secret=1a2b3c&redirect_uri=/path/to/pdf/" % reverse('spoof-user', kwargs={'user_id': 1234}))。您的新视图将在身份验证后端测试提供的shared_secret，登录用户请求并执行重定向到request.GET.get('redirect_uri')。

Answer 3

我搞定了！通过 PyPDF2 和 pdfkit 的结合，我得到了这个非常简单的工作。它适用于受保护的页面，因为 django 负责将完整的 html 作为字符串获取，我只是将其传递给 pdfkit。它还支持附加附件，但我怀疑（尽管我没有测试过）它是否适用于 pdf 以外的任何内容。

from django.template.loader import get_template
from PyPDF2 import PdfFileWriter, PdfFileReader
import pdfkit

def append_pdf(pdf, output):
    [output.addPage(pdf.getPage(page_num)) for page_num in range(pdf.numPages)]


def render_to_pdf():
    t = get_template('app/template.html')
    c = {'context_data': context_data}

    html = t.render(c)
    pdfkit.from_string(html, 'path/to/file.pdf')

    output = PdfFileWriter()
    append_pdf(PdfFileReader(open('path/to/file.pdf', "rb")), output)

    attaches = Attachment.objects.all()

    for attach in attaches:
        append_pdf(PdfFileReader(open(attach.file.path, "rb")), output)

    output.write(open('path/to/file_with_attachments.pdf', "wb"))