xZero707/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-12-18 00:05:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

[utils] Use compat_xpath

Browse Source
This commit is contained in:
Sergey M․ 2016-03-18 02:52:23 +06:00
parent 57f7e3c62d
commit 810c10baa1
1 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
youtube_dl/utils.py
View File

@ -50,6 +50,7 @@
compat_urllib_parse_urlparse,
compat_urllib_request,
compat_urlparse,
compat_xpath,
shlex_quote,
)
@ -165,12 +166,7 @@ def find_xpath_attr(node, xpath, key, val=None):
return node.find(expr)
else:
def find_xpath_attr(node, xpath, key, val=None):
# Here comes the crazy part: In 2.6, if the xpath is a unicode,
# .//node does not match if a node is a direct child of . !
if isinstance(xpath, compat_str):
xpath = xpath.encode('ascii')
for f in node.findall(xpath):
for f in node.findall(compat_xpath(xpath)):
if key not in f.attrib:
continue
if val is None or f.attrib.get(key) == val:
@ -195,9 +191,7 @@ def xpath_with_ns(path, ns_map):
def xpath_element(node, xpath, name=None, fatal=False, default=NO_DEFAULT):
def _find_xpath(xpath):
if sys.version_info < (2, 7): # Crazy 2.6
xpath = xpath.encode('ascii')
return node.find(xpath)
return node.find(compat_xpath(xpath))
if isinstance(xpath, (str, compat_str)):
n = _find_xpath(xpath)