From b9dc370e96a5dd3cdc8b6075625c4eec48b3197d Mon Sep 17 00:00:00 2001
From: wbt5 <wubangtu@qq.com>
Date: Sat, 9 Nov 2019 15:05:24 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E5=93=94=E5=93=A9=E5=93=94?=
 =?UTF-8?q?=E5=93=A9=E7=9B=B4=E6=92=AD=E6=BA=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 get_bilibili_real_url.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/get_bilibili_real_url.py b/get_bilibili_real_url.py
index 21a6599..aa93046 100644
--- a/get_bilibili_real_url.py
+++ b/get_bilibili_real_url.py
@@ -9,8 +9,12 @@ import re
 def get_real_rid(rid):
     room_url = 'https://api.live.bilibili.com/room/v1/Room/room_init?id=' + str(rid)
     response = requests.get(url=room_url).json()
-    live_status = response.get('data').get('live_status')
-    room_id = response.get('data').get('room_id')
+    data = response.get('data', 0)
+    if data:
+        live_status = data.get('live_status', 0)
+        room_id = data.get('room_id', 0)
+    else:
+        live_status = room_id = 0
     return live_status, room_id