diff --git a/src/pretix/base/exporter.py b/src/pretix/base/exporter.py index 18f6cea951..cb5d6ebd3c 100644 --- a/src/pretix/base/exporter.py +++ b/src/pretix/base/exporter.py @@ -1,9 +1,13 @@ import json +from collections import OrderedDict +from django import forms from django.core.serializers.json import DjangoJSONEncoder from django.dispatch import receiver +from django.utils.translation import ugettext as _ from typing import Tuple +from pretix.base.models import Order from pretix.base.signals import register_data_exporters @@ -164,13 +168,29 @@ class JSONExporter(BaseExporter): class MailExporter(BaseExporter): identifier = 'mailaddrs' - verbose_name = 'Email addresses (text file)' + verbose_name = _('Email addresses (text file)') def render(self, form_data: dict): - addrs = self.event.orders.values('email') + qs = self.event.orders.filter(status__in=form_data['status']) + addrs = qs.values('email') data = "\r\n".join(set(a['email'] for a in addrs)) return 'pretixemails.txt', 'text/plain', data.encode("utf-8") + @property + def export_form_fields(self): + return OrderedDict( + [ + ('status', + forms.MultipleChoiceField( + label=_('Filter by status'), + initial=[Order.STATUS_PENDING, Order.STATUS_PAID], + choices=Order.STATUS_CHOICE, + widget=forms.CheckboxSelectMultiple, + required=False + )), + ] + ) + @receiver(register_data_exporters, dispatch_uid="exporter_json") def register_json_export(sender, **kwargs):