bilibili.py

@@ -3,70 +3,49 @@
 # qn=250超清
 # qn=400蓝光
 # qn=10000原画
+import re
+
 import requests
 
 
 class BiliBili:
 
     def __init__(self, rid):
-        rid = rid
+        self.rid = rid
-        self.header = {
-            'User-Agent': 'Mozilla/5.0 (iPod; CPU iPhone OS 14_5 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) CriOS/87.0.4280.163 Mobile/15E148 Safari/604.1',
-        }
-        # 先获取直播状态和真实房间号
-        r_url = 'https://api.live.bilibili.com/room/v1/Room/room_init'
-        param = {
-            'id': rid
-        }
-        with requests.Session() as s:
-            res = s.get(r_url, headers=self.header, params=param).json()
-        if res['msg'] == '直播间不存在':
-            raise Exception(f'bilibili {rid} {res["msg"]}')
-        live_status = res['data']['live_status']
-        if live_status != 1:
-            raise Exception(f'bilibili {rid} 未开播')
-        self.real_room_id = res['data']['room_id']
 
-    def get_real_url(self, current_qn: int = 10000) -> list:
+    def get_real_url(self):
-        url = 'https://api.live.bilibili.com/xlive/web-room/v2/index/getRoomPlayInfo'
+        # 先获取直播状态和真实房间号
-        param = {
+        r_url = 'https://api.live.bilibili.com/room/v1/Room/room_init?id={}'.format(self.rid)
-            'device': 'pc',
+        with requests.Session() as s:
-            'platform': 'h5',
+            res = s.get(r_url).json()
-            'scale': 0,
+        code = res['code']
-            'build': '10000',
+        if code == 0:
-            'protocol': '0,1',
+            live_status = res['data']['live_status']
-            'format': '0,1,2',
+            if live_status == 1:
-            'codec': '0,1',
+                room_id = res['data']['room_id']
-            'room_id': self.real_room_id,
+
-            'qn': current_qn
+                def u(pf):
-        }
+                    f_url = 'https://api.live.bilibili.com/xlive/web-room/v1/playUrl/playUrl'
-        with requests.Session() as session:
+                    params = {
-            res = session.get(url, headers=self.header, params=param)
+                        'cid': room_id,
-            res = res.json()
+                        'platform': pf,
-            stream_info = res['data']['playurl_info']['playurl']['stream']
+                        'otype': 'json',
-            qn_max = 0
+                        'quality': 0
-            for data in stream_info:
+                    }
-                accept_qn = data['format'][0]['codec'][0]['accept_qn']
+                    resp = s.get(f_url, params=params).json()
-                for qn in accept_qn:
+                    try:
-                    qn_max = qn if qn > qn_max else qn_max
+                        durl = resp['data']['durl']
-            if qn_max != current_qn:
+                        real_url = durl[0]['url']
-                param['qn'] = qn_max
+                        real_url = re.sub(r'live_(\d+)_(\d+)_\d+.m3u8', r'live_\1_\2.m3u8', real_url)
-                res = session.get(url, headers=self.header, params=param)
+                        return real_url
-                res = res.json()
+                    except KeyError or IndexError:
-                stream_info = res['data']['playurl_info']['playurl']['stream']
+                        raise Exception('获取失败')
-            stream_url_list = []
+
-            for data in stream_info:
+                return u('h5')
-                format_name = data['format'][0]['format_name']
+            else:
-                if format_name == 'flv':
+                raise Exception('未开播')
-                    base_url = data['format'][0]['codec'][0]['base_url']
+        else:
-                    url_info = data['format'][0]['codec'][0]['url_info']
+            raise Exception('房间不存在')
-                    for info in url_info:
-                        host = info['host']
-                        extra = info['extra']
-                        # print(host + base_url + extra)
-                        stream_url_list.append(host + base_url + extra)
-                    break
-            return stream_url_list
 
 
 def get_real_url(rid):

douyu.py

@@ -126,10 +126,8 @@ class DouYu:
             raise Exception('房间未开播')
         else:
             key = self.get_js()
-        real_url = {}
+        return "http://dyscdnali1.douyucdn.cn/live/{}.flv?uuid=".format(key)
-        real_url["flv"] = "http://dyscdnali1.douyucdn.cn/live/{}.flv?uuid=".format(key)
+
-        real_url["x-p2p"] = "http://tx2play1.douyucdn.cn/live/{}.xs?uuid=".format(key)
-        return real_url
 
 if __name__ == '__main__':
     r = input('输入斗鱼直播间号：\n')

requirements.txt

@@ -8,7 +8,7 @@ multidict==4.7.6
 protobuf==3.12.2
 pycryptodome==3.9.8
 PyExecJS==1.5.1
-requests==2.26.0
+requests==2.24.0
 six==1.15.0
 typing-extensions==3.7.4.3
 urllib3==1.26.5