src/catalogue/admin.py

   1 # This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Wolne Lektury. 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     raw_id_fields = ['user']
  20
  21     def occurrences(self, tag):
  22         return tag.items.count()
  23     occurrences.short_description = 'Wystąpienia'
  24
  25     prepopulated_fields = {'slug': ('name',), 'sort_key': ('name',)}
  26     radio_fields = {'category': admin.HORIZONTAL}
  27
  28
  29 class MediaInline(admin.TabularInline):
  30     model = BookMedia
  31     readonly_fields = ['source_sha1']
  32     extra = 0
  33
  34
  35 class BookAdmin(admin.ModelAdmin):
  36     list_display = (
  37         'title', 'slug', 'created_at', 'has_epub_file', 'has_html_file', 'has_description',
  38     )
  39     list_filter = [
  40             'print_on_demand',
  41             ('wiki_link', BlankFieldListFilter),
  42             ('parent', EmptyFieldListFilter),
  43             ('media', admin.EmptyFieldListFilter),
  44             ]
  45     date_hierarchy = 'created_at'
  46     search_fields = ('title',)
  47     ordering = ('title',)
  48
  49     inlines = [MediaInline]
  50
  51
  52 class FragmentAdmin(admin.ModelAdmin):
  53     list_display = ('book', 'anchor',)
  54     ordering = ('book', 'anchor',)
  55
  56
  57 class CollectionAdmin(admin.ModelAdmin):
  58     list_display = ('title', 'listed', 'role')
  59     prepopulated_fields = {'slug': ('title',)}
  60
  61
  62 class SourceAdmin(admin.ModelAdmin):
  63     list_display = ('netloc', 'name')
  64
  65
  66 admin.site.register(Tag, TagAdmin)
  67 admin.site.register(Book, BookAdmin)
  68 admin.site.register(Fragment, FragmentAdmin)
  69 admin.site.register(Collection, CollectionAdmin)
  70 admin.site.register(Source, SourceAdmin)
  71
  72
  73 admin.site.register(Snippet)