From 72f84fdee714bcf38f6d828eb538f3dab4c94c7e Mon Sep 17 00:00:00 2001
From: ihmily <961532186@qq.com>
Date: Sat, 14 Sep 2024 19:04:24 +0800
Subject: [PATCH] fix huya game live record

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

diff --git a/spider.py b/spider.py
index f632893..1e1fc31 100644
--- a/spider.py
+++ b/spider.py
@@ -455,19 +455,27 @@ def get_huya_app_stream_url(url: str, proxy_addr: Union[str, None] = None, cooki
             flv_anti_code = i['sFlvAntiCode']
             s_hls_url = i['sHlsUrl']
             hls_anti_code = i['sHlsAntiCode']
+            m3u8_url = f'{s_hls_url}/{stream_name}.m3u8?{hls_anti_code}'
+            flv_url = f'{s_flv_url}/{stream_name}.flv?{flv_anti_code}'
             play_url_list.append(
                 {
                     'cdn_type': cdn_type,
-                    'm3u8_url': f'{s_hls_url}/{stream_name}.m3u8?{hls_anti_code}',
-                    'flv_url': f'{s_flv_url}/{stream_name}.flv?{flv_anti_code}',
+                    'm3u8_url': m3u8_url,
+                    'flv_url': flv_url,
                 }
             )
+        flv_url = 'https://' + play_url_list[0]['flv_url'].split('://')[1]
+        try:
+            record_url = get_req(flv_url, proxy_addr=proxy_addr, headers=headers, redirect_url=True, timeout=15)
+        except TimeoutError:
+            record_url = flv_url
+
         return {
             'anchor_name': anchor_name,
             'is_live': True,
             'm3u8_url': play_url_list[0]['m3u8_url'],
             'flv_url': play_url_list[0]['flv_url'],
-            'record_url': play_url_list[0]['flv_url'],
+            'record_url': record_url,
         }
 
 
@@ -2514,4 +2522,4 @@ if __name__ == '__main__':
     # print(get_shiguang_stream_url(room_url, proxy_addr=''))
     # print(get_yingke_stream_url(room_url, proxy_addr=''))
     # print(get_yinbo_stream_url(room_url, proxy_addr=''))
-    # print(get_zhihu_stream_url(room_url, proxy_addr=''))
+    # print(get_zhihu_stream_url(room_url, proxy_addr=''))
\ No newline at end of file