File size: 1,701 Bytes
fcbfedd 055922d fcbfedd b145a5f fcbfedd 055922d |
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 |
from fastapi import FastAPI, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
import httpx
import asyncio
app = FastAPI()
# Enable CORS
app.add_middleware(
CORSMiddleware,
allow_origins=["*"], # Allows all origins
allow_credentials=True,
allow_methods=["*"], # Allows all methods
allow_headers=["*"], # Allows all headers
)
class ProxyTest(BaseModel):
proxy: str
@app.post("/test-proxy")
async def test_proxy(proxy_test: ProxyTest):
proxy = proxy_test.proxy
test_url = "http://httpbin.org/ip"
timeout = 10
try:
async with httpx.AsyncClient(proxies={"http://": f"http://{proxy}", "https://": f"http://{proxy}"}) as client:
start_time = asyncio.get_event_loop().time()
response = await client.get(test_url, timeout=timeout)
end_time = asyncio.get_event_loop().time()
if response.status_code == 200:
response_time = round((end_time - start_time) * 1000, 2) # Convert to milliseconds
return {
"status": "success",
"message": f"Proxy {proxy} is working",
"response_time": f"{response_time} ms"
}
else:
raise HTTPException(status_code=400, detail=f"Proxy test failed with status code: {response.status_code}")
except httpx.TimeoutException:
raise HTTPException(status_code=408, detail=f"Proxy {proxy} timed out after {timeout} seconds")
except Exception as e:
raise HTTPException(status_code=400, detail=f"Proxy test failed: {str(e)}")
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=7860) |