From 07099f559216805a999ece5e5e92892433cc5e28 Mon Sep 17 00:00:00 2001
From: Chauncey Chen <32534829+colorsakura@users.noreply.github.com>
Date: Sun, 24 Oct 2021 10:18:26 +0800
Subject: [PATCH] =?UTF-8?q?=E7=BB=9F=E4=B8=80=E6=A0=BC=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bilibili.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/bilibili.py b/bilibili.py
index 9369785..638d485 100644
--- a/bilibili.py
+++ b/bilibili.py
@@ -27,7 +27,7 @@ class BiliBili:
             print(f'bilibili {rid} 未开播')
         self.real_room_id = res['data']['room_id']
 
-    def get_stream_address(self, current_qn: int = 10000) -> list:
+    def get_real_url(self, current_qn: int = 10000) -> list:
         url = 'https://api.live.bilibili.com/xlive/web-room/v2/index/getRoomPlayInfo'
         param = {
             'device': 'pc',
@@ -69,7 +69,15 @@ class BiliBili:
             return stream_url_list
 
 
+def get_real_url(rid):
+    try:
+        bilibili = BiliBili(rid)
+        return bilibili.get_real_url()
+    except Exception as e:
+        print('Exception：', e)
+        return False
+
+
 if __name__ == '__main__':
-    rid = input('请输入bilibili直播房间号：\n')
-    bilibili = BiliBili(rid)
-    print(bilibili.get_stream_address())
+    r = input('请输入bilibili直播房间号：\n')
+    print(get_real_url(r))