qushuiyin/api.py

48 lines
1.3 KiB
Python

from flask import Flask, request, jsonify, Response
import requests
from qushuiyin import clean_mask
import urllib.parse
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")
print(video_url)
# 检查视频链接是否提供
if not video_url:
return Response("No video URL provided", status=400)
headers = {
"User-Agent": "Mozilla/5.0 (iPhone; CPU iPhone OS 16_6 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.6 Mobile/15E148 Safari/604.1 Edg/122.0.0.0"
}
# 发送请求到视频链接
try:
response = requests.get(video_url, headers=headers, 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)