Restricting Page Hierarchy

CodingForEverybody · Mar 25, 2019 · c4cf47b · c4cf47b · KalobTaulien · Apr 4, 2019
1 parent ec1b75b
commit c4cf47b
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 2 deletions.
diff --git a/blog/models.py b/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,

diff --git a/contact/models.py b/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"

diff --git a/flex/models.py b/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()),

diff --git a/home/models.py b/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"])