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