Skip to content

Instantly share code, notes, and snippets.

@cfc603

cfc603/blocks.py

Forked from frankwiles/blocks.py
Created September 24, 2017 15:57
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save cfc603/dd5f9c4917fa446e90b4a990e584df53 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save cfc603/dd5f9c4917fa446e90b4a990e584df53 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. @frankwiles frankwiles created this gist May 17, 2015.
    57 changes: 57 additions & 0 deletions blocks.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,57 @@
    from django.utils.safestring import mark_safe
    from markdown import markdown
    from pygments import highlight
    from pygments.formatters import get_formatter_by_name
    from pygments.lexers import get_lexer_by_name

    from wagtail.wagtailcore import blocks


    class CodeBlock(blocks.StructBlock):
    """
    Code Highlighting Block
    """
    LANGUAGE_CHOICES = (
    ('python', 'Python'),
    ('bash', 'Bash/Shell'),
    ('html', 'HTML'),
    ('css', 'CSS'),
    ('scss', 'SCSS'),
    )

    language = blocks.ChoiceBlock(choices=LANGUAGE_CHOICES)
    code = blocks.TextBlock()

    class Meta:
    icon = 'code'

    def render(self, value):
    src = value['code'].strip('\n')
    lang = value['language']

    lexer = get_lexer_by_name(lang)
    formatter = get_formatter_by_name(
    'html',
    linenos=None,
    cssclass='codehilite',
    style='default',
    noclasses=False,
    )
    return mark_safe(highlight(src, lexer, formatter))


    class MarkDownBlock(blocks.TextBlock):
    """ MarkDown Block """

    class Meta:
    icon = 'code'

    def render_basic(self, value):
    md = markdown(
    value,
    [
    'markdown.extensions.fenced_code',
    'codehilite',
    ],
    )
    return mark_safe(md)