Restricting Page Hierarchy

Author: KalobTaulien

blog/models.py

@@ -154,6 +154,8 @@ class BlogListingPage(RoutablePageMixin, Page):
     """Listing page lists all the Blog Detail Pages."""
 
     template = "blog/blog_listing_page.html"
+    max_count = 1
+    subpage_types = ['blog.VideoBlogPage', 'blog.ArticleBlogPage']
 
     custom_title = models.CharField(
         max_length=100,
@@ -215,6 +217,9 @@ def get_sitemap_urls(self, request):
 class BlogDetailPage(Page):
     """Parental blog detail page."""
 
+    subpage_types = []
+    parent_page_types = ['blog.BlogListingPage']
+
     custom_title = models.CharField(
         max_length=100,
         blank=False,

contact/models.py

@@ -24,6 +24,9 @@ class FormField(AbstractFormField):
 class ContactPage(WagtailCaptchaEmailForm):
 
     template = "contact/contact_page.html"
+    subpage_types = []
+    parent_page_types = ['home.HomePage']
+
     # This is the default path.
     # If ignored, Wagtail adds _landing.html to your template name
     landing_page_template = "contact/contact_page_landing.html"

flex/models.py

@@ -12,7 +12,11 @@ class FlexPage(Page):
     """Flexible page class."""
 
     template = "flex/flex_page.html"
-
+    subpage_types = ['flex.FlexPage', 'contact.ContactPage']
+    parent_page_types = [
+        'flex.FlexPage',
+        'home.HomePage',
+    ]
     content = StreamField(
         [
             ("title_and_text", blocks.TitleAndTextBlock()),

home/models.py

@@ -43,7 +43,14 @@ class HomePage(RoutablePageMixin, Page):
     """Home page model."""
 
     template = "home/home_page.html"
-    max_count = 1
+    subpage_types = [
+        'blog.BlogListingPage',
+        'contact.ContactPage',
+        'flex.FlexPage',
+    ]
+    parent_page_type = [
+        'wagtailcore.Page'
+    ]
 
     banner_title = models.CharField(max_length=100, blank=False, null=True)
     banner_subtitle = RichTextField(features=["bold", "italic"])