Restore ability to provide a tree builder name as string (#559)

Co-authored-by: Łukasz Langa <lukasz@langa.pl>
html5lib · Mar 1, 2023 · c64a111 · c64a111
1 parent 1b39377
commit c64a111
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/html5lib/html5parser.py b/html5lib/html5parser.py
@@ -115,6 +115,9 @@ def __init__(self, tree=None, strict=False, namespaceHTMLElements=True, debug=Fa
 
         if tree is None:
             tree = treebuilders.getTreeBuilder("etree")
+        elif isinstance(tree, str):
+            tree = treebuilders.getTreeBuilder(tree)
+
         self.tree = tree(namespaceHTMLElements)
         self.errors = []