fix: handle embedded lists at the start of other lists

vzhd1701 · Dec 14, 2021 · f3b83ac · f3b83ac
1 parent 66ceb28
commit f3b83ac
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 0 deletions.
diff --git a/enex2notion/note_parser_elements.py b/enex2notion/note_parser_elements.py
@@ -34,6 +34,9 @@ def parse_list(element: Tag):
             nodes.append(li_item)
 
         elif subelement.name in {"ul", "ol"}:
+            if not nodes:
+                nodes.append(_make_blank_node(is_ul))
+
             nodes[-1].children.extend(parse_list(subelement))
 
         else:
@@ -46,6 +49,13 @@ def parse_list(element: Tag):
     return nodes
 
 
+def _make_blank_node(is_ul):
+    if is_ul:
+        return NotionBulletedListBlock(text_prop=TextProp(text=""))
+
+    return NotionNumberedListBlock(text_prop=TextProp(text=""))
+
+
 def _parse_odd_item(element: Tag):
     if isinstance(element, NavigableString):
         if not element.text.strip():

diff --git a/tests/test_note_parser.py b/tests/test_note_parser.py
@@ -67,6 +67,34 @@ def test_list_ul():
     ]
 
 
+def test_list_ul_ul():
+    test_note = parse_html(
+        "<ul><ul><li><div>test_sub</div></li></ul><li><div>test</div></li></ul>"
+    )
+
+    expected = [
+        NotionBulletedListBlock(text_prop=TextProp(text="")),
+        NotionBulletedListBlock(text_prop=TextProp("test")),
+    ]
+    expected[0].children = [NotionBulletedListBlock(text_prop=TextProp("test_sub"))]
+
+    assert parse_note_dom(test_note) == expected
+
+
+def test_list_ol_ol():
+    test_note = parse_html(
+        "<ol><ol><li><div>test_sub</div></li></ol><li><div>test</div></li></ol>"
+    )
+
+    expected = [
+        NotionNumberedListBlock(text_prop=TextProp(text="")),
+        NotionNumberedListBlock(text_prop=TextProp("test")),
+    ]
+    expected[0].children = [NotionNumberedListBlock(text_prop=TextProp("test_sub"))]
+
+    assert parse_note_dom(test_note) == expected
+
+
 def test_list_ul_todo():
     test_note = parse_html(
         (