Fix imports.
[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
6
7
8 class TagAdmin(admin.ModelAdmin):
9     list_display = ('name', 'slug', 'sort_key', 'category', 'has_description', 'occurrences')
10     list_filter = ('category',)
11     search_fields = ('name',)
12     ordering = ('name',)
13     readonly_fields = ('occurrences',)
14
15     def occurrences(self, tag):
16         return tag.items.count()
17     occurrences.short_description = 'Wystąpienia'
18
19     prepopulated_fields = {'slug': ('name',), 'sort_key': ('name',)}
20     radio_fields = {'category': admin.HORIZONTAL}
21
22
23 class MediaInline(admin.TabularInline):
24     model = BookMedia
25     readonly_fields = ['source_sha1']
26     extra = 0
27
28
29 class BookAdmin(admin.ModelAdmin):
30     list_display = (
31         'title', 'slug', 'created_at', 'has_epub_file', 'has_html_file', 'has_description',
32     )
33     list_filter = ['print_on_demand']
34     search_fields = ('title',)
35     ordering = ('title',)
36
37     inlines = [MediaInline]
38
39
40 class FragmentAdmin(admin.ModelAdmin):
41     list_display = ('book', 'anchor',)
42     ordering = ('book', 'anchor',)
43
44
45 class CollectionAdmin(admin.ModelAdmin):
46     list_display = ('title', 'listed')
47     prepopulated_fields = {'slug': ('title',)}
48
49
50 class SourceAdmin(admin.ModelAdmin):
51     list_display = ('netloc', 'name')
52
53
54 admin.site.register(Tag, TagAdmin)
55 admin.site.register(Book, BookAdmin)
56 admin.site.register(Fragment, FragmentAdmin)
57 admin.site.register(Collection, CollectionAdmin)
58 admin.site.register(Source, SourceAdmin)