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 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             ]
  43     search_fields = ('title',)
  44     ordering = ('title',)
  45
  46     inlines = [MediaInline]
  47
  48
  49 class FragmentAdmin(admin.ModelAdmin):
  50     list_display = ('book', 'anchor',)
  51     ordering = ('book', 'anchor',)
  52
  53
  54 class CollectionAdmin(admin.ModelAdmin):
  55     list_display = ('title', 'listed')
  56     prepopulated_fields = {'slug': ('title',)}
  57
  58
  59 class SourceAdmin(admin.ModelAdmin):
  60     list_display = ('netloc', 'name')
  61
  62
  63 admin.site.register(Tag, TagAdmin)
  64 admin.site.register(Book, BookAdmin)
  65 admin.site.register(Fragment, FragmentAdmin)
  66 admin.site.register(Collection, CollectionAdmin)
  67 admin.site.register(Source, SourceAdmin)