From b3883d18c69e1c89b5305bf2282aa59b228a5f64 Mon Sep 17 00:00:00 2001
From: aka <66773152+akavscode@users.noreply.github.com>
Date: Tue, 21 Sep 2021 06:19:39 +0800
Subject: [PATCH] douyu roomid length fix

---
 douyu.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/douyu.py b/douyu.py
index ca9710c..4f01ff0 100644
--- a/douyu.py
+++ b/douyu.py
@@ -22,7 +22,7 @@ class DouYu:
 
     def __init__(self, rid):
         """
-        房间号通常为1~7位纯数字，浏览器地址栏中看到的房间号不一定是真实rid.
+        房间号通常为1~8位纯数字，浏览器地址栏中看到的房间号不一定是真实rid.
         Args:
             rid:
         """
@@ -32,7 +32,7 @@ class DouYu:
 
         self.s = requests.Session()
         self.res = self.s.get('https://m.douyu.com/' + str(rid)).text
-        result = re.search(r'rid":(\d{1,7}),"vipId', self.res)
+        result = re.search(r'rid":(\d{1,8}),"vipId', self.res)
 
         if result:
             self.rid = result.group(1)
@@ -61,7 +61,7 @@ class DouYu:
         key = ''
         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)
+            key = re.search(r'(\d{1,8}[0-9a-zA-Z]+)_?\d{0,4}(/playlist|.m3u8)', rtmp_live).group(1)
         return error, key
 
     def get_js(self):
@@ -83,7 +83,7 @@ class DouYu:
 
         url = 'https://m.douyu.com/api/room/ratestream'
         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).group(1)
+        key = re.search(r'(\d{1,8}[0-9a-zA-Z]+)_?\d{0,4}(.m3u8|/playlist)', res).group(1)
 
         return key