Files
pretix_cgo/src/pretix/plugins/badges/tasks.py
2020-02-05 10:31:11 +01:00

26 lines
796 B
Python

import logging
from typing import List
from django.core.files.base import ContentFile
from pretix.base.models import (
CachedFile, Event, OrderPosition, cachedfile_name,
)
from pretix.base.services.orders import OrderError
from pretix.base.services.tasks import EventTask
from pretix.celery_app import app
from .exporters import OPTIONS, render_pdf
logger = logging.getLogger(__name__)
@app.task(base=EventTask, throws=(OrderError,))
def badges_create_pdf(event: Event, fileid: int, positions: List[int]) -> int:
file = CachedFile.objects.get(id=fileid)
pdfcontent = render_pdf(event, OrderPosition.objects.filter(id__in=positions), opt=OPTIONS['one'])
file.file.save(cachedfile_name(file, file.filename), ContentFile(pdfcontent.read()))
file.save()
return file.pk