初步完成接口

.gitignore

@@ -3,5 +3,10 @@ Download/
 logs/
 douyin_users.db
 douyin_videos.db
+__pycache__/
 
-.DS_Store
+.DS_Store
+
+.vscode/
+.idea/
+*.pyc

api.py

@@ -0,0 +1,19 @@
+from flask import Flask, request, jsonify
+from qushuiyin import qushuiyin
+
+app = Flask(__name__)
+
+
+@app.route("/get_dy_urls", methods=["POST"])
+async def get_dy_urls():
+    data = request.get_json()  # 从请求中获取JSON数据
+    if not data:
+        return jsonify({"error": "No data provided"}), 400
+
+    # 处理数据
+    dy_urls = await qushuiyin(data["share_url"])
+    return jsonify(dy_urls), 200
+
+
+if __name__ == "__main__":
+    app.run(debug=True)

qushuiyin.py

@@ -13,11 +13,9 @@ kwargs = {
 }
 
 
-async def get_video_id():
-    urls = input("请输入抖音作品分享链接：")
-
+async def get_video_id(share_url):
     raw_urls = [
-        urls,
+        share_url,
     ]
 
     # 提取有效URL
@@ -27,17 +25,15 @@ async def get_video_id():
     return await AwemeIdFetcher.get_all_aweme_id(urls)
 
 
-async def main():
-    aweme_id = await get_video_id()
+async def qushuiyin(share_url):
+    aweme_id = await get_video_id(share_url)
     video = await DouyinHandler(kwargs).fetch_one_video(aweme_id=aweme_id[0])
     video_info = video._to_dict()
     cover_url = video_info["cover"]
     music_url = video_info["music_play_url"]
-    video_url = video_info["video_play_addr"]
-    print(f"封面: {cover_url}")
-    print(f"音乐: {music_url}")
-    print(f"视频:{video_url[0]}")
+    video_urls = video_info["video_play_addr"]
+    return {"cover_url": cover_url, "music_url": music_url, "video_urls": video_urls}
 
 
 if __name__ == "__main__":
-    print(asyncio.run(main()))
+    print(asyncio.run(qushuiyin()))

run.py

@@ -0,0 +1,4 @@
+from api import app
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=5000)