douyu.py

@@ -2,7 +2,6 @@
 import hashlib
 import re
 import time
-import json
 
 import execjs
 import requests
@@ -25,9 +24,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()
@@ -51,8 +47,7 @@ class DouYu:
         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, data['rtmp_url'] + '/' + data['rtmp_live']
+        return error, key
-        # return error, key
 
     def get_js(self):
         result = re.search(r'(function ub98484234.*)\s(var.*)', self.res).group()
@@ -69,17 +64,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=-1'.format(self.rid)
-        params += '&ver=219032101&rid={}&rate={}'.format(self.rid, self.live_rate)
 
         url = 'https://m.douyu.com/api/room/ratestream'
-        res = self.s.post(url, params=params).json()['data']
+        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['url']).group(1)
+        key = re.search(r'(\d{1,7}[0-9a-zA-Z]+)_?\d{0,4}(.m3u8|/playlist)', res).group(1)
 
-        return res['url']
+        return key
-        # return key
 
-    def get_pc_js(self, cdn='ws-h5'):
+    def get_pc_js(self, cdn='ws-h5', rate=0):
         """
         通过PC网页端的接口获取完整直播源。
         :param cdn: 主线路ws-h5、备用线路tct-h5
@@ -102,32 +95,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()['data']
+        res = self.s.post(url, params=params).json()
 
-        return res['rtmp_url'] + '/' + res['rtmp_live']
+        return res
 
     def get_real_url(self):
-        ret = []
+        error, key = self.get_pre()
-        error, url = self.get_pre()
         if error == 0:
-            ret.append(url)
+            pass
         elif error == 102:
             raise Exception('房间不存在')
         elif error == 104:
             raise Exception('房间未开播')
-        # else:
+        else:
-        #     key = self.get_js()
+            key = self.get_js()
-        ret.append(self.get_js())
+        return "http://tx2play1.douyucdn.cn/live/{}.flv?uuid=".format(key)
-        ret.append(self.get_pc_js())
-        return ret
-        # return "http://tx2play1.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())