Django: сохранение pdf из представления в модель (reportlab)

Возможно ли в Django сохранить pdf из view в model (одновременно загружая его)?


Пока что эти шаги работают:

  • pdf отображается в новой вкладке, и я могу его скачать
  • экземпляр модели создан (но пустой)

Что не работает:

  • у созданных экземпляров модели нет пути к файлу и нигде на сервере не сохранен файл pdf

Мой код:

Модели

from django.db import models
class TestModel(models.Model):
    
    def pdf_upload_path(instance, filename):
        return f'utility_bills/{instance.created_date.strftime("%Y-%m-%d")}_test_{filename}'

    created_date = models.DateTimeField(
        auto_now=False, 
        auto_now_add=True, 
        null=True, 
        blank=True, 
    )
    pdf = models.FileField(upload_to=pdf_upload_path, blank=True)

Виды

import io
from django.core.files.base import ContentFile
from django.http import FileResponse
from reportlab.platypus import SimpleDocTemplate, Paragraph
from .models import (TestModel)

## Create PDF ##
def utility_pdf(request):
    
    # General Setup
    pdf_buffer = io.BytesIO()    
    my_doc = SimpleDocTemplate(pdf_buffer)
    sample_style_sheet = getSampleStyleSheet()

    # Instantiate flowables
    flowables = []
    test = Paragraph("Test", sample_style_sheet['BodyText'])
    
    # Append flowables and build doc
    flowables.append(test)    
    my_doc.build(flowables)
    
    # create and save pdf
    pdf_buffer.seek(0)    
    pdf = pdf_buffer.getvalue()
    file_data = ContentFile(pdf)
    pdf = TestModel(pdf=file_data)
    pdf.save()
    response = FileResponse(pdf_buffer, filename="some_file.pdf")
    
    return response

Я нашел его! 😀

Все, что мне нужно было сделать, это дать моему pdf имя - и теперь он работает!

screenshot

pdf_data = pdf_buffer.getvalue()
file_data = ContentFile(pdf_data)

## Here is the crucial part that was missing: ##
file_data.name = 'test.pdf'

pdf = UtilityBill(lease=pdf_lease, pdf=file_data)
pdf.save()
response = FileResponse(pdf_buffer, filename="some_file.pdf")
Вернуться на верх