print([x for x in range(1, 5)])'
11 | in test_page.body.render_as_block()
12 | )
13 |
--------------------------------------------------------------------------------
/tests/urls.py:
--------------------------------------------------------------------------------
1 | from django.urls import include, re_path
2 |
3 | from wagtail.admin import urls as wagtailadmin_urls
4 | from wagtail.documents import urls as wagtaildocs_urls
5 | from wagtail import urls as wagtail_urls
6 |
7 | urlpatterns = [
8 | re_path(r"^cms/", include(wagtailadmin_urls)),
9 | re_path(r"^documents/", include(wagtaildocs_urls)),
10 | re_path(r"", include(wagtail_urls)),
11 | ]
12 |
--------------------------------------------------------------------------------
/wagtailcodeblock/__init__.py:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/FlipperPA/wagtailcodeblock/11aaf676b236c22bceac8c9cf778a1f55003ccb9/wagtailcodeblock/__init__.py
--------------------------------------------------------------------------------
/wagtailcodeblock/blocks.py:
--------------------------------------------------------------------------------
1 | import wagtail
2 |
3 | from django.forms import Media
4 | from django.utils.functional import cached_property
5 | from django.utils.translation import gettext_lazy as _
6 |
7 | from wagtail.blocks import (
8 | StructBlock,
9 | TextBlock,
10 | ChoiceBlock,
11 | )
12 | from wagtail.blocks.struct_block import StructBlockAdapter
13 | from wagtail.telepath import register
14 |
15 | from .settings import get_language_choices
16 |
17 |
18 | class CodeBlock(StructBlock):
19 | """
20 | A Wagtail StreamField block for code syntax highlighting using PrismJS.
21 | """
22 |
23 | def __init__(self, local_blocks=None, **kwargs):
24 | # Languages included in PrismJS core
25 | # Review: https://github.com/PrismJS/prism/blob/gh-pages/prism.js#L602
26 | self.INCLUDED_LANGUAGES = (
27 | ("html", "HTML"),
28 | ("mathml", "MathML"),
29 | ("svg", "SVG"),
30 | ("xml", "XML"),
31 | )
32 |
33 | if local_blocks is None:
34 | local_blocks = []
35 | else:
36 | local_blocks = local_blocks.copy()
37 |
38 | language_choices, language_default = self.get_language_choice_list(**kwargs)
39 |
40 | local_blocks.extend(
41 | [
42 | (
43 | "language",
44 | ChoiceBlock(
45 | choices=language_choices,
46 | help_text=_("Coding language"),
47 | label=_("Language"),
48 | default=language_default,
49 | identifier="language",
50 | ),
51 | ),
52 | ("code", TextBlock(label=_("Code"), identifier="code")),
53 | ]
54 | )
55 |
56 | super().__init__(local_blocks, **kwargs)
57 |
58 | def get_language_choice_list(self, **kwargs):
59 | # Get default languages
60 | WCB_LANGUAGES = get_language_choices()
61 | # If a language is passed in as part of a code block, use it.
62 | language = kwargs.get("language", False)
63 |
64 | total_language_choices = WCB_LANGUAGES + self.INCLUDED_LANGUAGES
65 |
66 | if language in [lang[0] for lang in total_language_choices]:
67 | for language_choice in total_language_choices:
68 | if language_choice[0] == language:
69 | language_choices = (language_choice,)
70 | language_default = language_choice[0]
71 | else:
72 | language_choices = WCB_LANGUAGES
73 | language_default = kwargs.get("default_language")
74 |
75 | return language_choices, language_default
76 |
77 | class Meta:
78 | icon = "code"
79 | template = "wagtailcodeblock/code_block.html"
80 | form_classname = "code-block struct-block"
81 | form_template = "wagtailcodeblock/code_block_form.html"
82 |
83 |
84 | class CodeBlockAdapter(StructBlockAdapter):
85 | js_constructor = "wagtailcodeblock.blocks.CodeBlock"
86 |
87 | @cached_property
88 | def media(self):
89 | structblock_media = super().media
90 | return Media(
91 | js=structblock_media._js + ["wagtailcodeblock/js/wagtailcodeblock.js"],
92 | css=structblock_media._css,
93 | )
94 |
95 |
96 | register(CodeBlockAdapter(), CodeBlock)
97 |
--------------------------------------------------------------------------------
/wagtailcodeblock/settings.py:
--------------------------------------------------------------------------------
1 | from django.conf import settings
2 |
3 | PRISM_PREFIX = "//cdnjs.cloudflare.com/ajax/libs/prism/"
4 | PRISM_VERSION = "1.29.0"
5 |
6 |
7 | def get_language_choices():
8 | """
9 | Default list of language choices, if not overridden by Django.
10 | """
11 | DEFAULT_LANGUAGES = (
12 | ("bash", "Bash/Shell"),
13 | ("css", "CSS"),
14 | ("diff", "diff"),
15 | ("html", "HTML"),
16 | ("javascript", "Javascript"),
17 | ("json", "JSON"),
18 | ("python", "Python"),
19 | ("scss", "SCSS"),
20 | ("yaml", "YAML"),
21 | )
22 |
23 | return getattr(settings, "WAGTAIL_CODE_BLOCK_LANGUAGES", DEFAULT_LANGUAGES)
24 |
25 |
26 | def get_theme():
27 | """
28 | Returns a default theme, if not in the proejct's settings. Default theme is 'coy'.
29 | """
30 |
31 | return getattr(settings, "WAGTAIL_CODE_BLOCK_THEME", "coy")
32 |
33 |
34 | def get_line_numbers():
35 | """
36 | Returns the line numbers setting.
37 | """
38 |
39 | return getattr(settings, "WAGTAIL_CODE_BLOCK_LINE_NUMBERS", True)
40 |
41 |
42 | def get_copy_to_clipboard():
43 | """
44 | Returns the copy to clipboard setting.
45 | """
46 |
47 | return getattr(settings, "WAGTAIL_CODE_BLOCK_COPY_TO_CLIPBOARD", True)
48 |
--------------------------------------------------------------------------------
/wagtailcodeblock/static/wagtailcodeblock/css/wagtail-code-block.css:
--------------------------------------------------------------------------------
1 | .token.tag:before {
2 | content: none !important;
3 | }
4 |
5 | .token.tag {
6 | background-color: #fff !important;
7 | padding: 0 !important;
8 | }
9 |
10 | .code-block textarea {
11 | font-family: FreeMono, monospace;
12 | }
13 |
14 | .code-block code {
15 | display: block;
16 | }
17 |
--------------------------------------------------------------------------------
/wagtailcodeblock/static/wagtailcodeblock/css/wagtail-code-block.min.css:
--------------------------------------------------------------------------------
1 | .token.tag:before{content:none!important}.token.tag{background-color:#fff!important;padding:0!important}.code-block textarea {font-family:FreeMono,monospace;}.code-block code{display:block;}
--------------------------------------------------------------------------------
/wagtailcodeblock/static/wagtailcodeblock/js/wagtailcodeblock.js:
--------------------------------------------------------------------------------
1 | class CodeBlockDefinition extends window.wagtailStreamField.blocks
2 | .StructBlockDefinition {
3 | render(placeholder, prefix, initialState, initialError) {
4 | const block = super.render(
5 | placeholder,
6 | prefix,
7 | initialState,
8 | initialError,
9 | );
10 |
11 | var languageField = $(document).find('#' + prefix + '-language');
12 | var codeField = $(document).find('#' + prefix + '-code');
13 | var targetField = $(document).find('#' + prefix + '-target');
14 |
15 | function updateLanguage() {
16 | var languageCode = languageField.val();
17 | targetField.removeClass().addClass('language-' + languageCode);
18 | prismRepaint();
19 | }
20 |
21 | function prismRepaint() {
22 | Prism.highlightElement(targetField[0]);
23 | }
24 |
25 | function populateTargetCode() {
26 | var codeText = codeField.val();
27 | targetField.text(codeText);
28 | prismRepaint(targetField);
29 | }
30 |
31 | updateLanguage();
32 | populateTargetCode();
33 | languageField.on('change', updateLanguage);
34 | codeField.on('keyup', populateTargetCode);
35 |
36 | return block;
37 | }
38 | }
39 |
40 | window.telepath.register('wagtailcodeblock.blocks.CodeBlock', CodeBlockDefinition);
41 |
--------------------------------------------------------------------------------
/wagtailcodeblock/templates/wagtailcodeblock/code_block.html:
--------------------------------------------------------------------------------
1 | {% load static wagtailcodeblock_tags %}
2 | {% spaceless %}
3 | {% load_prism_css %}
4 | {% for key, val in self.items %}
5 | {% if key == "language" %}
6 |
48 | {% endif %}
49 | {% if key == "code" %}
50 |
51 | {{ val }}
52 |
53 |
59 | {% endif %}
60 | {% endfor %}
61 | {% endspaceless %}
62 |
--------------------------------------------------------------------------------
/wagtailcodeblock/templates/wagtailcodeblock/code_block_form.html:
--------------------------------------------------------------------------------
1 | {% load wagtailadmin_tags %}
2 |
3 |