X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/5c0ed2a6cb007fccf2cc4b58199f285707f7e974..4bb5959d246fa0528b83fc3e21f9e5782ec34236:/src/catalogue/admin.py

diff --git a/src/catalogue/admin.py b/src/catalogue/admin.py
index 189cd718..bd783b99 100644
--- a/src/catalogue/admin.py
+++ b/src/catalogue/admin.py
@@ -1,17 +1,21 @@
 from django.contrib import admin
 from . import models
+from .wikidata import WikidataAdminMixin
 
 
+class AuthorAdmin(WikidataAdminMixin, admin.ModelAdmin):
+    list_display = "first_name", "last_name", "notes"
+    search_fields = ["first_name", "last_name", "wikidata"]
+    prepopulated_fields = {"slug": ("first_name", "last_name")}
 
-class AuthorAdmin(admin.ModelAdmin):
-    search_fields = ['name']
 
 admin.site.register(models.Author, AuthorAdmin)
 
 
-class BookAdmin(admin.ModelAdmin):
-    raw_id_fields = ['authors']
-    autocomplete_fields = ['translators']
+class BookAdmin(WikidataAdminMixin, admin.ModelAdmin):
+    list_display = "title", "notes"
+    autocomplete_fields = ["authors", "translators"]
+    prepopulated_fields = {"slug": ("title",)}
 
-admin.site.register(models.Book, BookAdmin)
 
+admin.site.register(models.Book, BookAdmin)