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')


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

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