diff --git a/mwparserfromhell/nodes/template.py b/mwparserfromhell/nodes/template.py index 426a8d9..8e01d7a 100644 --- a/mwparserfromhell/nodes/template.py +++ b/mwparserfromhell/nodes/template.py @@ -70,7 +70,7 @@ class Template(Node): name, value = parse_anything(name), parse_anything(value) surface_text = value.filter_text(recursive=False) for node in surface_text: - node.replace("|", "|") # INSERT AS HTMLEntity INSTEAD OF RAW TEXT + value.replace(node, node.replace("|", "|")) if showkey is None: if any(["=" in node for node in surface_text]): @@ -84,7 +84,7 @@ class Template(Node): showkey = True elif not showkey: for node in surface_text: - node.replace("=", "|") # INSERT AS HTMLEntity INSTEAD OF RAW TEXT + value.replace(node, node.replace("=", "|")) if self.has_param(name): self.remove_param(name, keep_field=True) diff --git a/mwparserfromhell/nodes/text.py b/mwparserfromhell/nodes/text.py index 2d5079b..afdc235 100644 --- a/mwparserfromhell/nodes/text.py +++ b/mwparserfromhell/nodes/text.py @@ -34,6 +34,3 @@ class Text(Node): @property def value(self): return self._value - - def replace(self, old, new, count): # TODO - pass diff --git a/mwparserfromhell/string_mixin.py b/mwparserfromhell/string_mixin.py index 74ae42f..cc0a63b 100644 --- a/mwparserfromhell/string_mixin.py +++ b/mwparserfromhell/string_mixin.py @@ -79,3 +79,6 @@ class StringMixIn(object): if isinstance(item, StringMixIn): return unicode(item) in unicode(self) return item in unicode(self) + + def replace(self, old, new, count): + return unicode(self).replace(old, new, count)