import dramatiq
from django.conf import settings
from django.core.files.base import ContentFile
from django.urls import reverse
from gotenberg_client import GotenbergClient
from invoices.models import Invoice
@dramatiq.actor()
def generate_pdf(invoice_id: int):
invoice = Invoice.objects.get(pk=invoice_id)
print_url = f'{settings.INTERNAL_API_URL}{reverse("invoices:print_invoice", kwargs=dict(invoice_id=invoice.id))}'
invoice_file = invoice.files.create()
with GotenbergClient(settings.GOTENBERG_API_URL) as client:
with client.chromium.url_to_pdf() as task:
res = task.url(print_url).run()
invoice_file.file.save(
invoice_file.get_file_name('pdf'),
ContentFile(res.content),
)