_ Git - youtube-dl/blob - youtube_dl/extractor/youku.py

   1 # coding: utf-8
   2 from __future__ import unicode_literals
   3
   4 import base64
   5
   6 from .common import InfoExtractor
   7 from ..utils import ExtractorError
   8
   9 from ..compat import (
  10     compat_urllib_parse,
  11     compat_ord,
  12 )
  13
  14
  15 class YoukuIE(InfoExtractor):
  16     IE_NAME = 'youku'
  17     _VALID_URL = r'''(?x)
  18         (?:
  19             http://(?:v|player)\.youku\.com/(?:v_show/id_|player\.php/sid/)|
  20             youku:)
  21         (?P<id>[A-Za-z0-9]+)(?:\.html|/v\.swf|)
  22     '''
  23
  24     _TESTS = [{
  25         'url': 'http://v.youku.com/v_show/id_XMTc1ODE5Njcy.html',
  26         'md5': '5f3af4192eabacc4501508d54a8cabd7',
  27         'info_dict': {
  28             'id': 'XMTc1ODE5Njcy_part1',
  29             'title': '★Smile﹗♡ Git Fresh -Booty Music舞蹈.',
  30             'ext': 'flv'
  31         }
  32     }, {
  33         'url': 'http://player.youku.com/player.php/sid/XNDgyMDQ2NTQw/v.swf',
  34         'only_matching': True,
  35     }, {
  36         'url': 'http://v.youku.com/v_show/id_XODgxNjg1Mzk2_ev_1.html',
  37         'info_dict': {
  38             'id': 'XODgxNjg1Mzk2',
  39             'title': '武媚娘传奇 85',
  40         },
  41         'playlist_count': 11,
  42     }]
  43
  44     def construct_video_urls(self, data1, data2):
  45         # get sid, token
  46         def yk_t(s1, s2):
  47             ls = list(range(256))
  48             t = 0
  49             for i in range(256):
  50                 t = (t + ls[i] + compat_ord(s1[i % len(s1)])) % 256
  51                 ls[i], ls[t] = ls[t], ls[i]
  52             s = bytearray()
  53             x, y = 0, 0
  54             for i in range(len(s2)):
  55                 y = (y + 1) % 256
  56                 x = (x + ls[y]) % 256
  57                 ls[x], ls[y] = ls[y], ls[x]
  58                 s.append(compat_ord(s2[i]) ^ ls[(ls[x] + ls[y]) % 256])
  59             return bytes(s)
  60
  61         sid, token = yk_t(
  62             b'becaf9be', base64.b64decode(data2['ep'].encode('ascii'))
  63         ).decode('ascii').split('_')
  64
  65         # get oip
  66         oip = data2['ip']
  67
  68         # get fileid
  69         string_ls = list(
  70             'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ/\:._-1234567890')
  71         shuffled_string_ls = []
  72         seed = data1['seed']
  73         N = len(string_ls)
  74         for ii in range(N):
  75             seed = (seed * 0xd3 + 0x754f) % 0x10000
  76             idx = seed * len(string_ls) // 0x10000
  77             shuffled_string_ls.append(string_ls[idx])
  78             del string_ls[idx]
  79
  80         fileid_dict = {}
  81         for format in data1['streamtypes']:
  82             streamfileid = [
  83                 int(i) for i in data1['streamfileids'][format].strip('*').split('*')]
  84             fileid = ''.join(
  85                 [shuffled_string_ls[i] for i in streamfileid])
  86             fileid_dict[format] = fileid[:8] + '%s' + fileid[10:]
  87
  88         def get_fileid(format, n):
  89             fileid = fileid_dict[format] % hex(int(n))[2:].upper().zfill(2)
  90             return fileid
  91
  92         # get ep
  93         def generate_ep(format, n):
  94             fileid = get_fileid(format, n)
  95             ep_t = yk_t(
  96                 b'bf7e5f01',
  97                 ('%s_%s_%s' % (sid, fileid, token)).encode('ascii')
  98             )
  99             ep = base64.b64encode(ep_t).decode('ascii')
 100             return ep
 101
 102         # generate video_urls
 103         video_urls_dict = {}
 104         for format in data1['streamtypes']:
 105             video_urls = []
 106             for dt in data1['segs'][format]:
 107                 n = str(int(dt['no']))
 108                 param = {
 109                     'K': dt['k'],
 110                     'hd': self.get_hd(format),
 111                     'myp': 0,
 112                     'ts': dt['seconds'],
 113                     'ypp': 0,
 114                     'ctype': 12,
 115                     'ev': 1,
 116                     'token': token,
 117                     'oip': oip,
 118                     'ep': generate_ep(format, n)
 119                 }
 120                 video_url = \
 121                     'http://k.youku.com/player/getFlvPath/' + \
 122                     'sid/' + sid + \
 123                     '_' + str(int(n) + 1).zfill(2) + \
 124                     '/st/' + self.parse_ext_l(format) + \
 125                     '/fileid/' + get_fileid(format, n) + '?' + \
 126                     compat_urllib_parse.urlencode(param)
 127                 video_urls.append(video_url)
 128             video_urls_dict[format] = video_urls
 129
 130         return video_urls_dict
 131
 132     def get_hd(self, fm):
 133         hd_id_dict = {
 134             'flv': '0',
 135             'mp4': '1',
 136             'hd2': '2',
 137             'hd3': '3',
 138             '3gp': '0',
 139             '3gphd': '1'
 140         }
 141         return hd_id_dict[fm]
 142
 143     def parse_ext_l(self, fm):
 144         ext_dict = {
 145             'flv': 'flv',
 146             'mp4': 'mp4',
 147             'hd2': 'flv',
 148             'hd3': 'flv',
 149             '3gp': 'flv',
 150             '3gphd': 'mp4'
 151         }
 152         return ext_dict[fm]
 153
 154     def get_format_name(self, fm):
 155         _dict = {
 156             '3gp': 'h6',
 157             '3gphd': 'h5',
 158             'flv': 'h4',
 159             'mp4': 'h3',
 160             'hd2': 'h2',
 161             'hd3': 'h1'
 162         }
 163         return _dict[fm]
 164
 165     def _real_extract(self, url):
 166         video_id = self._match_id(url)
 167
 168         # request basic data
 169         data1_url = 'http://v.youku.com/player/getPlayList/VideoIDS/%s' % video_id
 170         data2_url = 'http://v.youku.com/player/getPlayList/VideoIDS/%s/Pf/4/ctype/12/ev/1' % video_id
 171
 172         raw_data1 = self._download_json(data1_url, video_id)
 173         raw_data2 = self._download_json(data2_url, video_id)
 174         data1 = raw_data1['data'][0]
 175         data2 = raw_data2['data'][0]
 176
 177         error_code = data1.get('error_code')
 178         if error_code:
 179             # -8 means blocked outside China.
 180             # Chinese and English, separated by newline.
 181             error = data1.get('error')
 182             raise ExtractorError(
 183                 error or 'Server reported error %i' %
 184                 error_code,
 185                 expected=True)
 186
 187         title = data1['title']
 188
 189         # generate video_urls_dict
 190         video_urls_dict = self.construct_video_urls(data1, data2)
 191
 192         # construct info
 193         entries = []
 194         for fm in data1['streamtypes']:
 195             video_urls = video_urls_dict[fm]
 196             for i in range(len(video_urls)):
 197                 if len(entries) < i + 1:
 198                     entries.append({'formats': []})
 199                 entries[i]['formats'].append(
 200                     {
 201                         'url': video_urls[i],
 202                         'format_id': self.get_format_name(fm),
 203                         'ext': self.parse_ext_l(fm),
 204                         'filesize': int(data1['segs'][fm][i]['size'])
 205                     }
 206                 )
 207
 208         for i in range(len(entries)):
 209             entries[i].update(
 210                 {
 211                     'id': '%s_part%d' % (video_id, i + 1),
 212                     'title': title,
 213                 }
 214             )
 215
 216         return {
 217             '_type': 'multi_video',
 218             'id': video_id,
 219             'title': title,
 220             'entries': entries,
 221         }