[rockstargames] Improve and add Youtube fallback (Closes #9737)

This commit is contained in:
Sergey M․ 2016-06-14 01:11:24 +07:00
parent 14d0f4e0f3
commit 16b6bd01d2
No known key found for this signature in database
GPG Key ID: 2C393E0F18A9236D

View File

@ -3,52 +3,67 @@
from .common import InfoExtractor
from ..utils import (
qualities,
parse_iso8601
int_or_none,
parse_iso8601,
)
class RockstarGamesIE(InfoExtractor):
_VALID_URL = r'https?://(?:www\.)?rockstargames\.com/videos/video/(?P<id>[0-9]+)'
_TEST = {
_VALID_URL = r'https?://(?:www\.)?rockstargames\.com/videos(?:/video/|#?/?\?.*\bvideo=)(?P<id>\d+)'
_TESTS = [{
'url': 'https://www.rockstargames.com/videos/video/11544/',
'md5': '03b5caa6e357a4bd50e3143fc03e5733',
'info_dict': {
'id': '11544',
'ext': 'mp4',
'title': 'Further Adventures in Finance and Felony Trailer',
'thumbnail': 're:^https?://.*\.jpg$',
'description': 'md5:6d31f55f30cb101b5476c4a379e324a3',
'thumbnail': 're:^https?://.*\.jpg$',
'timestamp': 1464876000,
'upload_date': '20160602',
'timestamp': 1464876000
}
}
}, {
'url': 'http://www.rockstargames.com/videos#/?video=48',
'only_matching': True,
}]
def _real_extract(self, url):
video_id = self._match_id(url)
json_data = self._download_json(
'https://www.rockstargames.com/videoplayer/videos/get-video.json?id=%s&locale=en_us' % video_id,
video_id
)['video']
video = self._download_json(
'https://www.rockstargames.com/videoplayer/videos/get-video.json',
video_id, query={
'id': video_id,
'locale': 'en_us',
})['video']
title = video['title']
formats = []
for video in json_data['files_processed']['video/mp4']:
for video in video['files_processed']['video/mp4']:
if not video.get('src'):
continue
height = video.get('resolution', '').replace('p', '')
resolution = video.get('resolution')
height = int_or_none(self._search_regex(
r'^(\d+)[pP]$', resolution or '', 'height', default=None))
formats.append({
'url': self._proto_relative_url(video['src']),
'height': int(height) if height.isdigit() else -1,
'format_id': resolution,
'height': height,
})
if not formats:
youtube_id = video.get('youtube_id')
if youtube_id:
return self.url_result(youtube_id, 'Youtube')
self._sort_formats(formats)
return {
'id': video_id,
'title': json_data['title'],
'description': json_data.get('description'),
'title': title,
'description': video.get('description'),
'thumbnail': self._proto_relative_url(video.get('screencap')),
'timestamp': parse_iso8601(video.get('created')),
'formats': formats,
'thumbnail': self._proto_relative_url(json_data.get('screencap')),
'timestamp': parse_iso8601(json_data.get('created'))
}