Spaces:
Sleeping
Sleeping
| from flask import Flask, request, Response | |
| import requests | |
| import os | |
| app = Flask(__name__) | |
| TARGET_DOMAIN = os.getenv("TARGET_API") | |
| def proxy(path): | |
| try: | |
| target_url = f"{TARGET_DOMAIN}/{path}" | |
| #print(f"Proxying to: {target_url}") | |
| headers = {key: value for (key, value) in request.headers if key.lower() not in ('host', 'content-length')} | |
| headers['Accept-Encoding'] = 'identity' # 禁止压缩 | |
| resp = requests.request( | |
| method=request.method, | |
| url=target_url, | |
| headers=headers, | |
| data=request.get_data(), | |
| cookies=request.cookies, | |
| params=request.args, | |
| stream=True | |
| ) | |
| excluded_headers = ['content-encoding', 'content-length', 'transfer-encoding', 'connection'] | |
| headers = [(name, value) for (name, value) in resp.raw.headers.items() if name.lower() not in excluded_headers] | |
| #headers.append(('Content-Type', 'application/json; charset=utf-8')) | |
| return Response(resp.iter_content(chunk_size=1024), headers=headers, status=resp.status_code) | |
| except Exception as e: | |
| return str(e), 500 | |
| if __name__ == '__main__': | |
| # 必须指定host和port | |
| app.run(host='0.0.0.0', port=7860, debug=False) |