from wagtail.blocks import RichTextBlock

from base.blocks import StreamBlock
from wagtailcodeblock.blocks import CodeBlock

class BlogPostBlock(StreamBlock):
    paragraph = RichTextBlock(blank=True)
    code = CodeBlock(label='Code snippet')