Permalink
Switch branches/tags
Nothing to show
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
37 lines (28 sloc) 1.11 KB
from django.db import models
from django.http import Http404
from wagtail.core.models import Page
from wagtail.core.fields import RichTextField
from wagtail.admin.edit_handlers import FieldPanel
class HomePage(Page):
body = RichTextField(blank=True)
def route(self, request, path_components):
# Check the request is for AMP page
is_amp_request = False
if path_components and path_components[0] == 'amp':
is_amp_request = True
# Remove the amp from path components to check if the page exist
path_components = path_components[1:]
page, args, kwargs = super(HomePage, self).route(
request, path_components
)
if is_amp_request:
# If the page has amp template serve it otherwise raise 404
if hasattr(page, 'get_template_amp'):
kwargs['is_amp_request'] = is_amp_request
else:
raise Http404
return page, args, kwargs
content_panels = Page.content_panels + [
FieldPanel('body', classname="full"),
]
parent_page_types = ['wagtailcore.Page']