"""Custom Page model(s)."""

from wagtail.wagtailadmin.edit_handlers import FieldPanel, MultiFieldPanel, StreamFieldPanel
from wagtail.wagtailcore.fields import StreamField
from wagtail.wagtailcore.models import Page
from wagtail.wagtailsearch import index

from .streamfields import *  # Replace * with the streamfields you want to use


class CustomPage(Page):
    """A custom page class."""

    template = 'templates/custom_page.html'
    content = StreamField([
        ('image_block', ImageBlock()),
        ('logo_parade', LogoParadeBlock()),
        ('content', RichTextBlock()),
    ], null=True)

    search_fields = Page.search_fields + [
        index.SearchField('content'),
    ]

    content_panels = [
        MultiFieldPanel([
            FieldPanel('title'),
            FieldPanel('subtitle'),
        ], 'Banner'),
        StreamFieldPanel('content'),
    ]