From 315ce05dd138001c7f1df8368e4fd03938f59388 Mon Sep 17 00:00:00 2001
From: wbt5 <wubangtu@qq.com>
Date: Sat, 20 Nov 2021 23:46:45 +0800
Subject: [PATCH] =?UTF-8?q?:bug:=20Fix=20=E5=85=AD=E9=97=B4=E6=88=BF?=
 =?UTF-8?q?=E7=9B=B4=E6=92=AD=20-=E4=BF=AE=E5=A4=8D=E6=AD=A3=E5=88=99?=
 =?UTF-8?q?=E5=8C=B9=E9=85=8D=E4=B8=8D=E5=88=B0=E7=9A=84=E9=97=AE=E9=A2=98?=
 =?UTF-8?q?=20-=E4=BC=98=E5=8C=96=E7=9B=B4=E6=92=AD=E5=9C=B0=E5=9D=80?=
 =?UTF-8?q?=E6=8B=BC=E6=8E=A5=E6=96=B9=E5=BC=8F=20-=E4=BC=98=E5=8C=96?=
 =?UTF-8?q?=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 v6cn.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/v6cn.py b/v6cn.py
index fd09c8b..908343e 100644
--- a/v6cn.py
+++ b/v6cn.py
@@ -11,14 +11,14 @@ class V6CN:
 
     def get_real_url(self):
         try:
-            response = requests.get('https://v.6.cn/' + str(self.rid)).text
+            response = requests.get(f'https://v.6.cn/{self.rid}').text
             result = re.findall(r'"flvtitle":"v(\d*?)-(\d*?)"', response)[0]
             uid = result[0]
             flvtitle = 'v{}-{}'.format(*result)
-            response = requests.get('https://rio.6rooms.com/live/?s=' + str(uid)).text
-            hip = 'https://' + re.findall(r'<watchip>(.*\.xiu123\.cn).*</watchip>', response)[0]
-            real_url = [hip + '/' + flvtitle + '/playlist.m3u8', hip + '/httpflv/' + flvtitle]
-        except:
+            response = requests.get(f'https://rio.6rooms.com/live/?s={uid}').text
+            hip = 'https://' + re.search(r'<watchip>(.*\.com).*?</watchip>', response).group(1)
+            real_url = [f'{hip}/{flvtitle}/palylist.m3u8', f'{hip}/httpflv/{flvtitle}']
+        except Exception:
             raise Exception('直播间不存在或未开播')
         return real_url
 
@@ -35,4 +35,3 @@ def get_real_url(rid):
 if __name__ == '__main__':
     r = input('请输入六间房直播房间号：\n')
     print(get_real_url(r))
-