Organizer/MultiEvent-Exports (#1684)

Co-authored-by: Raphael Michel <michel@rami.io>

src/pretix/base/services/export.py

@@ -5,9 +5,15 @@ from django.utils.timezone import override
 from django.utils.translation import gettext
 
 from pretix.base.i18n import LazyLocaleException, language
-from pretix.base.models import CachedFile, Event, cachedfile_name
-from pretix.base.services.tasks import ProfiledEventTask
-from pretix.base.signals import register_data_exporters
+from pretix.base.models import (
+    CachedFile, Event, Organizer, User, cachedfile_name,
+)
+from pretix.base.services.tasks import (
+    ProfiledEventTask, ProfiledOrganizerUserTask,
+)
+from pretix.base.signals import (
+    register_data_exporters, register_multievent_data_exporters,
+)
 from pretix.celery_app import app
 
 
@@ -32,3 +38,26 @@ def export(event: Event, fileid: str, provider: str, form_data: Dict[str, Any])
                 file.file.save(cachedfile_name(file, file.filename), ContentFile(data))
                 file.save()
     return file.pk
+
+
+@app.task(base=ProfiledOrganizerUserTask, throws=(ExportError,))
+def multiexport(organizer: Organizer, user: User, fileid: str, provider: str, form_data: Dict[str, Any]) -> None:
+    file = CachedFile.objects.get(id=fileid)
+    with language(user.locale), override(user.timezone):
+        allowed_events = user.get_events_with_permission('can_view_orders')
+
+        events = allowed_events.filter(pk__in=form_data.get('events'))
+        responses = register_multievent_data_exporters.send(organizer)
+
+        for receiver, response in responses:
+            ex = response(events)
+            if ex.identifier == provider:
+                d = ex.render(form_data)
+                if d is None:
+                    raise ExportError(
+                        gettext('Your export did not contain any data.')
+                    )
+                file.filename, file.type, data = d
+                file.file.save(cachedfile_name(file, file.filename), ContentFile(data))
+                file.save()
+    return file.pk

src/pretix/base/services/tasks.py

@@ -19,7 +19,7 @@ from django_scopes import scope, scopes_disabled
 from pretix.base.metrics import (
     pretix_task_duration_seconds, pretix_task_runs_total,
 )
-from pretix.base.models import Event
+from pretix.base.models import Event, Organizer, User
 from pretix.celery_app import app
 
 
@@ -88,10 +88,30 @@ class EventTask(app.Task):
         return ret
 
 
+class OrganizerUserTask(app.Task):
+    def __call__(self, *args, **kwargs):
+        organizer_id = kwargs['organizer']
+        with scopes_disabled():
+            organizer = Organizer.objects.get(pk=organizer_id)
+        kwargs['organizer'] = organizer
+
+        user_id = kwargs['user']
+        user = User.objects.get(pk=user_id)
+        kwargs['user'] = user
+
+        with scope(organizer=organizer):
+            ret = super().__call__(*args, **kwargs)
+        return ret
+
+
 class ProfiledEventTask(ProfiledTask, EventTask):
     pass
 
 
+class ProfiledOrganizerUserTask(ProfiledTask, OrganizerUserTask):
+    pass
+
+
 class TransactionAwareTask(ProfiledTask):
     """
     Task class which is aware of django db transactions and only executes tasks