summaryrefslogtreecommitdiff
path: root/youtube_dl/extractor/litv.py
blob: 337b1b15cf9d783750fa225b5538edfbc1fcfde2 (plain)
    1 # coding: utf-8
    2 from __future__ import unicode_literals
    3 
    4 import json
    5 
    6 from .common import InfoExtractor
    7 from ..utils import (
    8     ExtractorError,
    9     int_or_none,
   10     smuggle_url,
   11     unsmuggle_url,
   12 )
   13 
   14 
   15 class LiTVIE(InfoExtractor):
   16     _VALID_URL = r'https?://(?:www\.)?litv\.tv/(?:vod|promo)/[^/]+/(?:content\.do)?\?.*?\b(?:content_)?id=(?P<id>[^&]+)'
   17 
   18     _URL_TEMPLATE = 'https://www.litv.tv/vod/%s/content.do?id=%s'
   19 
   20     _TESTS = [{
   21         'url': 'https://www.litv.tv/vod/drama/content.do?brc_id=root&id=VOD00041610&isUHEnabled=true&autoPlay=1',
   22         'info_dict': {
   23             'id': 'VOD00041606',
   24             'title': '花千骨',
   25         },
   26         'playlist_count': 50,
   27     }, {
   28         'url': 'https://www.litv.tv/vod/drama/content.do?brc_id=root&id=VOD00041610&isUHEnabled=true&autoPlay=1',
   29         'md5': '969e343d9244778cb29acec608e53640',
   30         'info_dict': {
   31             'id': 'VOD00041610',
   32             'ext': 'mp4',
   33             'title': '花千骨第1集',
   34             'thumbnail': r're:https?://.*\.jpg$',
   35             'description': 'md5:c7017aa144c87467c4fb2909c4b05d6f',
   36             'episode_number': 1,
   37         },
   38         'params': {
   39             'noplaylist': True,
   40         },
   41         'skip': 'Georestricted to Taiwan',
   42     }, {
   43         'url': 'https://www.litv.tv/promo/miyuezhuan/?content_id=VOD00044841&',
   44         'md5': '88322ea132f848d6e3e18b32a832b918',
   45         'info_dict': {
   46             'id': 'VOD00044841',
   47             'ext': 'mp4',
   48             'title': '芈月傳第1集 霸星芈月降世楚國',
   49             'description': '楚威王二年,太史令唐昧夜觀星象,發現霸星即將現世。王后得知霸星的預言後,想盡辦法不讓孩子順利出生,幸得莒姬相護化解危機。沒想到眾人期待下出生的霸星卻是位公主,楚威王對此失望至極。楚王后命人將女嬰丟棄河中,居然奇蹟似的被少司命像攔下,楚威王認為此女非同凡響,為她取名芈月。',
   50         },
   51         'skip': 'Georestricted to Taiwan',
   52     }]
   53 
   54     def _extract_playlist(self, season_list, video_id, program_info, prompt=True):
   55         episode_title = program_info['title']
   56         content_id = season_list['contentId']
   57 
   58         if prompt:
   59             self.to_screen('Downloading playlist %s - add --no-playlist to just download video %s' % (content_id, video_id))
   60 
   61         all_episodes = [
   62             self.url_result(smuggle_url(
   63                 self._URL_TEMPLATE % (program_info['contentType'], episode['contentId']),
   64                 {'force_noplaylist': True}))  # To prevent infinite recursion
   65             for episode in season_list['episode']]
   66 
   67         return self.playlist_result(all_episodes, content_id, episode_title)
   68 
   69     def _real_extract(self, url):
   70         url, data = unsmuggle_url(url, {})
   71 
   72         video_id = self._match_id(url)
   73 
   74         noplaylist = self._downloader.params.get('noplaylist')
   75         noplaylist_prompt = True
   76         if 'force_noplaylist' in data:
   77             noplaylist = data['force_noplaylist']
   78             noplaylist_prompt = False
   79 
   80         webpage = self._download_webpage(url, video_id)
   81 
   82         program_info = self._parse_json(self._search_regex(
   83             r'var\s+programInfo\s*=\s*([^;]+)', webpage, 'VOD data', default='{}'),
   84             video_id)
   85 
   86         season_list = list(program_info.get('seasonList', {}).values())
   87         if season_list:
   88             if not noplaylist:
   89                 return self._extract_playlist(
   90                     season_list[0], video_id, program_info,
   91                     prompt=noplaylist_prompt)
   92 
   93             if noplaylist_prompt:
   94                 self.to_screen('Downloading just video %s because of --no-playlist' % video_id)
   95 
   96         # In browsers `getMainUrl` request is always issued. Usually this
   97         # endpoint gives the same result as the data embedded in the webpage.
   98         # If georestricted, there are no embedded data, so an extra request is
   99         # necessary to get the error code
  100         if 'assetId' not in program_info:
  101             program_info = self._download_json(
  102                 'https://www.litv.tv/vod/ajax/getProgramInfo', video_id,
  103                 query={'contentId': video_id},
  104                 headers={'Accept': 'application/json'})
  105         video_data = self._parse_json(self._search_regex(
  106             r'uiHlsUrl\s*=\s*testBackendData\(([^;]+)\);',
  107             webpage, 'video data', default='{}'), video_id)
  108         if not video_data:
  109             payload = {
  110                 'assetId': program_info['assetId'],
  111                 'watchDevices': program_info['watchDevices'],
  112                 'contentType': program_info['contentType'],
  113             }
  114             video_data = self._download_json(
  115                 'https://www.litv.tv/vod/getMainUrl', video_id,
  116                 data=json.dumps(payload).encode('utf-8'),
  117                 headers={'Content-Type': 'application/json'})
  118 
  119         if not video_data.get('fullpath'):
  120             error_msg = video_data.get('errorMessage')
  121             if error_msg == 'vod.error.outsideregionerror':
  122                 self.raise_geo_restricted('This video is available in Taiwan only')
  123             if error_msg:
  124                 raise ExtractorError('%s said: %s' % (self.IE_NAME, error_msg), expected=True)
  125             raise ExtractorError('Unexpected result from %s' % self.IE_NAME)
  126 
  127         formats = self._extract_m3u8_formats(
  128             video_data['fullpath'], video_id, ext='mp4',
  129             entry_protocol='m3u8_native', m3u8_id='hls')
  130         for a_format in formats:
  131             # LiTV HLS segments doesn't like compressions
  132             a_format.setdefault('http_headers', {})['Youtubedl-no-compression'] = True
  133 
  134         title = program_info['title'] + program_info.get('secondaryMark', '')
  135         description = program_info.get('description')
  136         thumbnail = program_info.get('imageFile')
  137         categories = [item['name'] for item in program_info.get('category', [])]
  138         episode = int_or_none(program_info.get('episode'))
  139 
  140         return {
  141             'id': video_id,
  142             'formats': formats,
  143             'title': title,
  144             'description': description,
  145             'thumbnail': thumbnail,
  146             'categories': categories,
  147             'episode_number': episode,
  148         }

Generated by cgit