mirror of https://github.com/yt-dlp/yt-dlp.git
272 lines
8.5 KiB
Python
272 lines
8.5 KiB
Python
# coding: utf-8
|
|
from __future__ import unicode_literals
|
|
|
|
import base64
|
|
import random
|
|
import string
|
|
import time
|
|
|
|
from .common import InfoExtractor
|
|
from ..compat import (
|
|
compat_urllib_parse,
|
|
compat_ord,
|
|
)
|
|
from ..utils import (
|
|
ExtractorError,
|
|
sanitized_Request,
|
|
)
|
|
|
|
|
|
class YoukuIE(InfoExtractor):
|
|
IE_NAME = 'youku'
|
|
IE_DESC = '优酷'
|
|
_VALID_URL = r'''(?x)
|
|
(?:
|
|
http://(?:v|player)\.youku\.com/(?:v_show/id_|player\.php/sid/)|
|
|
youku:)
|
|
(?P<id>[A-Za-z0-9]+)(?:\.html|/v\.swf|)
|
|
'''
|
|
|
|
_TESTS = [{
|
|
# MD5 is unstable
|
|
'url': 'http://v.youku.com/v_show/id_XMTc1ODE5Njcy.html',
|
|
'info_dict': {
|
|
'id': 'XMTc1ODE5Njcy_part1',
|
|
'title': '★Smile﹗♡ Git Fresh -Booty Music舞蹈.',
|
|
'ext': 'flv'
|
|
}
|
|
}, {
|
|
'url': 'http://player.youku.com/player.php/sid/XNDgyMDQ2NTQw/v.swf',
|
|
'only_matching': True,
|
|
}, {
|
|
'url': 'http://v.youku.com/v_show/id_XODgxNjg1Mzk2_ev_1.html',
|
|
'info_dict': {
|
|
'id': 'XODgxNjg1Mzk2',
|
|
'title': '武媚娘传奇 85',
|
|
},
|
|
'playlist_count': 11,
|
|
'skip': 'Available in China only',
|
|
}, {
|
|
'url': 'http://v.youku.com/v_show/id_XMTI1OTczNDM5Mg==.html',
|
|
'info_dict': {
|
|
'id': 'XMTI1OTczNDM5Mg',
|
|
'title': '花千骨 04',
|
|
},
|
|
'playlist_count': 13,
|
|
}, {
|
|
'url': 'http://v.youku.com/v_show/id_XNjA1NzA2Njgw.html',
|
|
'note': 'Video protected with password',
|
|
'info_dict': {
|
|
'id': 'XNjA1NzA2Njgw',
|
|
'title': '邢義田复旦讲座之想象中的胡人—从“左衽孔子”说起',
|
|
},
|
|
'playlist_count': 19,
|
|
'params': {
|
|
'videopassword': '100600',
|
|
},
|
|
}]
|
|
|
|
def construct_video_urls(self, data):
|
|
# get sid, token
|
|
def yk_t(s1, s2):
|
|
ls = list(range(256))
|
|
t = 0
|
|
for i in range(256):
|
|
t = (t + ls[i] + compat_ord(s1[i % len(s1)])) % 256
|
|
ls[i], ls[t] = ls[t], ls[i]
|
|
s = bytearray()
|
|
x, y = 0, 0
|
|
for i in range(len(s2)):
|
|
y = (y + 1) % 256
|
|
x = (x + ls[y]) % 256
|
|
ls[x], ls[y] = ls[y], ls[x]
|
|
s.append(compat_ord(s2[i]) ^ ls[(ls[x] + ls[y]) % 256])
|
|
return bytes(s)
|
|
|
|
sid, token = yk_t(
|
|
b'becaf9be', base64.b64decode(data['security']['encrypt_string'].encode('ascii'))
|
|
).decode('ascii').split('_')
|
|
|
|
# get oip
|
|
oip = data['security']['ip']
|
|
|
|
fileid_dict = {}
|
|
for stream in data['stream']:
|
|
format = stream.get('stream_type')
|
|
fileid = stream['stream_fileid']
|
|
fileid_dict[format] = fileid
|
|
|
|
def get_fileid(format, n):
|
|
number = hex(int(str(n), 10))[2:].upper()
|
|
if len(number) == 1:
|
|
number = '0' + number
|
|
streamfileids = fileid_dict[format]
|
|
fileid = streamfileids[0:8] + number + streamfileids[10:]
|
|
return fileid
|
|
|
|
# get ep
|
|
def generate_ep(format, n):
|
|
fileid = get_fileid(format, n)
|
|
ep_t = yk_t(
|
|
b'bf7e5f01',
|
|
('%s_%s_%s' % (sid, fileid, token)).encode('ascii')
|
|
)
|
|
ep = base64.b64encode(ep_t).decode('ascii')
|
|
return ep
|
|
|
|
# generate video_urls
|
|
video_urls_dict = {}
|
|
for stream in data['stream']:
|
|
format = stream.get('stream_type')
|
|
video_urls = []
|
|
for dt in stream['segs']:
|
|
n = str(stream['segs'].index(dt))
|
|
param = {
|
|
'K': dt['key'],
|
|
'hd': self.get_hd(format),
|
|
'myp': 0,
|
|
'ypp': 0,
|
|
'ctype': 12,
|
|
'ev': 1,
|
|
'token': token,
|
|
'oip': oip,
|
|
'ep': generate_ep(format, n)
|
|
}
|
|
video_url = \
|
|
'http://k.youku.com/player/getFlvPath/' + \
|
|
'sid/' + sid + \
|
|
'_00' + \
|
|
'/st/' + self.parse_ext_l(format) + \
|
|
'/fileid/' + get_fileid(format, n) + '?' + \
|
|
compat_urllib_parse.urlencode(param)
|
|
video_urls.append(video_url)
|
|
video_urls_dict[format] = video_urls
|
|
|
|
return video_urls_dict
|
|
|
|
@staticmethod
|
|
def get_ysuid():
|
|
return '%d%s' % (int(time.time()), ''.join([
|
|
random.choice(string.ascii_letters) for i in range(3)]))
|
|
|
|
def get_hd(self, fm):
|
|
hd_id_dict = {
|
|
'3gp': '0',
|
|
'3gphd': '1',
|
|
'flv': '0',
|
|
'flvhd': '0',
|
|
'mp4': '1',
|
|
'mp4hd': '1',
|
|
'mp4hd2': '1',
|
|
'mp4hd3': '1',
|
|
'hd2': '2',
|
|
'hd3': '3',
|
|
}
|
|
return hd_id_dict[fm]
|
|
|
|
def parse_ext_l(self, fm):
|
|
ext_dict = {
|
|
'3gp': 'flv',
|
|
'3gphd': 'mp4',
|
|
'flv': 'flv',
|
|
'flvhd': 'flv',
|
|
'mp4': 'mp4',
|
|
'mp4hd': 'mp4',
|
|
'mp4hd2': 'flv',
|
|
'mp4hd3': 'flv',
|
|
'hd2': 'flv',
|
|
'hd3': 'flv',
|
|
}
|
|
return ext_dict[fm]
|
|
|
|
def get_format_name(self, fm):
|
|
_dict = {
|
|
'3gp': 'h6',
|
|
'3gphd': 'h5',
|
|
'flv': 'h4',
|
|
'flvhd': 'h4',
|
|
'mp4': 'h3',
|
|
'mp4hd': 'h3',
|
|
'mp4hd2': 'h4',
|
|
'mp4hd3': 'h4',
|
|
'hd2': 'h2',
|
|
'hd3': 'h1',
|
|
}
|
|
return _dict[fm]
|
|
|
|
def _real_extract(self, url):
|
|
video_id = self._match_id(url)
|
|
|
|
self._set_cookie('youku.com', '__ysuid', self.get_ysuid())
|
|
|
|
def retrieve_data(req_url, note):
|
|
headers = {
|
|
'Referer': req_url,
|
|
}
|
|
self._set_cookie('youku.com', 'xreferrer', 'http://www.youku.com')
|
|
req = sanitized_Request(req_url, headers=headers)
|
|
|
|
cn_verification_proxy = self._downloader.params.get('cn_verification_proxy')
|
|
if cn_verification_proxy:
|
|
req.add_header('Ytdl-request-proxy', cn_verification_proxy)
|
|
|
|
raw_data = self._download_json(req, video_id, note=note)
|
|
|
|
return raw_data['data']
|
|
|
|
video_password = self._downloader.params.get('videopassword')
|
|
|
|
# request basic data
|
|
basic_data_url = 'http://play.youku.com/play/get.json?vid=%s&ct=12' % video_id
|
|
if video_password:
|
|
basic_data_url += '&pwd=%s' % video_password
|
|
|
|
data = retrieve_data(basic_data_url, 'Downloading JSON metadata')
|
|
|
|
error = data.get('error')
|
|
if error:
|
|
error_note = error.get('note')
|
|
if error_note is not None and '因版权原因无法观看此视频' in error_note:
|
|
raise ExtractorError(
|
|
'Youku said: Sorry, this video is available in China only', expected=True)
|
|
elif error_note and '该视频被设为私密' in error_note:
|
|
raise ExtractorError(
|
|
'Youku said: Sorry, this video is private', expected=True)
|
|
else:
|
|
msg = 'Youku server reported error %i' % error.get('code')
|
|
if error_note is not None:
|
|
msg += ': ' + error_note
|
|
raise ExtractorError(msg)
|
|
|
|
# get video title
|
|
title = data['video']['title']
|
|
|
|
# generate video_urls_dict
|
|
video_urls_dict = self.construct_video_urls(data)
|
|
|
|
# construct info
|
|
entries = [{
|
|
'id': '%s_part%d' % (video_id, i + 1),
|
|
'title': title,
|
|
'formats': [],
|
|
# some formats are not available for all parts, we have to detect
|
|
# which one has all
|
|
} for i in range(max(len(v.get('segs')) for v in data['stream']))]
|
|
for stream in data['stream']:
|
|
fm = stream.get('stream_type')
|
|
video_urls = video_urls_dict[fm]
|
|
for video_url, seg, entry in zip(video_urls, stream['segs'], entries):
|
|
entry['formats'].append({
|
|
'url': video_url,
|
|
'format_id': self.get_format_name(fm),
|
|
'ext': self.parse_ext_l(fm),
|
|
'filesize': int(seg['size']),
|
|
})
|
|
|
|
return {
|
|
'_type': 'multi_video',
|
|
'id': video_id,
|
|
'title': title,
|
|
'entries': entries,
|
|
}
|