代码初始化

utils/utils.py

@@ -0,0 +1,28 @@
+import hmac
+import hashlib
+import json
+
+SECRET_KEY = "OArmXWpMeX8n"  # 小程序中使用的 key
+
+
+def generate_signature(data: dict, key: str = SECRET_KEY) -> str:
+    """
+    生成签名
+    :param data: 请求数据 (dict)
+    :param key: 签名密钥
+    :return: 签名字符串
+    """
+    data_str = json.dumps(data, separators=(',', ':'), ensure_ascii=False)
+    return hmac.new(key.encode(), data_str.encode(), hashlib.sha256).hexdigest()
+
+
+def verify_signature(data: dict, client_signature: str, key: str = SECRET_KEY) -> bool:
+    """
+    校验签名
+    :param data: 请求数据 (dict)
+    :param client_signature: 客户端提供的签名
+    :param key: 签名密钥
+    :return: 签名是否有效
+    """
+    server_signature = generate_signature(data, key)
+    return hmac.compare_digest(server_signature, client_signature)

video/iqiyi.py

@@ -1,8 +1,9 @@
 from fastapi import APIRouter, Query, HTTPException
 import requests
+from ..utils.utils import generate_signature
+import time
 
 routerVideo = APIRouter()
-
 # 第三方API地址
 EXTERNAL_API_URL = "https://api.suyanw.cn/api/Iqiyi_video_search.php"
 
@@ -38,7 +39,7 @@ async def video_search(
 
 
 @routerVideo.get("/video-recommend/")
-def video_search(
+async def video_search(
         source: str = Query(..., description="资源类型"),
         search_type: str = Query(..., description="返回格式，默认text可选json")
 ):
@@ -66,3 +67,45 @@ def video_search(
     except requests.exceptions.RequestException as e:
         # 处理请求错误
         raise HTTPException(status_code=500, detail=f"请求第三方API出错: {str(e)}")
+
+
+@routerVideo.post("/video/search")
+def search_videos(date: str = Query(..., description="时间"),
+                  videoName: str = Query(..., description="视频名称")):
+    """
+    搜索视频接口
+    :param date: 时间
+    :param videoName: 视频名称
+    :return: 符合条件的视频列表
+    """
+    api_url = "https://api.q168.vip/api/search"
+    params = {
+        date: date,
+        videoName: videoName
+    }
+    # 签名
+    sign = generate_signature(params)
+
+    # 设置请求头
+    headers = {
+        "Sign": sign,
+        "Accept": "application/json, text/plain, */*",
+        "origin": "https://wantwatch.me"
+    }
+
+    try:
+        # 发送请求到第三方API
+        response = requests.post(api_url, data=params, headers=headers)
+        response.raise_for_status()  # 确保HTTP请求成功
+
+        # 返回第三方API的结果
+        return response.json()
+
+    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, "我是刑警")