Fix #389 -- Add event ical download feature (#413)

* added event ical download feature

* handle event settings and timezone

* add test cases for ical download

* fix failed test case for timezone settings

* using vobject lib to generate ical

* customised UID & add vobject dependency

src/pretix/presale/templates/pretixpresale/event/index.html

@@ -68,9 +68,20 @@
             {% endif %}
         </div>
     {% endif %}
-    {% if frontpage_text %}
-        {{ frontpage_text|rich_text }}
-    {% endif %}
+    <div>
+        {% if frontpage_text %}
+        <div class="pull-left">
+            {{ frontpage_text|rich_text }}
+        </div>
+        {% endif %}
+        <div class="pull-right">
+            <a href="{% eventurl event "presale:event.ical.download" %}" class="btn btn-link">
+                <i class="fa fa-calendar"></i> {% trans "Add to Calendar" %}
+            </a>
+        </div>
+        <div class="clearfix"></div>
+    </div>
+
     {% eventsignal event "pretix.presale.signals.front_page_top" %}
     {% if event.presale_is_running or event.settings.show_items_outside_presale_period %}
         <form method="post" data-asynctask

src/pretix/presale/urls.py

@@ -56,6 +56,9 @@ event_patterns = [
     url(r'^order/(?P<order>[^/]+)/(?P<secret>[A-Za-z0-9]+)/invoice/(?P<invoice>[0-9]+)$',
         pretix.presale.views.order.InvoiceDownload.as_view(),
         name='event.invoice.download'),
+    url(r'^ical$',
+        pretix.presale.views.event.EventIcalDownload.as_view(),
+        name='event.ical.download'),
     url(r'^auth/$', pretix.presale.views.event.EventAuth.as_view(), name='event.auth'),
     url(r'^$', pretix.presale.views.event.EventIndex.as_view(), name='event.index'),
 ]

src/pretix/presale/views/event.py

@@ -1,16 +1,22 @@
 import sys
+from datetime import datetime
 from importlib import import_module
 
+import pytz
+import vobject
 from django.conf import settings
 from django.core.exceptions import PermissionDenied
 from django.db.models import Count, Prefetch, Q
+from django.http import Http404, HttpResponse
 from django.shortcuts import redirect
 from django.utils.decorators import method_decorator
+from django.utils.functional import cached_property
 from django.utils.timezone import now
 from django.utils.translation import ugettext_lazy as _
 from django.views import View
 from django.views.decorators.csrf import csrf_exempt
 from django.views.generic import TemplateView
+from pytz import timezone
 
 from pretix.base.models import ItemVariation
 from pretix.multidomain.urlreverse import eventreverse
@@ -105,6 +111,48 @@ class EventIndex(EventViewMixin, CartMixin, TemplateView):
         return context
 
 
+class EventIcalDownload(EventViewMixin, View):
+
+    @cached_property
+    def event_timezone(self):
+        return timezone(self.request.event.settings.timezone)
+
+    def get(self, request, *args, **kwargs):
+        if not self.request.event:
+            raise Http404(_('Unknown event code or not authorized to access this event.'))
+
+        event = self.request.event
+        creation_time = datetime.now(pytz.utc)
+        cal = vobject.iCalendar()
+        cal.add('prodid').value = '-//pretix//{}//'.format(settings.PRETIX_INSTANCE_NAME)
+
+        vevent = cal.add('vevent')
+        vevent.add('summary').value = str(event.name)
+        vevent.add('dtstamp').value = creation_time
+        vevent.add('location').value = str(event.location)
+        vevent.add('organizer').value = event.organizer.name
+        vevent.add('uid').value = '{}-{}-{}'.format(
+            event.organizer.slug, event.slug, creation_time.strftime('%Y%m%d%H%M%S%f')
+        )
+
+        if event.settings.show_times:
+            vevent.add('dtstart').value = event.date_from.astimezone(self.event_timezone)
+        else:
+            vevent.add('dtstart').value = event.date_from.astimezone(self.event_timezone).date()
+
+        if event.settings.show_date_to:
+            if event.settings.show_times:
+                vevent.add('dtend').value = event.date_to.astimezone(self.event_timezone)
+            else:
+                vevent.add('dtend').value = event.date_to.astimezone(self.event_timezone).date()
+
+        resp = HttpResponse(cal.serialize(), content_type='text/calendar')
+        resp['Content-Disposition'] = 'attachment; filename="{}-{}.ics"'.format(
+            event.organizer.slug, event.slug
+        )
+        return resp
+
+
 class EventAuth(View):
     @method_decorator(csrf_exempt)
     def dispatch(self, request, *args, **kwargs):