Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Micro-optimization

Browse Source
This commit is contained in:
Kovid Goyal 2016-03-04 14:54:42 +05:30
parent 9d25d5ae50
commit 1a4b387038
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/calibre/ebooks/oeb/base.py
View File

@ -109,6 +109,8 @@ def itercsslinks(raw):
for match in _css_import_re.finditer(raw):
yield match.group(1), match.start(1)
_link_attrs = set(html.defs.link_attrs) | {XLINK('href'), 'poster'}
def iterlinks(root, find_links_in_css=True):
'''
Iterate over all links in a OEB Document.
@ -116,7 +118,6 @@ def iterlinks(root, find_links_in_css=True):
:param root: A valid lxml.etree element.
'''
assert etree.iselement(root)
link_attrs = set(html.defs.link_attrs) | {XLINK('href'), 'poster'}
for el in root.iter():
attribs = el.attrib
@ -146,7 +147,7 @@ def iterlinks(root, find_links_in_css=True):
yield (el, 'archive', value, match.start())
else:
for attr in attribs:
if attr in link_attrs:
if attr in _link_attrs:
yield (el, attr, attribs[attr], 0)
if not find_links_in_css: