From 3224b3857cc2f11226043ced1da586756403cbb1 Mon Sep 17 00:00:00 2001 From: Gertjan van den Burg Date: Mon, 27 Apr 2020 17:26:42 +0100 Subject: Use builtin iter() function to find img elements --- paper2remarkable/providers/html.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/paper2remarkable/providers/html.py b/paper2remarkable/providers/html.py index bbafe10..ba250e7 100644 --- a/paper2remarkable/providers/html.py +++ b/paper2remarkable/providers/html.py @@ -61,16 +61,9 @@ class ImgProcessor(markdown.treeprocessors.Treeprocessor): self._base_url = base_url super().__init__(*args, **kwargs) - def _find_img(self, node): - """ Find img nodes recursively """ - for img in node.findall("img"): - yield img - for child in node: - yield from self._find_img(child) - def run(self, root): """ Ensure all img src urls are absolute """ - for img in self._find_img(root): + for img in root.iter("img"): img.attrib["src"] = urllib.parse.urljoin( self._base_url, img.attrib["src"] ) -- cgit v1.2.3