[cntv] Add extractor

2 files changed, 124 insertions, 0 deletions

diff --git a/youtube_dl/extractor/cntv.py b/youtube_dl/extractor/cntv.py
new file mode 100644
index 000000000..0b26ad32d
--- /dev/null
+++ b/youtube_dl/extractor/cntv.py
@@ -0,0 +1,123 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
+from .common import InfoExtractor
+from ..compat import compat_urllib_request
+
+class CntvIE(InfoExtractor):
+    IE_NAME = 'cntv'
+    IE_DESC = '央视网'
+    _VALID_URL = r'''(?x)
+        (?:
+            http://(?:ent|tv|sports|english|chunwan)\.(?:cntv|cctv)\.(?:com|cn)/
+            (?:
+                (?:video|special)/(?:[_A-Za-z0-9]+)|(?P<year>\d+)/(?P<month>\d+)/(?P<day>\d+)
+            )
+        )
+        /(?P<id>[A-Za-z0-9]+)(index)?(?:\.shtml|)
+    '''
+
+    _TESTS = [{
+        'url': 'http://sports.cntv.cn/2016/02/12/ARTIaBRxv4rTT1yWf1frW2wi160212.shtml',
+        'md5': 'd0f8c3e043fe43534517ed47c831fd4e',
+        'info_dict': {
+            'id': '5ecdbeab623f4973b40ff25f18b174e8',
+            'ext': 'flv',
+            'title': '[NBA]二少联手砍下46分 雷霆主场击败鹈鹕（快讯）',
+            'description': 'md5:7e14a5328dc5eb3d1cd6afbbe0574e95',
+            'uploader': 'songjunjie'
+        }
+    },{
+        'url': 'http://tv.cctv.com/2016/02/05/VIDEUS7apq3lKrHG9Dncm03B160205.shtml',
+        'md5': '3bde627de65b6bad064f3e67ec9e71a1',
+        'info_dict': {
+            'id': 'efc5d49e5b3b4ab2b34f3a502b73d3ae',
+            'ext': 'flv',
+            'title': '[赛车]“车王”舒马赫恢复情况成谜（快讯）',
+            'description': '2月4日，蒙特泽莫罗透露了关于“车王”舒马赫恢复情况，但情况是否属实遭到了质疑。',
+            'uploader': 'shujun'
+        }
+    },{
+        'url': 'http://english.cntv.cn/special/four_comprehensives/index.shtml',
+        'md5': 'f37a56d5f16647eae24c3e618f8d23a2',
+        'info_dict': {
+            'id': '4bb9bb4db7a6471ba85fdeda5af0381e',
+            'ext': 'flv',
+            'title': 'NHnews008 ANNUAL POLITICAL SEASON',
+            'description': 'Four Comprehensives',
+            'uploader': 'zhangyunlei'
+        }
+    },{
+        'url': 'http://ent.cntv.cn/2016/01/18/ARTIjprSSJH8DryTVr5Bx8Wb160118.shtml',
+        'md5': '37ad2407b5f28336ddf26aacea570ee5',
+        'info_dict': {
+            'id': '45336758dd474ef39dc6887f115e0375',
+            'ext': 'flv',
+            'title': '《爱国之恋》中国梦新歌展播  综艺视频精切',
+            'description': 'md5:856836ae2660f000e458ead597b1e2bd',
+            'uploader': 'baiyuqing'
+        }
+    },{
+        'url': 'http://tv.cntv.cn/video/C39296/e0210d949f113ddfb38d31f00a4e5c44',
+        'md5': 'd7188ecc2a2f8447c70023895724c31c',
+        'info_dict': {
+            'id': 'e0210d949f113ddfb38d31f00a4e5c44',
+            'ext': 'flv',
+            'title': '《传奇故事》 20150409 金钱“魔术”（下）',
+            'description': '本期节目主要内容：     2014年10月23日凌晨，浙江余杭警方出动2000多名警力，一举摧毁1040阳光工程传销组织，抓获65名涉嫌传销的组织头目，教育遣返400多名参与传销人员。',
+            'uploader': '陆银凯'
+        }
+    }]
+
+    def _real_extract(self, url):
+        video_id = self._match_id(url)
+        webpage = self._download_webpage(url, video_id)
+
+        video_id_regex = [r'var guid[ ="]+([a-zA-Z0-9]+)"',
+                    r'"videoCenterId"[ ,"]+([a-zA-Z0-9]+)"',
+                    r'"changePlayer\(\'([a-zA-Z0-9]+)\'\)\"']
+        video_id = self._search_regex(video_id_regex, webpage, 'video_id', default=None)
+
+        if video_id is None:
+            video_id = self._match_id(url)
+
+        # refer to youku.py
+        def retrieve_data(req_url, note):
+            req = compat_urllib_request.Request(req_url)
+
+            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
+
+        get_info_url = \
+            'http://vdn.apps.cntv.cn/api/getHttpVideoInfo.do' + \
+            '?pid=%s' % video_id + \
+            '&tz=-8' + \
+            '&from=000tv' + \
+            '&url=%s' % url + \
+            '&idl=32' + \
+            '&idlr=32' + \
+            '&modifyed=false'
+
+        data = retrieve_data(
+            get_info_url,
+            'Downloading JSON metadata')
+
+        title = data.get('title')
+        uploader = data.get('editer_name')
+        hls_url = data.get('hls_url')
+        formats = self._extract_m3u8_formats(hls_url, video_id, 'flv')
+        self._sort_formats(formats)
+
+        description = self._html_search_meta('description', webpage)
+
+        return {
+            'id': video_id,
+            'title': title,
+            'uploader': uploader,
+            'formats': formats,
+            'description': description,
+        }
diff --git a/youtube_dl/extractor/extractors.py b/youtube_dl/extractor/extractors.py
index 3017bf56c..4cd12562e 100644
--- a/youtube_dl/extractor/extractors.py
+++ b/youtube_dl/extractor/extractors.py
@@ -178,6 +178,7 @@ from .cnn import (
     CNNBlogsIE,
     CNNArticleIE,
 )
+from .cntv import CntvIE
 from .coub import CoubIE
 from .collegerama import CollegeRamaIE
 from .comedycentral import (