ebd14b9cb68681fbba1ace5fe7da9385496f5b2b
[wolnelektury.git] / src / catalogue / admin.py
1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
3 #
4 from django.contrib import admin
5 from catalogue.models import Tag, Book, Fragment, BookMedia, Collection, Source, Snippet
6 from pz.admin import EmptyFieldListFilter
7
8
9 class BlankFieldListFilter(EmptyFieldListFilter):
10     with_empty_str = True
11
12
13 class TagAdmin(admin.ModelAdmin):
14     list_display = ('name', 'slug', 'sort_key', 'category', 'has_description', 'occurrences')
15     list_filter = ('category',)
16     search_fields = ('name',)
17     ordering = ('name',)
18     readonly_fields = ('occurrences',)
19
20     def occurrences(self, tag):
21         return tag.items.count()
22     occurrences.short_description = 'Wystąpienia'
23
24     prepopulated_fields = {'slug': ('name',), 'sort_key': ('name',)}
25     radio_fields = {'category': admin.HORIZONTAL}
26
27
28 class MediaInline(admin.TabularInline):
29     model = BookMedia
30     readonly_fields = ['source_sha1']
31     extra = 0
32
33
34 class BookAdmin(admin.ModelAdmin):
35     list_display = (
36         'title', 'slug', 'created_at', 'has_epub_file', 'has_html_file', 'has_description',
37     )
38     list_filter = [
39             'print_on_demand',
40             ('wiki_link', BlankFieldListFilter),
41             ('parent', EmptyFieldListFilter),
42             ('media', admin.EmptyFieldListFilter),
43             ]
44     date_hierarchy = 'created_at'
45     search_fields = ('title',)
46     ordering = ('title',)
47
48     inlines = [MediaInline]
49
50
51 class FragmentAdmin(admin.ModelAdmin):
52     list_display = ('book', 'anchor',)
53     ordering = ('book', 'anchor',)
54
55
56 class CollectionAdmin(admin.ModelAdmin):
57     list_display = ('title', 'listed', 'role')
58     prepopulated_fields = {'slug': ('title',)}
59
60
61 class SourceAdmin(admin.ModelAdmin):
62     list_display = ('netloc', 'name')
63
64
65 admin.site.register(Tag, TagAdmin)
66 admin.site.register(Book, BookAdmin)
67 admin.site.register(Fragment, FragmentAdmin)
68 admin.site.register(Collection, CollectionAdmin)
69 admin.site.register(Source, SourceAdmin)
70
71
72 admin.site.register(Snippet)