def __init__(self):
self.renderer = MarkdownRenderer()
# Nested lists.
self._lists = []
def __init__(self):
self.renderer = MarkdownRenderer()
# Nested lists.
self._lists = []
class ASTToMarkdown(TreeTransformer):
"""Read an AST and render a Markdown string."""
def __init__(self):
self.renderer = MarkdownRenderer()
# Nested lists.
self._lists = []
def get_inner_contents(self, node):
delim = ''
# What is the delimiter between children? If the children are
# blocks, we should insert a new line between consecutive blocks.
# Otherwise we just concatenate the children.
if node.children:
child = node.children[0]
if isinstance(child, ASTNode) and child.is_block():
delim = '\n\n'
return delim.join(self.transform_children(node))
def transform_str(self, text):
return text
def transform_Node(self, node):
return self.get_inner_contents(node)
# Block nodes
# -------------------------------------------------------------------------
def transform_Plain(self, node):
return self.renderer.text(self.get_inner_contents(node))
def transform_Para(self, node):
return self.transform_Plain(node)
def transform_Header(self, node):
return self.renderer.heading(self.get_inner_contents(node),
level=node.level)
def transform_CodeBlock(self, node):
return self.renderer.code(self.get_inner_contents(node),
lang=node.lang)
def transform_BlockQuote(self, node):
return self.renderer.quote(self.get_inner_contents(node))
def transform_MathBlock(self, node):
return self.renderer.math_block(self.get_inner_contents(node))
def _write_list(self, node, list_type):
assert list_type in ('bullet', 'ordered')
if list_type == 'bullet':
bullet = node.bullet_char
suffix = node.delimiter
elif list_type == 'ordered':
bullet = node.start
# NOTE: CommonMark doesn't appear to support non-decimal bullets
# node.style is not used
suffix = ')' if node.delimiter == 'OneParen' else '.'
if not suffix.endswith(' '):
suffix += ' '
# This is a list of processed items.
items = self.transform_children(node)
out = []
for item in items:
# We indent all lines in the item.
item_lines = item.splitlines()
item = '\n'.join(((' ' if i else '') + line)
for i, line in enumerate(item_lines))
# We add the bullet and suffix to the first line in the item.
out.append(str(bullet) + suffix + item)
# We increase the current ordered list number.
if list_type == 'ordered':
bullet += 1
return '\n'.join(out)
def transform_BulletList(self, node):
return self._write_list(node, 'bullet')
def transform_OrderedList(self, node):
return self._write_list(node, 'ordered')
def transform_ListItem(self, node):
out = self.get_inner_contents(node)
return out
# Inline nodes
# -------------------------------------------------------------------------
def transform_Emph(self, node):
return self.renderer.emph(self.get_inner_contents(node))
def transform_Strong(self, node):
return self.renderer.strong(self.get_inner_contents(node))
def transform_Code(self, node):
return self.renderer.inline_code(
self.get_inner_contents(node))
def transform_LineBreak(self, node):
return self.renderer.linebreak()
def transform_Math(self, node):
return self.renderer.math(self.get_inner_contents(node))
def transform_Link(self, node):
return self.renderer.link(self.get_inner_contents(node),
node.url)
def transform_Image(self, node):
return self.renderer.image(self.get_inner_contents(node),
node.url)
class ASTToMarkdown(TreeTransformer):
"""Read an AST and render a Markdown string."""
def __init__(self):
self.renderer = MarkdownRenderer()
# Nested lists.
self._lists = []
def fold(self, transformed_children, node=None):
delim = '\n\n' if node.name == 'root' else ''
return delim.join(transformed_children)
def get_inner_contents(self, node):
return self.fold(self.transform_children(node), node)
def transform_str(self, text):
return text
def transform_Node(self, node):
return self.get_inner_contents(node)
# Block nodes
# -------------------------------------------------------------------------
def transform_Plain(self, node):
return self.renderer.text(self.get_inner_contents(node))
def transform_Para(self, node):
return self.transform_Plain(node)
def transform_Header(self, node):
return self.renderer.heading(self.get_inner_contents(node),
level=node.level)
def transform_CodeBlock(self, node):
return self.renderer.code(self.get_inner_contents(node),
lang=node.lang)
def transform_BlockQuote(self, node):
return self.renderer.quote(self.get_inner_contents(node))
def transform_MathBlock(self, node):
return self.renderer.math_block(self.get_inner_contents(node))
def _write_list(self, node, list_type):
assert list_type in ('bullet', 'ordered')
if list_type == 'bullet':
bullet = node.bullet_char
suffix = node.delimiter
elif list_type == 'ordered':
bullet = node.start
# NOTE: CommonMark doesn't appear to support non-decimal bullets
# node.style is not used
suffix = ')' if node.delimiter == 'OneParen' else '.'
if not suffix.endswith(' '):
suffix += ' '
# This is a list of processed items.
items = self.transform_children(node)
out = []
for item in items:
# We indent all lines in the item.
item_lines = item.splitlines()
item = '\n'.join(((' ' if i else '') + line)
for i, line in enumerate(item_lines))
# We add the bullet and suffix to the first line in the item.
out.append(str(bullet) + suffix + item)
# We increase the current ordered list number.
if list_type == 'ordered':
bullet += 1
return '\n'.join(out)
def transform_BulletList(self, node):
return self._write_list(node, 'bullet')
def transform_OrderedList(self, node):
return self._write_list(node, 'ordered')
def transform_ListItem(self, node):
out = self.get_inner_contents(node)
return out
# Inline nodes
# -------------------------------------------------------------------------
def transform_Emph(self, node):
return self.renderer.emph(self.get_inner_contents(node))
def transform_Strong(self, node):
return self.renderer.strong(self.get_inner_contents(node))
def transform_Code(self, node):
return self.renderer.inline_code(
self.get_inner_contents(node))
def transform_LineBreak(self, node):
return self.renderer.linebreak()
def transform_Math(self, node):
return self.renderer.math(self.get_inner_contents(node))
def transform_Link(self, node):
return self.renderer.link(self.get_inner_contents(node),
node.url)
def transform_Image(self, node):
return self.renderer.image(self.get_inner_contents(node),
node.url)
