File size: 1,436 Bytes
70ca71a e6927ac 70ca71a |
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 |
from fastapi.responses import StreamingResponse, JSONResponse
from fastapi import FastAPI, Response, Request, Query
from fastapi.middleware.cors import CORSMiddleware
from dotenv import load_dotenv
import requests
import os
load_dotenv()
app = FastAPI()
# uvicorn main:app --reload
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.get("/{path:path}")
async def redirect(request:Request, path: str):
try:
base_url = os.environ.get("BASE_URL")
query_params = request.query_params
provider = query_params.get("provider")
base_url = base_url.replace("<provider>", provider)
full_url = f"{base_url}/{path}"
if query_params:
full_url += "?" + "&".join([f"{key}={value}" for key, value in query_params.items()])
auth_token = os.environ.get("API_KEY")
headers = { "Authorization": f"Bearer {auth_token}" }
response = requests.get(full_url, stream=True, headers=headers)
if response.status_code == 404:
return JSONResponse({ "msg": "bro, i think you provider is wrong" })
def stream_content():
for chunk in response.iter_content(chunk_size=1024):
yield chunk
content_type = response.headers.get('content-type', 'application/octet-stream')
return StreamingResponse(stream_content(), media_type=content_type)
except Exception as err:
print(f"{err} [ERROR]")
return JSONResponse({ "msg": "try hard bro" })
|