CGM_Public/pretix_cgo
1
0
Fork 0
forked from CGM_Public/pretix_original
Code Pull Requests Actions Activity

Add company to checkin list

Browse Source
This commit is contained in:
Raphael Michel
2018-11-08 11:01:29 +01:00
parent 4477f8001e
commit 0a442e712b
1 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/pretix/plugins/checkinlists/exporters.py
View File

@@ -17,7 +17,9 @@ from reportlab.lib.units import mm
from reportlab.platypus import Flowable, Paragraph, Spacer, Table, TableStyle from reportlab.platypus import Flowable, Paragraph, Spacer, Table, TableStyle
from pretix.base.exporter import BaseExporter from pretix.base.exporter import BaseExporter
from pretix.base.models import Checkin, Order, OrderPosition, Question from pretix.base.models import (
Checkin, InvoiceAddress, Order, OrderPosition, Question,
)
from pretix.base.settings import PERSON_NAME_SCHEMES from pretix.base.settings import PERSON_NAME_SCHEMES
from pretix.base.templatetags.money import money_filter from pretix.base.templatetags.money import money_filter
from pretix.control.forms.widgets import Select2 from pretix.control.forms.widgets import Select2
@@ -332,9 +334,15 @@ class CSVCheckinList(BaseCheckinList):
for q in questions: for q in questions:
headers.append(str(q.question)) headers.append(str(q.question))
headers.append(_('Company'))
writer.writerow(headers) writer.writerow(headers)
for op in qs: for op in qs:
try:
ia = op.order.invoice_address
except InvoiceAddress.DoesNotExist:
ia = InvoiceAddress()
last_checked_in = None last_checked_in = None
if isinstance(op.last_checked_in, str): # SQLite if isinstance(op.last_checked_in, str): # SQLite
last_checked_in = dateutil.parser.parse(op.last_checked_in) last_checked_in = dateutil.parser.parse(op.last_checked_in)
@@ -344,12 +352,16 @@ class CSVCheckinList(BaseCheckinList):
last_checked_in = make_aware(last_checked_in, UTC) last_checked_in = make_aware(last_checked_in, UTC)
row = [ row = [
op.order.code, op.order.code,
op.attendee_name or (op.addon_to.attendee_name if op.addon_to else ''), op.attendee_name or (op.addon_to.attendee_name if op.addon_to else '') or ia.name,
] ]
if len(name_scheme['fields']) > 1: if len(name_scheme['fields']) > 1:
for k, label, w in name_scheme['fields']: for k, label, w in name_scheme['fields']:
row.append( row.append(
(op.attendee_name_parts or (op.addon_to.attendee_name_parts if op.addon_to else {})).get(k, '') (
op.attendee_name_parts or
(op.addon_to.attendee_name_parts if op.addon_to else {}) or
ia.name_parts
).get(k, '')
) )
row += [ row += [
str(op.item) + (" – " + str(op.variation.value) if op.variation else ""), str(op.item) + (" – " + str(op.variation.value) if op.variation else ""),
@@ -373,6 +385,7 @@ class CSVCheckinList(BaseCheckinList):
for q in questions: for q in questions:
row.append(acache.get(q.pk, '')) row.append(acache.get(q.pk, ''))
row.append(ia.company)
writer.writerow(row) writer.writerow(row)
return '{}_checkin.csv'.format(self.event.slug), 'text/csv', output.getvalue().encode("utf-8") return '{}_checkin.csv'.format(self.event.slug), 'text/csv', output.getvalue().encode("utf-8")