from django.contrib import admin
from .models import Book, BookHighlight


class BookAdmin(admin.ModelAdmin):
    list_display = ('title', 'admin_thumbnail', 'rating', 'isbn', 'author_name', 'read_date', 'publish_date')
    search_fields = ['title', 'body_markdown']
    list_filter = ('rating', 'read_date')

    class Media:
        js = ('next-prev-links.js',)

class BookHighlightAdmin(admin.ModelAdmin):
    list_display = ('book', 'page', 'date_added')

admin.site.register(Book, BookAdmin)
admin.site.register(BookHighlight, BookHighlightAdmin)