from django.db import models

from modelcluster.fields import ParentalKey

from wagtail.core.models import Page, Orderable
from wagtail.core.fields import RichTextField
from wagtail.admin.edit_handlers import FieldPanel, InlinePanel
from wagtail.images.edit_handlers import ImageChooserPanel
from wagtail.search import index


class Stranka(Page):
    obsah = RichTextField(blank=True, verbose_name="Obsah stránky")

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

    content_panels = Page.content_panels + [
        FieldPanel('obsah', classname="full"),
        InlinePanel('obrazky', label="Obrázky"),
    ]


class StrankaObrazek(Orderable):
    page = ParentalKey(Stranka, on_delete=models.CASCADE, related_name='obrazky')
    soubor = models.ForeignKey('wagtailimages.Image', on_delete=models.CASCADE, related_name='+', verbose_name='Obrázek')
    popisek = models.CharField(blank=True, max_length=250, verbose_name='Popisek')

    panels = [
        ImageChooserPanel('soubor'),
        FieldPanel('popisek'),
    ]