mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-30 18:00:24 -05:00
118 lines
3.6 KiB
Python
118 lines
3.6 KiB
Python
# 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,
|
|
}
|