Django 1.8

[prawokultury.git] / events / templatetags / events_tags.py

diff --git a/events/templatetags/events_tags.py b/events/templatetags/events_tags.py
index a51bda9..3bd5135 100644 (file)
--- a/events/templatetags/events_tags.py
+++ b/events/templatetags/events_tags.py
@@ -3,6 +3,7 @@
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 from django import template
+from django.utils.timezone import now
 from events import app_settings
 from events.models import Event
 
@@ -12,5 +13,12 @@ from datetime import datetime
 
 @register.inclusion_tag('events/snippets/events_box.html', takes_context=True)
 def events_box(context, limit=app_settings.BOX_LENGTH):
-    objects = Event.objects.filter(date__gte=datetime.now())[:limit]
+    lang = context['request'].LANGUAGE_CODE
+    objects = Event.objects.filter(**{"published_%s" % lang: True})
+    objects = objects.filter(date_end__gte=now())[:limit]
     return {'objects': objects}
+
+
+@register.inclusion_tag('events/snippets/event_short.html')
+def event_short(event):
+    return {'event': event}