prozy-anywhere / main.py
ka1kuk's picture
Update main.py
466745f
from fastapi import FastAPI, HTTPException, Path, Response, Request
from fastapi.middleware.cors import CORSMiddleware
import httpx
app = FastAPI()
# Add CORS middleware
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.get("/{url:path}")
async def proxy(url: str = Path(..., title="The URL to be proxied", description="The entire URL to be proxied")):
headers = {
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
}
# Ensure the URL has a valid format
if not url.startswith(('http://', 'https://')):
raise HTTPException(status_code=400, detail="Invalid URL format")
async with httpx.AsyncClient() as client:
try:
response = await client.get(url, headers=headers)
return Response(content=response.content, media_type=response.headers.get('Content-Type'))
except httpx.RequestError as e:
raise HTTPException(status_code=500, detail=f"Error fetching URL: {e}")
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=8000)