真实链接测试

douyin_parser/main.py

@@ -54,6 +54,7 @@ class DouyinParser:
         """获取真实视频链接"""
         max_retries = 3  # 最大重试次数
         retry_delay = 2  # 重试延迟秒数
+        max_redirects = 10  # 最大重定向次数，防止死循环
         proxies = {"http": self.http_proxy, "https": self.http_proxy} if self.http_proxy else None
         redirect_history = []
 
@@ -67,28 +68,36 @@ class DouyinParser:
                     'Range': 'bytes=0-'
                 }
 
-                current_url = video_url
-                while True:
-                    response = requests.get(current_url, headers=headers, proxies=proxies, allow_redirects=False,
-                                            timeout=60)
+                # 默认使用 allow_redirects=True 获取历史记录
+                response = requests.get(video_url, headers=headers, proxies=proxies, allow_redirects=True, timeout=60)
+
+                if response.history:
+                    redirect_history = [resp.url for resp in response.history]
+                    real_url = response.url
+                else:
+                    # response.history 为空，手动解析重定向
+                    current_url = video_url
+                    for _ in range(max_redirects):  # 限制最大重定向次数
+                        resp = requests.get(current_url, headers=headers, proxies=proxies, allow_redirects=False,
+                                            timeout=60)
+                        new_url = resp.headers.get('Location')
 
-                    if response.status_code in [301, 302, 303, 307, 308]:  # 处理重定向
-                        new_url = response.headers.get('Location')
                         if not new_url:
-                            self.LOG.info("[抖音] 发现重定向但没有 Location 头部，停止解析")
-                            break
+                            break  # 没有新的 Location，停止
+
                         if not new_url.startswith("http"):
                             from urllib.parse import urljoin
                             new_url = urljoin(current_url, new_url)  # 处理相对路径重定向
 
+                        if new_url in redirect_history:
+                            self.LOG.info("[抖音] 检测到循环重定向: %s", new_url)
+                            break  # 避免死循环
+
                         redirect_history.append(new_url)
                         self.LOG.info("[抖音] 发现重定向: %s -> %s", current_url, new_url)
                         current_url = new_url
-                    else:
-                        break  # 没有进一步重定向，停止循环
 
-                real_url = current_url
-                self.LOG.info("[抖音] 最终获取到的链接: %s", real_url)
+                    real_url = current_url
 
                 if redirect_history:
                     self.LOG.info("[抖音] 重定向历史: %s", redirect_history)