1 files changed, 29 insertions, 1 deletions
diff --git a/app/books/admin.py b/app/books/admin.py
index f488fe5..f0303a4 100644
--- a/app/books/admin.py
+++ b/app/books/admin.py
@@ -1,17 +1,45 @@
 from django.contrib import admin
 from .models import Book, BookHighlight
 
+class BookHighlightInline(admin.StackedInline):
+    model = BookHighlight
+    extra = 0
+    fieldsets = (
+        (None, {
+            'fields': (
+                'book',
+                'body_markdown',
+                'page',
+                'is_public',
+            )
+        }),
+    )
+
 
 @admin.register(Book)
 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')
+    inlines = [BookHighlightInline,]
 
     class Media:
         js = ('next-prev-links.js',)
 
-
 @admin.register(BookHighlight)
 class BookHighlightAdmin(admin.ModelAdmin):
     list_display = ('book', 'page', 'date_highlighted')
+
+    fieldsets = (
+        (None, {
+            'fields': (
+                'book',
+                'body_markdown',
+                'is_public',
+                'page',
+                'date_highlighted',
+            )
+        }),
+    )
+    class Media:
+        js = ('next-prev-links.js',)