23 lines
778 B
Python
23 lines
778 B
Python
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),
|
|
)
|