[RutubePlaylistIE] correct retrieval of category

This commit is contained in:
luceatnobis 2017-07-23 13:19:30 +02:00
parent d6391e5e46
commit bbfca487d0

View File

@ -255,6 +255,7 @@ class RutubePlaylistIE(InfoExtractor):
results = page['results'] results = page['results']
for result in results: for result in results:
entry = self.url_result(result.get('video_url'), 'Rutube') entry = self.url_result(result.get('video_url'), 'Rutube')
category = try_get(result, lambda x: x['category']['name'])
entry.update({ entry.update({
'id': result.get('id'), 'id': result.get('id'),
'uploader': try_get(result, lambda x: x['author']['name']), 'uploader': try_get(result, lambda x: x['author']['name']),
@ -264,7 +265,7 @@ class RutubePlaylistIE(InfoExtractor):
'description': result.get('description'), 'description': result.get('description'),
'thumbnail': result.get('thumbnail_url'), 'thumbnail': result.get('thumbnail_url'),
'duration': int_or_none(result.get('duration')), 'duration': int_or_none(result.get('duration')),
'category': [try_get(result, lambda x: x['category']['name'])], 'category': [category] if category else None,
'age_limit': 18 if result.get('is_adult') else 0, 'age_limit': 18 if result.get('is_adult') else 0,
'view_count': int_or_none(result.get('hits')), 'view_count': int_or_none(result.get('hits')),
'is_live': result.get('is_livestream'), 'is_live': result.get('is_livestream'),