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 имя - и теперь он работает!
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")
