add support for previewing pages

dnscms/venues/models.py

@@ -11,6 +11,7 @@ from wagtail.admin.panels import FieldPanel, FieldRowPanel, MultiFieldPanel
 from wagtail.fields import RichTextField, StreamField
 from wagtail.models import Page
 from wagtail.search import index
+from wagtail_headless_preview.models import HeadlessMixin
 
 from dnscms.blocks import ImageSliderBlock
 from dnscms.fields import CommonStreamField
@@ -18,7 +19,7 @@ from dnscms.wordpress.models import WPImportedPageMixin
 
 
 @register_singular_query_field("venueIndex")
-class VenueIndex(Page):
+class VenueIndex(HeadlessMixin, Page):
     # there can only be one venue index page
     max_count = 1
     subpage_types = ["venues.VenuePage"]
@@ -35,7 +36,7 @@ class VenueIndex(Page):
 
 
 @register_singular_query_field("venueRentalIndex")
-class VenueRentalIndex(Page):
+class VenueRentalIndex(HeadlessMixin, Page):
     # there can only be one venue index page
     max_count = 1
     subpage_types = []
@@ -51,7 +52,7 @@ class VenueRentalIndex(Page):
     graphql_fields = [GraphQLRichText("lead"), GraphQLStreamfield("body")]
 
 
-class VenuePage(WPImportedPageMixin, Page):
+class VenuePage(HeadlessMixin, WPImportedPageMixin, Page):
     # no children
     subpage_types = []
     parent_page_types = ["venues.VenueIndex"]