代码初始化

video/iqiyi.py

@@ -1,7 +1,10 @@
 from fastapi import APIRouter, Query, HTTPException
 import requests
 from pydantic import BaseModel
-from utils.utils import generate_signature
+# from utils.utils import generate_signature
+import hmac
+import hashlib
+import json
 import time
 
 routerVideo = APIRouter()
@@ -74,10 +77,11 @@ async def video_search(
 class SearchVideoRequest(BaseModel):
     date: str  # 时间
     videoName: str  # 视频名称
+    sign: str
 
 
 @routerVideo.post("/video/search")
-async def search_videos(request: SearchVideoRequest):
+def search_videos(request: SearchVideoRequest):
     """
     搜索视频接口
     :param date: 时间
@@ -90,13 +94,17 @@ async def search_videos(request: SearchVideoRequest):
         "videoName": request.videoName
     }
     # 签名
-    sign = generate_signature(params)
+    # data_str = json.dumps(params)
+    # sign = hmac.new("OArmXWpMeX8n".encode('utf-8'), data_str.encode('utf-8'), hashlib.sha256).hexdigest()
+    # sign = generate_signature(params)
 
     # 设置请求头
     headers = {
-        "Sign": sign,
+        "Sign": request.sign,
         "Accept": "application/json, text/plain, */*",
-        "origin": "https://wantwatch.me"
+        "Accept-Encoding": "gzip, deflate, br",  # 加入 br
+        "origin": "https://wantwatch.me",
+        "Content-Type": "application/json; charset=utf-8",
     }
 
     try:
@@ -110,8 +118,3 @@ async def search_videos(request: SearchVideoRequest):
     except requests.exceptions.RequestException as e:
         # 处理请求错误
         raise HTTPException(status_code=500, detail=f"请求第三方API出错: {str(e)}")
-
-
-if __name__ == '__main__':
-    timestamp_ms = int(time.time() * 1000)
-    search_videos(timestamp_ms, "我是刑警")