You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
3.6 KiB
118 lines
3.6 KiB
# coding: utf-8 |
|
from __future__ import unicode_literals |
|
|
|
from .common import InfoExtractor |
|
from ..compat import compat_str |
|
from ..utils import ( |
|
determine_ext, |
|
float_or_none, |
|
int_or_none, |
|
try_get, |
|
urlencode_postdata, |
|
) |
|
|
|
|
|
class YandexDiskIE(InfoExtractor): |
|
_VALID_URL = r'https?://yadi\.sk/[di]/(?P<id>[^/?#&]+)' |
|
|
|
_TESTS = [{ |
|
'url': 'https://yadi.sk/i/VdOeDou8eZs6Y', |
|
'md5': '33955d7ae052f15853dc41f35f17581c', |
|
'info_dict': { |
|
'id': 'VdOeDou8eZs6Y', |
|
'ext': 'mp4', |
|
'title': '4.mp4', |
|
'duration': 168.6, |
|
'uploader': 'y.botova', |
|
'uploader_id': '300043621', |
|
'view_count': int, |
|
}, |
|
}, { |
|
'url': 'https://yadi.sk/d/h3WAXvDS3Li3Ce', |
|
'only_matching': True, |
|
}] |
|
|
|
def _real_extract(self, url): |
|
video_id = self._match_id(url) |
|
|
|
status = self._download_webpage( |
|
'https://disk.yandex.com/auth/status', video_id, query={ |
|
'urlOrigin': url, |
|
'source': 'public', |
|
'md5': 'false', |
|
}) |
|
|
|
sk = self._search_regex( |
|
r'(["\'])sk(?:External)?\1\s*:\s*(["\'])(?P<value>(?:(?!\2).)+)\2', |
|
status, 'sk', group='value') |
|
|
|
webpage = self._download_webpage(url, video_id) |
|
|
|
models = self._parse_json( |
|
self._search_regex( |
|
r'<script[^>]+id=["\']models-client[^>]+>\s*(\[.+?\])\s*</script', |
|
webpage, 'video JSON'), |
|
video_id) |
|
|
|
data = next( |
|
model['data'] for model in models |
|
if model.get('model') == 'resource') |
|
|
|
video_hash = data['id'] |
|
title = data['name'] |
|
|
|
models = self._download_json( |
|
'https://disk.yandex.com/models/', video_id, |
|
data=urlencode_postdata({ |
|
'_model.0': 'videoInfo', |
|
'id.0': video_hash, |
|
'_model.1': 'do-get-resource-url', |
|
'id.1': video_hash, |
|
'version': '13.6', |
|
'sk': sk, |
|
}), query={'_m': 'videoInfo'})['models'] |
|
|
|
videos = try_get(models, lambda x: x[0]['data']['videos'], list) or [] |
|
source_url = try_get( |
|
models, lambda x: x[1]['data']['file'], compat_str) |
|
|
|
formats = [] |
|
if source_url: |
|
formats.append({ |
|
'url': source_url, |
|
'format_id': 'source', |
|
'ext': determine_ext(title, 'mp4'), |
|
'quality': 1, |
|
}) |
|
for video in videos: |
|
format_url = video.get('url') |
|
if not format_url: |
|
continue |
|
if determine_ext(format_url) == 'm3u8': |
|
formats.extend(self._extract_m3u8_formats( |
|
format_url, video_id, 'mp4', entry_protocol='m3u8_native', |
|
m3u8_id='hls', fatal=False)) |
|
else: |
|
formats.append({ |
|
'url': format_url, |
|
}) |
|
self._sort_formats(formats) |
|
|
|
duration = float_or_none(try_get( |
|
models, lambda x: x[0]['data']['duration']), 1000) |
|
uploader = try_get( |
|
data, lambda x: x['user']['display_name'], compat_str) |
|
uploader_id = try_get( |
|
data, lambda x: x['user']['uid'], compat_str) |
|
view_count = int_or_none(try_get( |
|
data, lambda x: x['meta']['views_counter'])) |
|
|
|
return { |
|
'id': video_id, |
|
'title': title, |
|
'duration': duration, |
|
'uploader': uploader, |
|
'uploader_id': uploader_id, |
|
'view_count': view_count, |
|
'formats': formats, |
|
}
|
|
|