From 6689f45469f5f14dd338ae7743313f7b8c6c5174 Mon Sep 17 00:00:00 2001
From: steven7851 <steven7851@msn.com>
Date: Fri, 29 Jul 2016 13:35:05 +0800
Subject: [PATCH] [Douyutv]Update api

Thanks #1299 @cnbeining #1235 @hellofwy
---
 src/you_get/extractors/douyutv.py | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/src/you_get/extractors/douyutv.py b/src/you_get/extractors/douyutv.py
index 449022ca..29bcedd4 100644
--- a/src/you_get/extractors/douyutv.py
+++ b/src/you_get/extractors/douyutv.py
@@ -6,27 +6,44 @@ from ..common import *
 import json
 import hashlib
 import time
+import random
+import string
+import requests
 
 def douyutv_download(url, output_dir = '.', merge = True, info_only = False, **kwargs):
     room_id = url[url.rfind('/')+1:]
-    #Thanks to @yan12125 for providing decoding method!!
-    suffix = 'room/%s?aid=android&client_sys=android&time=%d' % (room_id, int(time.time()))
-    sign = hashlib.md5((suffix + '1231').encode('ascii')).hexdigest()
-    json_request_url = "http://www.douyu.com/api/v1/%s&auth=%s" % (suffix, sign)
+
+    json_request_url = "http://m.douyu.com/html5/live?roomId=%s" % room_id
     content = get_html(json_request_url)
     data = json.loads(content)['data']
     server_status = data.get('error',0)
     if server_status is not 0:
         raise ValueError("Server returned error:%s" % server_status)
+
     title = data.get('room_name')
     show_status = data.get('show_status')
     if show_status is not "1":
         raise ValueError("The live stream is not online! (Errno:%s)" % server_status)
+
+    tt = int(time.time() / 60)
+    did = ''.join([random.choice(string.ascii_uppercase + string.digits) for n in range(32)])
+    sign = hashlib.md5((room_id + did + 'A12Svb&%1UUmf@hC' + "%d" % tt).encode("utf-8")).hexdigest()
+    json_request_url = "http://www.douyu.com/lapi/live/getPlay/%s" % room_id
+
+    payload = {'cdn': 'ws', 'rate': '0', 'tt': tt, 'did': did, 'sign': sign}
+    r = requests.post(json_request_url, data=payload)
+    content = r.json()
+    data = content['data']
+
+    server_status = data.get('error',0)
+    if server_status is not 0:
+        raise ValueError("Server returned error:%s" % server_status)
+
     real_url = data.get('rtmp_url')+'/'+data.get('rtmp_live')
 
     print_info(site_info, title, 'flv', float('inf'))
     if not info_only:
-        download_urls([real_url], title, 'flv', None, output_dir, merge = merge)
+        download_url_ffmpeg(real_url, title, 'flv', None, output_dir, merge = merge)
 
 site_info = "douyu.com"
 download = douyutv_download