Generate organizer-level iCal files

src/pretix/presale/ical.py

@@ -0,0 +1,61 @@
+import datetime
+from urllib.parse import urlparse
+
+import pytz
+import vobject
+from django.conf import settings
+from django.utils.formats import date_format
+from django.utils.translation import ugettext as _
+
+from pretix.base.models import Event
+from pretix.multidomain.urlreverse import build_absolute_uri
+
+
+def get_ical(events):
+    cal = vobject.iCalendar()
+    cal.add('prodid').value = '-//pretix//{}//'.format(settings.PRETIX_INSTANCE_NAME.replace(" ", "_"))
+    creation_time = datetime.datetime.now(pytz.utc)
+
+    for ev in events:
+        event = ev if isinstance(ev, Event) else ev.event
+        tz = pytz.timezone(event.settings.timezone)
+
+        vevent = cal.add('vevent')
+        vevent.add('summary').value = str(ev.name)
+        vevent.add('dtstamp').value = creation_time
+        if ev.location:
+            vevent.add('location').value = str(ev.location)
+        vevent.add('organizer').value = event.organizer.name
+        vevent.add('uid').value = 'pretix-{}-{}-{}@{}'.format(
+            event.organizer.slug, event.slug,
+            ev.pk if not isinstance(ev, Event) else '0',
+            urlparse(settings.SITE_URL).netloc
+        )
+
+        if event.settings.show_times:
+            vevent.add('dtstart').value = ev.date_from.astimezone(tz)
+        else:
+            vevent.add('dtstart').value = ev.date_from.astimezone(tz).date()
+
+        if event.settings.show_date_to and ev.date_to:
+            if event.settings.show_times:
+                vevent.add('dtend').value = ev.date_to.astimezone(tz)
+            else:
+                vevent.add('dtend').value = ev.date_to.astimezone(tz).date()
+
+        descr = []
+
+        if isinstance(ev, Event):
+            descr.append(_('Tickets: {url}').format(url=build_absolute_uri(event, 'presale:event.index')))
+        else:
+            descr.append(_('Tickets: {url}').format(url=build_absolute_uri(event, 'presale:event.index', {
+                'subevent': ev.pk
+            })))
+
+        if ev.date_admission:
+            descr.append(str(_('Admission: {datetime}')).format(
+                datetime=date_format(ev.date_admission.astimezone(tz), 'SHORT_DATETIME_FORMAT')
+            ))
+
+        vevent.add('description').value = '\n'.join(descr)
+    return cal