Merge pull request #328 from tatuylonen/find-html

find_html: allow list of tags to find.
tatuylonen · Oct 22, 2024 · d2151af · d2151af
2 parents a709d4b + 3400f75
commit d2151af
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/wikitextprocessor/parser.py b/src/wikitextprocessor/parser.py
@@ -496,7 +496,7 @@ def filter_empty_str_child(self) -> Iterator[Union[str, "WikiNode"]]:
     @overload
     def find_html(
         self,
-        target_tag: str,
+        target_tags: str | list[str],
         with_index: Literal[True],
         attr_name: str,
         attr_value: str,
@@ -505,15 +505,15 @@ def find_html(
     @overload
     def find_html(
         self,
-        target_tag: str,
+        target_tags: str | list[str],
         with_index: Literal[False] = ...,
         attr_name: str = ...,
         attr_value: str = ...,
     ) -> Iterator["HTMLNode"]: ...
 
     def find_html(
         self,
-        target_tag: str,
+        target_tags: str | list[str],
         with_index: bool = False,
         attr_name: str = "",
         attr_value: str = "",
@@ -523,7 +523,9 @@ def find_html(
             if TYPE_CHECKING:
                 assert isinstance(node, HTMLNode)
             # node.tag is an alias for node.sarg defined in HTMLNode
-            if node.tag == target_tag:
+            if isinstance(target_tags, str):
+                target_tags = [target_tags]
+            if node.tag in target_tags:
                 if len(attr_name) > 0 and attr_value not in node.attrs.get(
                     attr_name, {}
                 ):