WIP: feat: add parent attribute (#71)

Author: pbodnar

mistletoe/ast_renderer.py

@@ -40,7 +40,7 @@ def get_ast(token):
         node[attrname] = getattr(token, attrname)
     if 'header' in vars(token):
         node['header'] = get_ast(getattr(token, 'header'))
-    if 'children' in vars(token):
+    if token.children is not None:
         node['children'] = [get_ast(child) for child in token.children]
     return node
 

mistletoe/html_renderer.py

@@ -49,7 +49,7 @@ def __exit__(self, *args):
         super().__exit__(*args)
 
     def render_to_plain(self, token) -> str:
-        if hasattr(token, 'children'):
+        if token.children is not None:
             inner = [self.render_to_plain(child) for child in token.children]
             return ''.join(inner)
         return html.escape(token.content)

mistletoe/span_token.py

@@ -71,7 +71,7 @@ def __init__(self, match):
             self.content = match.group(self.parse_group)
 
     def __contains__(self, text):
-        if hasattr(self, 'children'):
+        if self.children is not None:
             return any(text in child for child in self.children)
         return text in self.content
 

mistletoe/token.py

@@ -54,7 +54,7 @@ def __repr__(self):
             self.__class__.__name__
         )
 
-        if "children" in vars(self):
+        if self.children is not None:
             count = len(self.children)
             if count == 1:
                 output += " with 1 child"
@@ -69,3 +69,18 @@ def __repr__(self):
             output += " {}={}".format(attrname, _short_repr(attrvalue))
         output += " at {:#x}>".format(id(self))
         return output
+    
+    @property
+    def parent(self):
+        return getattr(self, '_parent', None)
+
+    @property
+    def children(self):
+        return getattr(self, '_children', None)
+
+    @children.setter
+    def children(self, value):
+        self._children = value
+        if value:
+            for child in value:
+                child._parent = self

mistletoe/utils.py

@@ -20,14 +20,14 @@ def traverse(source, klass=None, depth=None, include_source=False):
     current_depth = 0
     if include_source and (klass is None or isinstance(source, klass)):
         yield TraverseResult(source, None, current_depth)
-    next_children = [(source, c) for c in getattr(source, 'children', [])]
+    next_children = [(source, c) for c in source.children or []]
     while next_children and (depth is None or current_depth < depth):
         current_depth += 1
         new_children = []
         for parent, child in next_children:
             if klass is None or isinstance(child, klass):
                 yield TraverseResult(child, parent, current_depth)
             new_children.extend(
-                [(child, c) for c in getattr(child, 'children', [])]
+                [(child, c) for c in child.children or []]
             )
         next_children = new_children

test/test_block_token.py

@@ -612,6 +612,18 @@ def test_contains(self):
         self.assertFalse('foo' in token)
 
 
+class TestParent(unittest.TestCase):
+    def test_parent(self):
+        lines = ['# heading\n', '\n', 'paragraph\n']
+        token = block_token.Document(lines)
+        self.assertEqual(len(token.children), 2)
+        self.assertIsNone(token.parent)
+        for child in token.children:
+            self.assertEqual(child.parent, token)
+            for grandchild in child.children:
+                self.assertEqual(grandchild.parent, child)
+
+
 class TestHtmlBlock(unittest.TestCase):
     def setUp(self):
         block_token.add_token(block_token.HtmlBlock)

test/test_span_token.py

@@ -151,8 +151,7 @@ def test_attribute(self):
 
     def test_no_children(self):
         token = span_token.RawText('some text')
-        with self.assertRaises(AttributeError):
-            token.children
+        self.assertIsNone(token.children)
 
     def test_valid_html_entities(self):
         tokens = span_token.tokenize_inner('  合')
@@ -189,6 +188,13 @@ def test_contains(self):
         self.assertFalse('foo' in token)
 
 
+class TestParent(unittest.TestCase):
+    def test_parent(self):
+        token, = span_token.tokenize_inner('**some text**')
+        self.assertIsInstance(token.children[0], span_token.RawText)
+        self.assertEqual(token.children[0].parent, token)
+
+
 class TestHtmlSpan(unittest.TestCase):
     def setUp(self):
         span_token.add_token(span_token.HtmlSpan)