f56322e9d2abc3e5cd5a9c7376bbccde8020283b
[prawokultury.git] / migdal / templatetags / migdal_tags.py
1 # -*- coding: utf-8 -*-
2 # This file is part of PrawoKultury, licensed under GNU Affero GPLv3 or later.
3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
4 #
5 from django_comments_xtd.models import XtdComment
6 from django.contrib import comments
7 from django.core.urlresolvers import reverse
8 from django import template
9 from migdal.models import Category, Entry
10 from migdal import settings
11 from django.utils.translation import ugettext_lazy as _
12
13 register = template.Library()
14
15
16 @register.simple_tag(takes_context=True)
17 def entry_begin(context, entry):
18     t = template.loader.select_template((
19         'migdal/entry/%s/entry_begin.html' % entry.type,
20         'migdal/entry/entry_begin.html',
21     ))
22     context = {
23         'request': context['request'],
24         'object': entry,
25     }
26     return t.render(template.Context(context))
27
28
29 @register.simple_tag(takes_context=True)
30 def entry_short(context, entry):
31     t = template.loader.select_template((
32         'migdal/entry/%s/entry_short.html' % entry.type,
33         'migdal/entry/entry_short.html',
34     ))
35     context = {
36         'request': context['request'],
37         'object': entry,
38     }
39     return t.render(template.Context(context))
40
41
42 @register.inclusion_tag('migdal/categories.html', takes_context=True)
43 def categories(context, taxonomy):
44     context = {
45         'request': context['request'],
46         'object_list': Category.objects.filter(taxonomy=taxonomy)
47     }
48     return context
49
50
51 @register.inclusion_tag('migdal/last_comments.html')
52 def last_comments(limit=10):
53     return {'object_list': 
54         XtdComment.objects.filter(is_public=True, is_removed=False).order_by('-submit_date')[:limit]}
55
56
57 @register.inclusion_tag(['comments/form.html'])
58 def entry_comment_form(entry):
59     return {
60             'form': comments.get_form()(entry),
61             'next': entry.get_absolute_url(),
62         }
63
64
65 class MenuItem(object):
66     html_id = None
67
68     def __init__(self, title, url, html_id=None):
69         self.title = title
70         self.url = url
71         self.html_id = html_id
72
73     def check_active(self, chooser, value):
74         self.active = chooser == 'url' and value == self.url
75
76
77 class ModelMenuItem(object):
78     def __init__(self, obj, html_id=None):
79         self.obj = obj
80         self.title = unicode(obj)
81         self.url = obj.get_absolute_url()
82         self.html_id = html_id
83
84     def check_active(self, chooser, value):
85         self.active = (chooser == 'object' and value == self.obj or
86                         chooser == 'objects' and self.obj in value)
87
88
89 class EntryTypeMenuItem(object):
90     def __init__(self, title, type_, html_id=None):
91         self.type = type_
92         self.title = title
93         self.url = reverse('migdal_entry_list_%s' % type_)
94         self.html_id = html_id
95
96     def check_active(self, chooser, value):
97         self.active = (chooser == 'object' and isinstance(value, Entry)
98                         and value.type == self.type or
99                         chooser == 'entry_type' and value == self.type)
100
101 @register.inclusion_tag('migdal/menu.html', takes_context=True)
102 def main_menu(context, chooser=None, value=None):
103     items = [
104         ModelMenuItem(Entry.objects.get(slug_pl='o-nas')),
105         EntryTypeMenuItem(_(u'Publications'), u'publications'),
106         MenuItem(_(u'Events'), reverse('events')),
107         ModelMenuItem(Category.objects.get(slug_pl='stanowisko')),
108         ModelMenuItem(Category.objects.get(slug_pl='pierwsza-pomoc')),
109     ]
110     if context['request'].LANGUAGE_CODE == 'pl':
111         items.append(MenuItem(u'en', '/en/', html_id='item-lang'))
112     else:
113         items.append(MenuItem(u'pl', '/', html_id='item-lang'))
114     for item in items:
115         item.check_active(chooser, value)
116     return {'items': items}