Files
pretix_original/src/pretix/plugins/badges/tasks.py
2019-02-01 16:48:58 +01:00

26 lines
740 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.celery_app import app
from .exporters import render_pdf
logger = logging.getLogger(__name__)
@app.task(throws=(OrderError,))
def badges_create_pdf(fileid: int, event: int, positions: List[int]) -> int:
file = CachedFile.objects.get(id=fileid)
event = Event.objects.get(id=event)
pdfcontent = render_pdf(event, OrderPosition.objects.filter(id__in=positions))
file.file.save(cachedfile_name(file, file.filename), ContentFile(pdfcontent.read()))
file.save()
return file.pk