Update app.py

app.py

@@ -1,3 +1,4 @@
+
 from flask import Flask, request, Response
 import requests
 import os
@@ -7,13 +8,24 @@ app = Flask(__name__)
 # 代理的目标域名
 TARGET_DOMAIN = "https://kusa.pics"
 
-@app.route('/', defaults={'path': ''}, methods=['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'OPTIONS'])
-@app.route('/<path:path>', methods=['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'OPTIONS'])
+@app.route('/', defaults={'path': ''}, methods=['GET', 'POST'])
+@app.route('/<path:path>', methods=['GET', 'POST'])
 def proxy(path):
     """
-    代理所有请求到目标域名。
+    代理 POST 请求到目标域名，GET 请求返回运行状态。
     """
-    target_url = f"{TARGET_DOMAIN}/{path}?{request.query_string.decode()}"
+    # 处理 GET 请求，直接返回 "api is running"
+    if request.method == 'GET':
+        return Response('api is running', status=200, headers={'Content-Type': 'text/plain; charset=utf-8'})
+
+    # 只允许 POST 请求进行代理
+    if request.method != 'POST':
+        return Response('Method Not Allowed', status=405, headers={'Allow': 'GET, POST'})
+
+    # 构造目标 URL
+    target_url = f"{TARGET_DOMAIN}/{path}"
+    if request.query_string:
+        target_url += f"?{request.query_string.decode()}"
 
     try:
         # 构造请求头
@@ -21,7 +33,7 @@ def proxy(path):
         headers.pop('Host', None)
         headers.pop('Content-Length', None)
 
-        # 发送请求
+        # 发送代理请求
         response = requests.request(
             method=request.method,
             url=target_url,
@@ -35,24 +47,25 @@ def proxy(path):
         encoding = response.encoding or 'utf-8'
         content = response.content.decode(encoding, errors='replace')
 
-        # 构造响应
+        # 构造响应头，排除特定头
         excluded_headers = ['content-encoding', 'content-length', 'transfer-encoding', 'connection']
         response_headers = [(name, value) for (name, value) in response.headers.items()
                             if name.lower() not in excluded_headers]
 
-        # 检查原始响应的 Content-Type，如果包含 json，则设置为 application/json
+        # 设置 Content-Type
         original_content_type = response.headers.get('Content-Type', 'text/plain')
         if 'application/json' in original_content_type.lower():
             content_type = 'application/json; charset=utf-8'
         else:
             content_type = f'text/plain; charset={encoding}'
 
+        # 构造响应
         proxy_response = Response(content, response.status_code, headers=response_headers)
         proxy_response.headers['Content-Type'] = content_type
 
         # 添加 CORS 头
         proxy_response.headers['Access-Control-Allow-Origin'] = '*'
-        proxy_response.headers['Access-Control-Allow-Methods'] = 'GET, POST, OPTIONS, PUT, DELETE, PATCH'
+        proxy_response.headers['Access-Control-Allow-Methods'] = 'GET, POST'
         proxy_response.headers['Access-Control-Allow-Headers'] = 'Content-Type, Authorization'
         proxy_response.headers['Access-Control-Allow-Credentials'] = 'true'
 
@@ -66,19 +79,16 @@ def proxy(path):
         print(f"其他错误: {e}")
         return Response(f"代理服务器出错: {e}", status=500)
 
-@app.after_request
-def after_request(response):
+@app.route('/', defaults={'path': ''}, methods=['PUT', 'DELETE', 'PATCH', 'OPTIONS'])
+@app.route('/<path:path>', methods=['PUT', 'DELETE', 'PATCH', 'OPTIONS'])
+def method_not_allowed(path):
     """
-    处理 OPTIONS 请求，添加 CORS 头。
+    拒绝非 GET 和 POST 请求。
     """
-    if request.method == 'OPTIONS':
-        response.headers['Access-Control-Allow-Origin'] = '*'
-        response.headers['Access-Control-Allow-Methods'] = 'GET, POST, OPTIONS, PUT, DELETE, PATCH'
-        response.headers['Access-Control-Allow-Headers'] = 'Content-Type, Authorization'
-        response.headers['Access-Control-Allow-Credentials'] = 'true'
-    return response
+    return Response('Method Not Allowed', status=405, headers={'Allow': 'GET, POST'})
 
 if __name__ == '__main__':
     port = int(os.environ.get("PORT", 7860))
     print(f"Flask 服务器已启动，监听 {port} 端口")
     app.run(debug=True, host='0.0.0.0', port=port)
+