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

Allow passing ElementTree to Select()

Browse Source
This commit is contained in:
Kovid Goyal 2015-02-20 18:21:23 +05:30
parent ee214d6df5
commit 3c8984d2a5
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/css_selectors/select.py
View File

@ -125,6 +125,8 @@ class Select(object):
}
def __init__(self, root, default_lang=None, dispatch_map=None, trace=False):
if hasattr(root, 'getroot'):
root = root.getroot()
self.root = root
self.dispatch_map = dispatch_map or default_dispatch_map
self.invalidate_caches()
@ -568,4 +570,4 @@ if __name__ == '__main__':
from pprint import pprint
root = etree.fromstring('<body xmlns="xxx" xml:lang="en"><p id="p" class="one two" lang="fr"><a id="a"/><b/><c/><d/></p></body>')
select = Select(root, trace=True)
pprint(list(select('p *:root')))
pprint(list(select('p')))