46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
|
from re import Match
|
||
|
|
from textwrap import dedent
|
||
|
|
from typing import Any
|
||
|
|
|
||
|
|
from mistune import HTMLRenderer
|
||
|
|
from mistune.block_parser import BlockParser
|
||
|
|
from mistune.core import BlockState
|
||
|
|
from mistune.directives import DirectivePlugin, RSTDirective
|
||
|
|
from mistune.markdown import Markdown
|
||
|
|
|
||
|
|
|
||
|
|
class Column(DirectivePlugin):
|
||
|
|
def parse(
|
||
|
|
self, block: BlockParser, m: Match, state: BlockState
|
||
|
|
) -> dict[str, Any]:
|
||
|
|
info = m.groupdict()
|
||
|
|
|
||
|
|
new_state = block.state_cls()
|
||
|
|
new_state.process(dedent(info["text"]))
|
||
|
|
block.parse(new_state)
|
||
|
|
|
||
|
|
return {
|
||
|
|
"type": "column",
|
||
|
|
"text": info["text"],
|
||
|
|
"children": new_state.tokens,
|
||
|
|
"attrs": {},
|
||
|
|
}
|
||
|
|
|
||
|
|
def __call__( # type: ignore
|
||
|
|
self, directive: RSTDirective, md: Markdown
|
||
|
|
) -> None:
|
||
|
|
directive.register("column", self.parse)
|
||
|
|
|
||
|
|
if md.renderer.NAME == "html":
|
||
|
|
md.renderer.register("column", self._render_column)
|
||
|
|
|
||
|
|
def _render_column(self, renderer: HTMLRenderer, text: str, **attrs):
|
||
|
|
start = (
|
||
|
|
'<div class="columns mt-3 is-multiline">\n'
|
||
|
|
if attrs.get("first")
|
||
|
|
else ""
|
||
|
|
)
|
||
|
|
end = "</div>\n" if attrs.get("last") else ""
|
||
|
|
col = f'<div class="column is-half">{text}</div>\n'
|
||
|
|
return start + (col) + end
|