yt_dlp/extractor/baidu.py

   1 from .common import InfoExtractor
   2 from ..utils import unescapeHTML
   3
   4
   5 class BaiduVideoIE(InfoExtractor):
   6     IE_DESC = '百度视频'
   7     _VALID_URL = r'https?://v\.baidu\.com/(?P<type>[a-z]+)/(?P<id>\d+)\.htm'
   8     _TESTS = [{
   9         'url': 'http://v.baidu.com/comic/1069.htm?frp=bdbrand&q=%E4%B8%AD%E5%8D%8E%E5%B0%8F%E5%BD%93%E5%AE%B6',
  10         'info_dict': {
  11             'id': '1069',
  12             'title': '中华小当家 TV版国语',
  13             'description': 'md5:51be07afe461cf99fa61231421b5397c',
  14         },
  15         'playlist_count': 52,
  16     }, {
  17         'url': 'http://v.baidu.com/show/11595.htm?frp=bdbrand',
  18         'info_dict': {
  19             'id': '11595',
  20             'title': 're:^奔跑吧兄弟',
  21             'description': 'md5:1bf88bad6d850930f542d51547c089b8',
  22         },
  23         'playlist_mincount': 12,
  24     }]
  25
  26     def _call_api(self, path, category, playlist_id, note):
  27         return self._download_json('http://app.video.baidu.com/%s/?worktype=adnative%s&id=%s' % (
  28             path, category, playlist_id), playlist_id, note)
  29
  30     def _real_extract(self, url):
  31         category, playlist_id = self._match_valid_url(url).groups()
  32         if category == 'show':
  33             category = 'tvshow'
  34         if category == 'tv':
  35             category = 'tvplay'
  36
  37         playlist_detail = self._call_api(
  38             'xqinfo', category, playlist_id, 'Download playlist JSON metadata')
  39
  40         playlist_title = playlist_detail['title']
  41         playlist_description = unescapeHTML(playlist_detail.get('intro'))
  42
  43         episodes_detail = self._call_api(
  44             'xqsingle', category, playlist_id, 'Download episodes JSON metadata')
  45
  46         entries = [self.url_result(
  47             episode['url'], video_title=episode['title']
  48         ) for episode in episodes_detail['videos']]
  49
  50         return self.playlist_result(
  51             entries, playlist_id, playlist_title, playlist_description)