RFC: html: add parse_html

Signed-off-by: Florian Scherf <[email protected]>
lona-web-org · Jul 7, 2023 · b8ddd67 · b8ddd67
1 parent d0b50b9
commit b8ddd67
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/lona/html/parsing.py b/lona/html/parsing.py
@@ -198,8 +198,12 @@ def handle_endtag(self, tag):
         self.set_current_node(self._node.parent)
 
 
-def html_string_to_node_list(html_string, use_high_level_nodes=True,
-                             node_classes=None):
+def parse_html(
+        html_string: str,
+        use_high_level_nodes: bool = True,
+        node_classes: Dict[str, AbstractNode] | None = None,
+        flat: bool = True,
+) -> AbstractNode | List[AbstractNode]:
 
     root_node = Node()
     nodes = []
@@ -221,6 +225,9 @@ def html_string_to_node_list(html_string, use_high_level_nodes=True,
         node.remove()
         nodes.append(node)
 
+    if flat and len(nodes) == 1:
+        return nodes[0]
+
     return nodes
 
 
@@ -230,6 +237,8 @@ def HTML(
         node_classes: Dict[str, AbstractNode] | None = None,
 ) -> AbstractNode:
 
+    # TODO: remove HTML parsing in 2.0
+
     _nodes: List[AbstractNode] = []
 
     for node in nodes:
@@ -243,10 +252,11 @@ def HTML(
 
             # html string
             elif '<' in node or '>' in node:
-                parsed_nodes = html_string_to_node_list(
+                parsed_nodes = parse_html(
                     html_string=node,
                     use_high_level_nodes=use_high_level_nodes,
                     node_classes=node_classes or {},
+                    flat=False,
                 )
 
                 if len(nodes) > 1:

diff --git a/lona/html/widgets.py b/lona/html/widgets.py
@@ -1,4 +1,4 @@
-from lona.html.parsing import html_string_to_node_list
+from lona.html.parsing import parse_html
 from lona.html.text_node import TextNode
 from lona.html.widget import Widget
 
@@ -22,10 +22,11 @@ def __init__(self, *nodes, use_high_level_nodes=True, node_classes=None):
                         self.nodes.append(HTML(node))
 
                     else:
-                        self.nodes = html_string_to_node_list(
+                        self.nodes = parse_html(
                             html_string=node,
                             use_high_level_nodes=use_high_level_nodes,
                             node_classes=node_classes or {},
+                            flat=False,
                         )
 
                 else: