from wagtail.core import blocks
from wagtail.images.blocks import ImageChooserBlock

class ImageListBlock(blocks.StreamBlock):
    image = ImageChooserBlock()

    class Meta:
        template = "blocks/image_list.html"


class YellowStripBlock(blocks.StreamBlock):
    body = blocks.RichTextBlock()

    class Meta:
        template = "blocks/yellow_strip.html"


class ModalBlock(blocks.StructBlock):
    link_text = blocks.CharBlock()
    title = blocks.CharBlock()
    content = blocks.RichTextBlock()

    class Meta:
        template = "blocks/modal.html"


class NewsletterBlock(blocks.StructBlock):
    link_text = blocks.CharBlock()
    link_url = blocks.CharBlock()
    
    class Meta:
        template = "blocks/yellow_strip.html"