diff --git a/.gitignore b/.gitignore index 8bfe434..1ca849a 100644 --- a/.gitignore +++ b/.gitignore @@ -3,5 +3,10 @@ Download/ logs/ douyin_users.db douyin_videos.db +__pycache__/ -.DS_Store \ No newline at end of file +.DS_Store + +.vscode/ +.idea/ +*.pyc \ No newline at end of file diff --git a/api.py b/api.py new file mode 100644 index 0000000..c7cf453 --- /dev/null +++ b/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) diff --git a/qushuiyin.py b/qushuiyin.py index 4a54db7..d2f4967 100644 --- a/qushuiyin.py +++ b/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())) diff --git a/requirements.txt b/requirements.txt index 300153b..66117f7 100644 Binary files a/requirements.txt and b/requirements.txt differ diff --git a/run.py b/run.py new file mode 100644 index 0000000..6026ae6 --- /dev/null +++ b/run.py @@ -0,0 +1,4 @@ +from api import app + +if __name__ == "__main__": + app.run(host="0.0.0.0", port=5000)