소스 검색

Register Country model with a ModelViewSet

This serves as an example of how ModelViewSet can be mixed with snippets
in a SnippetViewSetGroup
Sage Abdullah 1 년 전
부모
커밋
7368667f96
1개의 변경된 파일9개의 추가작업 그리고 2개의 파일을 삭제
  1. 9 2
      bakerydemo/breads/wagtail_hooks.py

+ 9 - 2
bakerydemo/breads/wagtail_hooks.py

@@ -1,4 +1,6 @@
 from wagtail.admin.filters import WagtailFilterSet
+from wagtail.admin.panels import FieldPanel
+from wagtail.admin.viewsets.model import ModelViewSet
 from wagtail.snippets.models import register_snippet
 from wagtail.snippets.views.snippets import SnippetViewSet, SnippetViewSetGroup
 
@@ -35,10 +37,15 @@ class BreadTypeSnippetViewSet(SnippetViewSet):
     filterset_class = BreadTypeFilterSet
 
 
-class CountrySnippetViewSet(SnippetViewSet):
+class CountryModelViewSet(ModelViewSet):
     model = Country
     ordering = ("title",)
     search_fields = ("title",)
+    icon = "globe"
+
+    panels = [
+        FieldPanel("title"),
+    ]
 
 
 # We want to group several snippets together in the admin menu.
@@ -57,7 +64,7 @@ class BreadMenuGroup(SnippetViewSetGroup):
     items = (
         BreadIngredientSnippetViewSet,
         BreadTypeSnippetViewSet,
-        CountrySnippetViewSet,
+        CountryModelViewSet,
     )