fix: update rednote live stream fetch

README.md

@@ -144,8 +144,7 @@ https://www.yy.com/22490906/22490906
 B站:
 https://live.bilibili.com/320
 
-小红书（推荐使用主页地址):
+小红书（直播间分享地址):
-https://www.xiaohongshu.com/user/profile/6330049c000000002303c7ed?appuid=5f3f478a00000000010005b3
 http://xhslink.com/xpJpfM
 
 bigo直播:

src/spider.py

@@ -683,54 +683,7 @@ async def get_bilibili_stream_data(url: str, qn: str = '10000', platform: str =
         extra = stream_data['url_info'][0]['extra']
         m3u8_url = host + base_url + extra
         return m3u8_url
-
+    
-
-@trace_error_decorator
-async def get_xhs_stream_url_profile(
-        url: str, proxy_addr: OptionalStr = None, cookies: OptionalStr = None, author: OptionalStr = None
-) -> dict:
-    headers = {
-        'User-Agent': 'ios/7.830 (ios 17.0; ; iPhone 15 (A2846/A3089/A3090/A3092))',
-        'xy-common-params': 'platform=iOS&sid=session.1722166379345546829388',
-        'referer': 'https://app.xhs.cn/',
-    }
-    if cookies:
-        headers['Cookie'] = cookies
-    host_id = get_params(url, 'host_id')
-    user_id = re.search('/user/profile/(.*?)(?=/|\\?|$)', url)
-    user_id = user_id.group(1) if user_id else host_id
-    result = {"anchor_name": '', "is_live": False}
-    if user_id:
-        params = {'user_id_list': user_id}
-        app_api = f'https://live-room.xiaohongshu.com/api/sns/v1/live/user_status?{urllib.parse.urlencode(params)}'
-        json_str = await async_req(app_api, proxy_addr=proxy_addr, headers=headers)
-        json_data = json.loads(json_str)
-        if json_data.get('data'):
-            live_link = json_data['data'][0]['live_link']
-            anchor_name = get_params(live_link, "host_nickname")
-            flv_url = get_params(live_link, "flvUrl")
-            room_id = flv_url.split('live/')[1].split('.')[0]
-            flv_url = f'http://live-source-play.xhscdn.com/live/{room_id}.flv'
-            m3u8_url = flv_url.replace('.flv', '.m3u8')
-            result |= {
-                "anchor_name": anchor_name,
-                "is_live": True,
-                "flv_url": flv_url,
-                "m3u8_url": m3u8_url,
-                'record_url': flv_url
-            }
-        if not result["anchor_name"]:
-            if author:
-                result['anchor_name'] = author
-            else:
-                html_str = await async_req(url, proxy_addr=proxy_addr, headers=headers)
-                json_str = re.search('window.__INITIAL_STATE__=(.*?)</script>', html_str, re.S).group(1)
-                json_data = json.loads(json_str)
-                anchor_name = json_data['profile']['userInfo']['nickname']
-                result['anchor_name'] = anchor_name
-    return result
-
-
 @trace_error_decorator
 async def get_xhs_stream_url(url: str, proxy_addr: OptionalStr = None, cookies: OptionalStr = None) -> dict:
     headers = {
@@ -741,38 +694,49 @@ async def get_xhs_stream_url(url: str, proxy_addr: OptionalStr = None, cookies:
     if cookies:
         headers['Cookie'] = cookies
 
-    if 'xhslink.com' in url:
+    if "xhslink.com" in url:
         url = await async_req(url, proxy_addr=proxy_addr, headers=headers, redirect_url=True)
 
+    host_id = get_params(url, "host_id")
+    user_id = re.search("/user/profile/(.*?)(?=/|\\?|$)", url)
+    user_id = user_id.group(1) if user_id else host_id
     result = {"anchor_name": '', "is_live": False}
-    room_id = re.search('/livestream/(.*?)(?=/|\\?|$)', url)
+    html_str = await async_req(url, proxy_addr=proxy_addr, headers=headers)
-    if room_id:
+    match_data = re.search("<script>window.__INITIAL_STATE__=(.*?)</script>", html_str)
-        html_str = await async_req(url, proxy_addr=proxy_addr, headers=headers)
+
-        json_str = re.search('window.__INITIAL_STATE__=(.*?)</script>', html_str, re.S).group(1)
+    if match_data:
+        json_str = match_data.group(1).replace("undefined", "null")
         json_data = json.loads(json_str)
-        room_data = json_data['liveStream']['roomData']
+
-        anchor_name = room_data['hostInfo']['nickName']
+        if json_data.get("liveStream"):
-        live_status = json_data['liveStream']['liveStatus']
+            stream_data = json_data["liveStream"]
-        result['anchor_name'] = anchor_name
+            if stream_data.get("liveStatus") == "success":
-        if live_status == 'end':
+                room_info = stream_data["roomData"]["roomInfo"]
-            return await get_xhs_stream_url_profile(url, proxy_addr, cookies, anchor_name)
+                title = room_info.get("roomTitle")
-        title = room_data['roomInfo']['roomTitle']
+                if title and "回放" not in title:
-        play_data = json.loads(room_data['roomInfo']['pullConfig'])
+                    live_link = room_info["deeplink"]
-        m3u8_url_list = []
+                    anchor_name = get_params(live_link, "host_nickname")
-        flv_url_list = []
+                    flv_url = get_params(live_link, "flvUrl")
-        for item in play_data['h264']:
+                    room_id = flv_url.split('live/')[1].split('.')[0]
-            play_url = item['master_url']
+                    flv_url = f"http://live-source-play.xhscdn.com/live/{room_id}.flv"
-            if play_url.endswith('.m3u8'):
+                    m3u8_url = flv_url.replace('.flv', '.m3u8')
-                m3u8_url_list.append(play_url)
+                    result |= {
-            else:
+                        "anchor_name": anchor_name,
-                flv_url_list.append(play_url)
+                        "is_live": True,
-        flv_url = flv_url_list[0]
+                        "title": title,
-        m3u8_url = m3u8_url_list[0]
+                        "flv_url": flv_url,
-        record_url = m3u8_url if m3u8_url else flv_url
+                        "m3u8_url": m3u8_url,
-        result |= {'is_live': True, "title": title, "flv_url": flv_url, "m3u8_url": m3u8_url, 'record_url': record_url}
+                        'record_url': flv_url
-        return result
+                    }
-    else:
+                    return result
-        return await get_xhs_stream_url_profile(url, proxy_addr, cookies)
+
+    profile_url = f"https://www.xiaohongshu.com/user/profile/{user_id}"
+    html_str = await async_req(profile_url, proxy_addr=proxy_addr, headers=headers)
+    anchor_name = re.search("<title>@(.*?) 的个人主页</title>", html_str)
+    if anchor_name:
+        result["anchor_name"] = anchor_name.group(1)
+
+    return result
 
 
 @trace_error_decorator