🐛 Fix 斗鱼直播 (#185 #189 #191 #193 #195 #198 #200)

感谢两位大佬@wjxgzz @4bbu6j5885o3gpv6ss8找到的的CDN和@moozik @BoneAsh @wangyusen提交的代码

douyu.py

@@ -1,17 +1,31 @@
-# 获取斗鱼直播间的真实流媒体地址，默认最高画质。
+# 获取斗鱼直播间的真实流媒体地址，默认最高画质
+# 使用 https://github.com/wbt5/real-url/issues/185 中两位大佬@wjxgzz @4bbu6j5885o3gpv6ss8找到的的CDN，在此感谢！
 import hashlib
 import re
 import time
-import json
 
 import execjs
 import requests
 
 
 class DouYu:
+    """
+    可用来替换返回链接中的主机部分
+    两个阿里的CDN：
+    dyscdnali1.douyucdn.cn
+    dyscdnali3.douyucdn.cn
+    墙外不用带尾巴的akm cdn：
+    hls3-akm.douyucdn.cn
+    hlsa-akm.douyucdn.cn
+    hls1a-akm.douyucdn.cn
+    """
 
     def __init__(self, rid):
-        # 房间号通常为1~7位纯数字，浏览器地址栏中看到的房间号不一定是真实rid.
+        """
+        房间号通常为1~7位纯数字，浏览器地址栏中看到的房间号不一定是真实rid.
+        Args:
+            rid:
+        """
         self.did = '10000000000000000000000000001501'
         self.t10 = str(int(time.time()))
         self.t13 = str(int((time.time() * 1000)))
@@ -25,9 +39,6 @@ class DouYu:
         else:
             raise Exception('房间号错误')
 
-    def set_rate(self, rate):
-        self.live_rate = rate
-    
     @staticmethod
     def md5(data):
         return hashlib.md5(data.encode('utf-8')).hexdigest()
@@ -45,9 +56,13 @@ class DouYu:
             'auth': auth
         }
         res = self.s.post(url, headers=headers, data=data).json()
-        if 0 != res['error']:
-            raise Exception(res["msg"])
-        return res['data']['rtmp_url'] + '/' + res['data']['rtmp_live']
+        error = res['error']
+        data = res['data']
+        key = ''
+        if data:
+            rtmp_live = data['rtmp_live']
+            key = re.search(r'(\d{1,7}[0-9a-zA-Z]+)_?\d{0,4}(/playlist|.m3u8)', rtmp_live).group(1)
+        return error, key
 
     def get_js(self):
         result = re.search(r'(function ub98484234.*)\s(var.*)', self.res).group()
@@ -64,16 +79,15 @@ class DouYu:
 
         js = execjs.compile(func_sign)
         params = js.call('sign', self.rid, self.did, self.t10)
-        # params += '&ver=219032101&rid={}&rate=-1'.format(self.rid)
-        params += '&ver=219032101&rid={}&rate={}'.format(self.rid, self.live_rate)
+        params += '&ver=219032101&rid={}&rate=-1'.format(self.rid)
 
         url = 'https://m.douyu.com/api/room/ratestream'
-        res = self.s.post(url, params=params).json()
-        if 0 != res['code']:
-            return None
-        return res['data']['url']
+        res = self.s.post(url, params=params).text
+        key = re.search(r'(\d{1,7}[0-9a-zA-Z]+)_?\d{0,4}(.m3u8|/playlist)', res).group(1)
 
-    def get_pc_js(self, cdn='ws-h5'):
+        return key
+
+    def get_pc_js(self, cdn='ws-h5', rate=0):
         """
         通过PC网页端的接口获取完整直播源。
         :param cdn: 主线路ws-h5、备用线路tct-h5
@@ -96,24 +110,26 @@ class DouYu:
         js = execjs.compile(func_sign)
         params = js.call('sign', self.rid, self.did, self.t10)
 
-        params += '&cdn={}&rate={}'.format(cdn, self.live_rate)
+        params += '&cdn={}&rate={}'.format(cdn, rate)
         url = 'https://www.douyu.com/lapi/live/getH5Play/{}'.format(self.rid)
         res = self.s.post(url, params=params).json()
-        if 0 != res['error']:
-            return None
-        return res['data']['rtmp_url'] + '/' + res['data']['rtmp_live']
+
+        return res
 
     def get_real_url(self):
-        ret = []
-        ret.append(self.get_pre())
-        ret.append(self.get_js())
-        ret.append(self.get_pc_js())
-        return ret
+        error, key = self.get_pre()
+        if error == 0:
+            pass
+        elif error == 102:
+            raise Exception('房间不存在')
+        elif error == 104:
+            raise Exception('房间未开播')
+        else:
+            key = self.get_js()
+        return "http://dyscdnali1.douyucdn.cn/live/{}.flv?uuid=".format(key)
 
 
 if __name__ == '__main__':
     r = input('输入斗鱼直播间号：\n')
     s = DouYu(r)
-    r = input('输入画质(1流畅；2高清；3超清；4蓝光4M；0蓝光8M或10M)：\n')
-    s.set_rate(r)
     print(s.get_real_url())