from __future__ import unicode_literals from django.db import models from modelcluster.fields import ParentalKey from wagtail.wagtailcore import blocks from wagtail.wagtailimages import blocks as imageblocks from wagtail.wagtailcore.models import Page from wagtail.wagtailcore.models import Orderable from wagtail.wagtailcore.fields import RichTextField from wagtail.wagtailcore.fields import StreamField from wagtail.wagtailimages.edit_handlers import ImageChooserPanel from wagtail.wagtailadmin.edit_handlers import InlinePanel from wagtail.wagtailadmin.edit_handlers import FieldPanel from wagtail.wagtailadmin.edit_handlers import PageChooserPanel from wagtail.wagtailadmin.edit_handlers import StreamFieldPanel from wagtail.wagtailsearch import index ILLUSTRATION_ANTARCTICA = "antarctica.svg" ILLUSTRATION_BRIDGE = "bridge.svg" ILLUSTRATION_CASINO = "casino.svg" ILLUSTRATION_CRADLE = "cradle.svg" ILLUSTRATION_DEVIL = "devil.svg" ILLUSTRATION_FALLS = "falls.svg" ILLUSTRATION_HOBART = "hobart.svg" ILLUSTRATION_LAVENDER = "lavender.svg" ILLUSTRATION_TUZ = "tuz.svg" ILLUSTRATION_WINEGLASS = "wineglass.svg" ILLUSTRATION_TYPES = ( (ILLUSTRATION_ANTARCTICA, "Antarctica"), (ILLUSTRATION_BRIDGE, "Bridge"), (ILLUSTRATION_CASINO, "Casino"), (ILLUSTRATION_CRADLE, "Cradle Mountain"), (ILLUSTRATION_DEVIL, "Tasmanian Devil"), (ILLUSTRATION_FALLS, "Waterfall"), (ILLUSTRATION_HOBART, "Hobart"), (ILLUSTRATION_LAVENDER, "Lavender"), (ILLUSTRATION_TUZ, "Tuz"), (ILLUSTRATION_WINEGLASS, "Wineglass"), ) class BasicContentBlock(blocks.StructBlock): class Meta: template = "cms_pages/home_page_blocks/basic_content.html" PANEL_BLUE_LEFT = "blue_left" PANEL_WHITE_RIGHT = "white_right" PANEL_TYPES = ( (PANEL_BLUE_LEFT, "Left-aligned image, blue-filtered image BG"), (PANEL_WHITE_RIGHT, "Right-aligned image, white background"), ) EXTERNAL_LINK_TWITTER = "twitter" EXTERNAL_LINK_GENERIC = "generic" EXTERNAL_LINK_TYPES = ( (EXTERNAL_LINK_TWITTER, "Twitter"), (EXTERNAL_LINK_GENERIC, "Generic URL"), ) panel_type = blocks.ChoiceBlock( choices=PANEL_TYPES, required=True, ) heading = blocks.CharBlock(required=True) inset_illustration = blocks.ChoiceBlock( choices=ILLUSTRATION_TYPES, required=True, ) background_image = imageblocks.ImageChooserBlock( required=False, help_text="This is used as the background image of a " "blue-left block. It's not used for white-right." ) body = blocks.RichTextBlock(required=True) link = blocks.StructBlock([ ("page", blocks.PageChooserBlock()), ("title", blocks.CharBlock(required=True)), ]) external_links = blocks.ListBlock( blocks.StructBlock([ ("alt", blocks.CharBlock(required=True)), ("icon", blocks.ChoiceBlock( choices=EXTERNAL_LINK_TYPES, required=True, )), ("url", blocks.URLBlock(required=True) )]) ) class HomePage(Page): body = StreamField([ ("basic_content", BasicContentBlock()), # TODO: keynotes # TODO: other bits ]) content_panels = Page.content_panels + [ StreamFieldPanel('body') ] # Content pages class AbstractContentPage(Page): class Meta: abstract = True intro = models.CharField(max_length=250) body = RichTextField(blank=True) background_image = models.ForeignKey( 'wagtailimages.Image', null=True, blank=True, on_delete=models.SET_NULL, related_name='+' ) search_fields = Page.search_fields + [ index.SearchField('intro'), index.SearchField('body'), ] content_panels = Page.content_panels + [ ImageChooserPanel('background_image'), FieldPanel('intro'), FieldPanel('body', classname="full") ] class ContentPage(AbstractContentPage): inset_illustration = models.CharField( choices=ILLUSTRATION_TYPES, max_length=256, ) content_panels = AbstractContentPage.content_panels + [ FieldPanel('inset_illustration') ] # News pages class NewsIndexPage(AbstractContentPage): def child_pages(self): return NewsPage.objects.live().child_of(self).specific().order_by("-date") subpage_types = [ "NewsPage", ] content_panels = AbstractContentPage.content_panels class NewsPage(AbstractContentPage): date = models.DateField("Post date") portrait_image = models.ForeignKey( 'wagtailimages.Image', null=True, blank=True, on_delete=models.SET_NULL, related_name='+' ) parent_page_types = [ NewsIndexPage, ] content_panels = AbstractContentPage.content_panels + [ FieldPanel('date'), ImageChooserPanel('portrait_image'), ]