Spaces:
Sleeping
Sleeping
File size: 1,871 Bytes
7ecdd01 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
import asyncio
import json
import urllib.parse
from fastapi.params import Header
from fastapi.responses import JSONResponse, PlainTextResponse
import requests
import fastapi
app = fastapi.FastAPI()
# Define the root endpoint
@app.get("/")
def read_root():
return {"status": "ok"}
# Define the proxy endpoint -> https://hostname/proxy?url=stream_url&headers=headers&output=json/text
@app.get("/proxy")
async def proxy(
url: str = fastapi.Query(..., description="The target URL to fetch"),
headers: str = fastapi.Query(..., description="URL-encoded JSON headers"),
output: str = fastapi.Query("text", description="Output format: json or text"),
):
try:
url = urllib.parse.unquote(url)
decoded_headers = urllib.parse.unquote(headers)
final_headers = json.loads(decoded_headers)
response = await asyncio.to_thread(requests.get, url, headers=final_headers)
if response.status_code == 200:
if output.lower() == "json":
try:
return JSONResponse(response.json())
except json.JSONDecodeError:
return JSONResponse({"error": "Response is not valid JSON"}, status_code=502)
return PlainTextResponse(response.text)
else:
return JSONResponse(
{"error": f"{response.status_code} - {response.reason}"},
status_code=response.status_code
)
except json.JSONDecodeError:
return JSONResponse({"error": "Invalid JSON format for headers."}, status_code=400)
except requests.RequestException as e:
return JSONResponse({"error": str(e)}, status_code=502)
except Exception as e:
return JSONResponse({"error": str(e)}, status_code=500)
# if __name__ == "__main__":
# import uvicorn
# uvicorn.run(app, host="localhost", port=8000)
|