From 651e472f3b4f275e3445751825c60fa4977cbd90 Mon Sep 17 00:00:00 2001
From: cnbeining <cnbeining@gmail.com>
Date: Mon, 26 Oct 2015 01:34:22 -0400
Subject: [PATCH] [dilidili] Add support

---
 README.md                          |  1 +
 src/you_get/common.py              |  1 +
 src/you_get/extractors/__init__.py |  1 +
 src/you_get/extractors/dilidili.py | 65 ++++++++++++++++++++++++++++++
 4 files changed, 68 insertions(+)
 create mode 100644 src/you_get/extractors/dilidili.py

diff --git a/README.md b/README.md
index 99fa1283..303346e5 100644
--- a/README.md
+++ b/README.md
@@ -335,6 +335,7 @@ Use `--url`/`-u` to get a list of downloadable resource URLs extracted from the
 | **Baidu<br/>百度贴吧** | <http://tieba.baidu.com/> |✓|✓| |
 | 爆米花网 | <http://www.baomihua.com/>     |✓| | |
 | **bilibili<br/>哔哩哔哩** | <http://www.bilibili.com/> |✓| | |
+| Dilidili | <http://www.dilidili.com/>    |✓| | |
 | 豆瓣     | <http://www.douban.com/>       | | |✓|
 | 斗鱼     | <http://www.douyutv.com/>      |✓| | |
 | 凤凰视频 | <http://v.ifeng.com/>          |✓| | |
diff --git a/src/you_get/common.py b/src/you_get/common.py
index d0df4c65..5ee03631 100755
--- a/src/you_get/common.py
+++ b/src/you_get/common.py
@@ -12,6 +12,7 @@ SITES = {
     'cntv'       : 'cntv',
     'cbs'        : 'cbs',
     'dailymotion': 'dailymotion',
+    'dilidili'   : 'dilidili',
     'dongting'   : 'dongting',
     'douban'     : 'douban',
     'douyutv'    : 'douyutv',
diff --git a/src/you_get/extractors/__init__.py b/src/you_get/extractors/__init__.py
index 147d57d2..d78a111e 100755
--- a/src/you_get/extractors/__init__.py
+++ b/src/you_get/extractors/__init__.py
@@ -9,6 +9,7 @@ from .bilibili import *
 from .cbs import *
 from .cntv import *
 from .dailymotion import *
+from .dilidili import *
 from .douban import *
 from .douyutv import *
 from .ehow import *
diff --git a/src/you_get/extractors/dilidili.py b/src/you_get/extractors/dilidili.py
new file mode 100644
index 00000000..1c5340a6
--- /dev/null
+++ b/src/you_get/extractors/dilidili.py
@@ -0,0 +1,65 @@
+#!/usr/bin/env python
+
+__all__ = ['dilidili_download']
+
+from ..common import *
+
+#----------------------------------------------------------------------
+def dilidili_parser_data_to_stream_types(typ ,vid ,hd2 ,sign):
+    """->list"""
+    parse_url = 'http://player.005.tv/parse.php?xmlurl=null&type={typ}&vid={vid}&hd={hd2}&sign={sign}'.format(typ = typ, vid = vid, hd2 = hd2, sign = sign)
+    html = get_html(parse_url)
+    
+    info = re.search(r'(\{[^{]+\})(\{[^{]+\})(\{[^{]+\})(\{[^{]+\})(\{[^{]+\})', html).groups()
+    info = [i.strip('{}').split('->') for i in info]
+    info = {i[0]: i [1] for i in info}
+    
+    stream_types = []
+    for i in zip(info['deft'].split('|'), info['defa'].split('|')):
+        stream_types.append({'id': str(i[1][-1]), 'container': 'mp4', 'video_profile': i[0]})
+    return stream_types
+
+#----------------------------------------------------------------------
+def dilidili_parser_data_to_download_url(typ ,vid ,hd2 ,sign):
+    """->str"""
+    parse_url = 'http://player.005.tv/parse.php?xmlurl=null&type={typ}&vid={vid}&hd={hd2}&sign={sign}'.format(typ = typ, vid = vid, hd2 = hd2, sign = sign)
+    html = get_html(parse_url)
+    
+    return match1(html, r'<file><!\[CDATA\[(.+)\]\]></file>')
+
+#----------------------------------------------------------------------
+def dilidili_download(url, output_dir = '.', merge = False, info_only = False, **kwargs):
+    if re.match(r'http://www.dilidili.com/watch/\w+', url):
+        html = get_html(url)
+        title = match1(html, r'<title>(.+)丨(.+)</title>')  #title
+        
+        # player loaded via internal iframe
+        frame_url = re.search(r'<iframe (.+)src="(.+)\" f(.+)</iframe>', html).group(2)
+        #https://player.005.tv:60000/?vid=a8760f03fd:a04808d307&v=yun&sign=a68f8110cacd892bc5b094c8e5348432
+        html = get_html(frame_url)
+        
+        match = re.search(r'(.+?)var video =(.+?);', html)
+        vid = match1(html, r'var vid="(.+)"')
+        hd2 = match1(html, r'var hd2="(.+)"')
+        typ = match1(html, r'var typ="(.+)"')
+        sign = match1(html, r'var sign="(.+)"')
+        
+        # here s the parser...
+        stream_types = dilidili_parser_data_to_stream_types(typ, vid, hd2, sign)
+        
+        #get best
+        best_id = max([i['id'] for i in stream_types])
+        
+        url = dilidili_parser_data_to_download_url(typ, vid, best_id, sign)
+
+        type_ = ''
+        size = 0
+
+        type_, ext, size = url_info(url)
+        print_info(site_info, title, type_, size)
+        if not info_only:
+            download_urls([url], title, ext, total_size=None, output_dir=output_dir, merge=merge)
+
+site_info = "dilidili"
+download = dilidili_download
+download_playlist = playlist_not_supported('dilidili')