diff options
Diffstat (limited to 'htmtl')
| -rw-r--r-- | htmtl/attribute_parsers/inner_text.py | 8 | ||||
| -rw-r--r-- | htmtl/attribute_parsers/outer_text.py | 8 |
2 files changed, 12 insertions, 4 deletions
diff --git a/htmtl/attribute_parsers/inner_text.py b/htmtl/attribute_parsers/inner_text.py index 9195802..75ad01e 100644 --- a/htmtl/attribute_parsers/inner_text.py +++ b/htmtl/attribute_parsers/inner_text.py @@ -1,9 +1,13 @@ +from typing import Optional + from dompa.nodes import Node, TextNode from ..attribute_parser import AttributeParser class InnerText(AttributeParser): - def traverse(self, node: Node): + def traverse(self, node: Node) -> Optional[Node]: if "inner-text" in node.attributes: node.children = [TextNode(value=self.expression(node.attributes["inner-text"]))] - node.attributes.pop("inner-text")
\ No newline at end of file + node.attributes.pop("inner-text") + + return node
\ No newline at end of file diff --git a/htmtl/attribute_parsers/outer_text.py b/htmtl/attribute_parsers/outer_text.py index 8d89b14..8e9bead 100644 --- a/htmtl/attribute_parsers/outer_text.py +++ b/htmtl/attribute_parsers/outer_text.py @@ -1,8 +1,12 @@ +from typing import Optional + from dompa.nodes import Node, TextNode from ..attribute_parser import AttributeParser class OuterText(AttributeParser): - def traverse(self, node: Node): + def traverse(self, node: Node) -> Optional[Node]: if "outer-text" in node.attributes: - node.replace_with(TextNode(value=self.expression(node.attributes["outer-text"]))) + return TextNode(value=self.expression(node.attributes["outer-text"])) + + return node |
