Madeorsk/youtube-dl
1
0
Fork 0
mirror of https://codeberg.org/polarisfm/youtube-dl synced 2025-01-10 15:17:53 +01:00
Code Issues Releases Wiki Activity

[picta] Fix KeyError when missing attributes in MPD

Browse Source 
Get element attribute using  element.get('key') instead of element.attrib['index'] manifests like:
    SegmentList/Initialization@sourceURL
    SegmentList/SegmentURL@media

Example MPD manifest https://www.picta.cu/videos/3bf20311b6564d61a0fd519b3c7dcced/manifest.mpd
This commit is contained in:
Oleksis 2019-04-06 02:29:01 -04:00
parent 3fb09eecd5
commit b5626f1979
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
youtube_dl/extractor/common.py
View File

@ -2054,8 +2054,8 @@ class InfoExtractor(object):
def extract_Initialization(source): def extract_Initialization(source):
initialization = source.find(_add_ns('Initialization')) initialization = source.find(_add_ns('Initialization'))
if initialization is not None: if initialization is not None and initialization.get('sourceURL') is not None:
ms_info['initialization_url'] = initialization.attrib['sourceURL'] ms_info['initialization_url'] = initialization.get('sourceURL')
segment_list = element.find(_add_ns('SegmentList')) segment_list = element.find(_add_ns('SegmentList'))
if segment_list is not None: if segment_list is not None:
@ -2063,7 +2063,9 @@ class InfoExtractor(object):
extract_Initialization(segment_list) extract_Initialization(segment_list)
segment_urls_e = segment_list.findall(_add_ns('SegmentURL')) segment_urls_e = segment_list.findall(_add_ns('SegmentURL'))
if segment_urls_e: if segment_urls_e:
ms_info['segment_urls'] = [segment.attrib['media'] for segment in segment_urls_e] segment_urls = [segment.get('media') for segment in segment_urls_e if segment.get('media') is not None]
if len(segment_urls) > 0:
ms_info['segment_urls'] = segment_urls
else: else:
segment_template = element.find(_add_ns('SegmentTemplate')) segment_template = element.find(_add_ns('SegmentTemplate'))
if segment_template is not None: if segment_template is not None: