143 lines
4.7 KiB
Python
143 lines
4.7 KiB
Python
# coding: utf-8
|
|
from __future__ import unicode_literals
|
|
|
|
import json
|
|
|
|
from .common import InfoExtractor
|
|
from ..compat import compat_HTTPError
|
|
from ..utils import (
|
|
determine_ext,
|
|
ExtractorError,
|
|
float_or_none,
|
|
int_or_none,
|
|
mimetype2ext,
|
|
parse_iso8601,
|
|
urljoin,
|
|
)
|
|
|
|
|
|
class YandexDiskIE(InfoExtractor):
|
|
_VALID_URL = r'''(?x)https?://
|
|
(?:
|
|
(?:www\.)?yadi\.sk|
|
|
disk\.yandex\.
|
|
(?:
|
|
az|
|
|
by|
|
|
co(?:m(?:\.(?:am|ge|tr))?|\.il)|
|
|
ee|
|
|
fr|
|
|
k[gz]|
|
|
l[tv]|
|
|
md|
|
|
t[jm]|
|
|
u[az]|
|
|
ru
|
|
)
|
|
)/(?:[di]/|public.*?\bhash=)(?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',
|
|
'timestamp': 1421396809,
|
|
'upload_date': '20150116',
|
|
'view_count': int,
|
|
},
|
|
}, {
|
|
'url': 'https://yadi.sk/d/h3WAXvDS3Li3Ce',
|
|
'only_matching': True,
|
|
}, {
|
|
'url': 'https://yadi.sk/public?hash=5DZ296JK9GWCLp02f6jrObjnctjRxMs8L6%2B%2FuhNqk38%3D',
|
|
'only_matching': True,
|
|
}]
|
|
|
|
def _real_extract(self, url):
|
|
video_id = self._match_id(url)
|
|
|
|
try:
|
|
resource = self._download_json(
|
|
'https://cloud-api.yandex.net/v1/disk/public/resources',
|
|
video_id, query={'public_key': url})
|
|
except ExtractorError as e:
|
|
if isinstance(e.cause, compat_HTTPError) and e.cause.code == 403:
|
|
error_description = self._parse_json(
|
|
e.cause.read().decode(), video_id)['description']
|
|
raise ExtractorError(error_description, expected=True)
|
|
raise
|
|
|
|
title = resource['name']
|
|
public_url = resource.get('public_url')
|
|
if public_url:
|
|
video_id = self._match_id(public_url)
|
|
|
|
self._set_cookie('yadi.sk', 'yandexuid', '0')
|
|
|
|
def call_api(action):
|
|
return (self._download_json(
|
|
urljoin(url, '/public/api/') + action, video_id, data=json.dumps({
|
|
'hash': url,
|
|
# obtain sk if needed from call_api('check-auth') while
|
|
# the yandexuid cookie is set and sending an empty JSON object
|
|
'sk': 'ya6b52f8c6b12abe91a66d22d3a31084b'
|
|
}).encode(), headers={
|
|
'Content-Type': 'text/plain',
|
|
}, fatal=False) or {}).get('data') or {}
|
|
|
|
formats = []
|
|
source_url = resource.get('file')
|
|
if not source_url:
|
|
source_url = call_api('download-url').get('url')
|
|
if source_url:
|
|
formats.append({
|
|
'url': source_url,
|
|
'format_id': 'source',
|
|
'ext': determine_ext(title, mimetype2ext(resource.get('mime_type')) or 'mp4'),
|
|
'quality': 1,
|
|
'filesize': int_or_none(resource.get('size'))
|
|
})
|
|
|
|
video_streams = call_api('get-video-streams')
|
|
for video in (video_streams.get('videos') or []):
|
|
format_url = video.get('url')
|
|
if not format_url:
|
|
continue
|
|
if video.get('dimension') == 'adaptive':
|
|
formats.extend(self._extract_m3u8_formats(
|
|
format_url, video_id, 'mp4', 'm3u8_native',
|
|
m3u8_id='hls', fatal=False))
|
|
else:
|
|
size = video.get('size') or {}
|
|
height = int_or_none(size.get('height'))
|
|
format_id = 'hls'
|
|
if height:
|
|
format_id += '-%dp' % height
|
|
formats.append({
|
|
'ext': 'mp4',
|
|
'format_id': format_id,
|
|
'height': height,
|
|
'protocol': 'm3u8_native',
|
|
'url': format_url,
|
|
'width': int_or_none(size.get('width')),
|
|
})
|
|
self._sort_formats(formats)
|
|
|
|
owner = resource.get('owner') or {}
|
|
|
|
return {
|
|
'id': video_id,
|
|
'title': title,
|
|
'duration': float_or_none(video_streams.get('duration'), 1000),
|
|
'uploader': owner.get('display_name'),
|
|
'uploader_id': owner.get('uid'),
|
|
'view_count': int_or_none(resource.get('views_count')),
|
|
'timestamp': parse_iso8601(resource.get('created')),
|
|
'formats': formats,
|
|
}
|