qushuiyin/api.py

42 lines
1.1 KiB
Python

from flask import Flask, request, jsonify, Response
import requests
from qushuiyin import clean_mask
app = Flask(__name__)
@app.route("/get_urls", methods=["POST"])
def get_urls():
data = request.get_json() # 从请求中获取JSON数据
if not data:
return jsonify({"error": "No data provided"}), 400
# 处理数据
url = clean_mask(data["share_url"])
print(url)
return jsonify(url), 200
@app.route("/proxy")
def proxy():
# 从请求参数中获取视频链接
video_url = request.args.get("url")
# 检查视频链接是否提供
if not video_url:
return Response("No video URL provided", status=400)
# 发送请求到视频链接
try:
response = requests.get(video_url, stream=True)
response.raise_for_status() # 检查请求是否成功
except requests.RequestException as e:
return Response(f"Error fetching video: {e}", status=500)
# 将视频内容作为响应返回
return Response(response.content, content_type=response.headers["Content-Type"])
if __name__ == "__main__":
app.run(debug=True)