use dict.get in case upload_date does not exist

pull/8/head
Peter Rowlands 9 years ago
parent 8d93c21466
commit 0fdbe3146c

@ -66,7 +66,7 @@ class AfreecaTVIE(InfoExtractor):
@staticmethod @staticmethod
def parse_video_key(key): def parse_video_key(key):
video_key = {'upload_date': None, 'part': '0'} video_key = {}
m = re.match(r'^(?P<upload_date>\d{8})_\w+_(?P<part>\d+)$', key) m = re.match(r'^(?P<upload_date>\d{8})_\w+_(?P<part>\d+)$', key)
if m: if m:
video_key['upload_date'] = m.group('upload_date') video_key['upload_date'] = m.group('upload_date')
@ -92,12 +92,12 @@ class AfreecaTVIE(InfoExtractor):
thumbnail = xpath_text(video_xml, './track/titleImage', 'thumbnail') thumbnail = xpath_text(video_xml, './track/titleImage', 'thumbnail')
entries = [] entries = []
for video_file in video_xml.findall('./track/video/file'): for i, video_file in enumerate(video_xml.findall('./track/video/file')):
video_key = self.parse_video_key(video_file.get('key')) video_key = self.parse_video_key(video_file.get('key'))
entries.append({ entries.append({
'id': '%s_%s' % (video_id, video_key['part']), 'id': '%s_%s' % (video_id, video_key.get('part', i + 1)),
'title': title, 'title': title,
'upload_date': video_key['upload_date'], 'upload_date': video_key.get('upload_date'),
'duration': int_or_none(video_file.get('duration')), 'duration': int_or_none(video_file.get('duration')),
'url': video_file.text, 'url': video_file.text,
}) })

Loading…
Cancel
Save