a03d16c00129e20c81285de04f48a7d6057f8ae2
[django-migdal.git] / migdal / admin.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.conf import settings
6 from django.contrib import admin
7 from django.utils.translation import ugettext_lazy as _
8 from migdal.models import Entry, Attachment
9 from migdal import app_settings
10 from fnpdjango.utils.models import filtered_model
11 from fnpdjango.utils.models.translation import translated_fields
12
13
14 class AttachmentInline(admin.TabularInline):
15     model = Attachment
16     readonly_fields = ['url']
17
18
19 def filtered_entry_admin(typ):
20     class EntryAdmin(admin.ModelAdmin):
21         def get_queryset(self, request):
22             return self.model.objects.filter(type=typ.db)
23
24         def has_add_permission(self, request):
25             return request.user.has_perm('migdal.add_entry')
26
27         def has_change_permission(self, request, obj=None):
28             return request.user.has_perm('migdal.change_entry')
29
30         def has_delete_permission(self, request, obj=None):
31             return request.user.has_perm('migdal.delete_entry')
32
33         def formfield_for_dbfield(self, db_field, **kwargs):
34             field = super(EntryAdmin, self).formfield_for_dbfield(db_field, **kwargs)
35             if db_field.name == 'categories':
36                 field.widget.attrs['style'] = 'height: 10em'
37             return field
38
39         date_hierarchy = 'date'
40         readonly_fields = ('date', 'changed_at') + translated_fields(('published_at',))
41         if app_settings.PUBLISH_DATE_EDITABLE:
42             readonly_fields += ('first_published_at',)
43         _promo_if_necessary = ('promo',) if typ.promotable else ()
44
45         fieldsets = (
46             (None, {
47                 'fields': _promo_if_necessary + (
48                     'in_stream', 'author', 'author_email', 'canonical_url', 'image',
49                     'date', 'first_published_at', 'changed_at')
50                 }),
51         ) + tuple(
52             (ln, {'fields': (
53                 ('published_%s' % lc),
54                 'published_at_%s' % lc,
55                 'title_%s' % lc,
56                 'slug_%s' % lc,
57                 'lead_%s' % lc,
58                 'body_%s' % lc,
59                 )})
60             for lc, ln in app_settings.OBLIGATORY_LANGUAGES
61         ) + tuple(
62             (ln, {'fields': (
63                 ('needed_%s' % lc, 'published_%s' % lc),
64                 'published_at_%s' % lc,
65                 'title_%s' % lc,
66                 'slug_%s' % lc,
67                 'lead_%s' % lc,
68                 'body_%s' % lc,
69                 )})
70             for lc, ln in app_settings.OPTIONAL_LANGUAGES
71         )
72
73         if typ.categorized:
74             fieldsets += (
75                 (_('Categories'), {'fields': ('categories',)}),
76             )
77         prepopulated_fields = dict([
78                 ("slug_%s" % lang_code, ("title_%s" % lang_code,))
79                 for lang_code, lang_name in settings.LANGUAGES
80             ]) 
81
82         list_display = translated_fields(('title',), app_settings.OBLIGATORY_LANGUAGES) + \
83             ('date', 'author') + \
84             _promo_if_necessary + \
85             ('in_stream', 'first_published_at',) + \
86             translated_fields(('published_at',)) + \
87             translated_fields(('needed',), app_settings.OPTIONAL_LANGUAGES)
88         list_filter = _promo_if_necessary + \
89             translated_fields(('published',)) + \
90             translated_fields(('needed',), app_settings.OPTIONAL_LANGUAGES)
91         inlines = (AttachmentInline,)
92         search_fields = ('title_pl', 'title_en')
93     return EntryAdmin
94
95
96 for typ in app_settings.TYPES:
97     newmodel = filtered_model("Entry_%s" % typ.db, Entry, 'type', typ.db, typ.slug)
98     admin.site.register(newmodel, filtered_entry_admin(typ))
99
100
101 if app_settings.TAXONOMIES:
102     from migdal.models import Category
103
104     class CategoryAdmin(admin.ModelAdmin):
105         list_display = translated_fields(('title', 'slug')) + ('taxonomy',)
106         prepopulated_fields = dict([
107                 ("slug_%s" % lang_code, ("title_%s" % lang_code,))
108                 for lang_code, lang_name in settings.LANGUAGES
109             ]) 
110     admin.site.register(Category, CategoryAdmin)