Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, request, jsonify  
+from flask import Flask, request, jsonify, make_response  
 import requests  
 import logging  
 
@@ -9,39 +9,45 @@ app = Flask(__name__)
 def proxy_request(full_url):  
     try:  
         headers = {key: value for key, value in request.headers.items() if key.lower() != 'host'}  
-        headers['User-Agent'] = request.headers.get('User-Agent', 'Mozilla/5.0 (compatible; ProxyBot/1.0)')  # 添加默认User-Agent  
+        headers['User-Agent'] = request.headers.get('User-Agent', 'Mozilla/5.0 (compatible; SpacesProxy/1.0)')  
         response = requests.request(  
             method=request.method,  
-            url=full_url,  # 确保是完整的URL  
+            url=full_url,  
             headers=headers,  
             data=request.get_data(),  
             params=request.args,  
-            timeout=60,  # 延长超时到60秒，防止网络延迟  
-            verify=True  # 确保SSL验证  
+            timeout=60,  
+            stream=True  # 启用流式以处理大响应，如HTML页面  
         )  
-        return response.content, response.status_code, response.headers  
+        return response.iter_content(chunk_size=1024), response.status_code, response.headers  # 返回迭代器以流式输出  
     except requests.exceptions.RequestException as e:  
         logging.error(f"代理请求错误: {e}")  
         return jsonify({"error": str(e)}), 500  
 
 @app.route('/proxy/<path:url>')  
 def handle_proxy(url):  
-    # 确保url是完整的；如果不是，添加协议  
     if not url.startswith('http'):  
         full_url = f"https://{url}"  
     else:  
-        full_url = url  # 如果用户输入了完整的URL，就直接用  
-    content, status_code, headers = proxy_request(full_url)  
-    response = app.response_class(  
-        response=content,  
-        status=status_code,  
-        mimetype=headers.get('Content-Type', 'text/html')  # 自动获取正确的MIME类型  
-    )  
-    # 过滤并传递头部，避免冲突  
+        full_url = url  # 使用传入的完整URL  
+    
+    content_iter, status_code, headers = proxy_request(full_url)  # 获取迭代器  
+    
+    resp = make_response('')  # 初始化响应  
+    resp.status_code = status_code  
     for key, value in headers.items():  
-        if key.lower() not in ['transfer-encoding', 'connection']:  # 忽略可能导致502的头部  
-            response.headers[key] = value  
-    return response  
+        if key.lower() not in ['content-encoding', 'content-length', 'transfer-encoding', 'connection', 'host']:  # 更严格过滤可能冲突的头部  
+            resp.headers[key] = value  
+    resp.mimetype = headers.get('Content-Type', 'text/html')  # 设置正确的MIME类型  
+    
+    # 流式写入内容  
+    def generate():  
+        for chunk in content_iter:  # 逐块输出内容  
+            yield chunk  
+    resp.response = generate()  # 将生成器设置为响应体  
+    resp.headers['Content-Length'] = headers.get('Content-Length', '0')  # 如果可用，设置内容长度  
+    
+    return resp  
 
 if __name__ == '__main__':  
     app.run(host='0.0.0.0', port=7860, debug=True)