frankwiles · May 17, 2015 03:13
diff --git a/blocks.py b/blocks.py
 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)
	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)