Pārlūkot izejas kodu

修改视频ID提取方法

zhangliang 3 nedēļas atpakaļ
vecāks
revīzija
d2c3d7a995
1 mainītis faili ar 6 papildinājumiem un 1 dzēšanām
  1. 6 1
      utils/dy_ks_get_url.py

+ 6 - 1
utils/dy_ks_get_url.py

@@ -42,7 +42,12 @@ class Dy_KS:
                     }
                     response = requests.request(url=url, method='GET', headers=headers, allow_redirects=False, timeout=30)
                     location = response.headers.get('Location', None)
-                    video_id = re.search(r'/video/(\d+)/?', location.split('?')[0] if location else url).group(1)
+                    match = re.search(r'/video/(\d+)/?', location.split('?')[0] if location else url)
+                    if match:
+                        video_id = match.group(1)
+                    elif "&vid=" in location:
+                        video_id = re.search(r'vid=(\d+)', location).group(1)
+
                 url = "http://8.217.192.46:8889/crawler/dou_yin/detail"
                 if not video_id or not video_id.strip():
                     return None, None, None