在 FastAPI 中设置 CORS 头

在 FastAPI 中设置 CORS 头

可以使用 fastapi.middleware.cors.CORSMiddleware 来设置 CORS 头信息。

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddlewareapp = FastAPI()# 设置允许的源
origins = ["http://example.com","https://example.com","http://localhost","http://localhost:8000",
]# 添加 CORS 中间件
app.add_middleware(CORSMiddleware,allow_origins=origins,  # 允许的源allow_credentials=True,allow_methods=["*"],  # 允许的方法allow_headers=["*"],  # 允许的头
)@app.post("/remove_background")
async def remove_background(file: Optional[UploadFile] = File(None),response: Response = Response(),  # 这里提供默认值db: Session = Depends(get_db),api_key: Optional[str] = Form(default=None),mode: int = Form(default=1, description="处理模式: 1为默认返回二进制; 2为返回base64; 3为通过图片URL返回base64结果"),url: Optional[str] = Form(default=''),
):# 在这里处理你的逻辑# 例如：# if mode == 1:#     response.body = some_binary_data# elif mode == 2:#     response.body = some_base64_data# elif mode == 3:#     response.body = some_base64_data_from_urlresponse.headers["access-control-allow-origin"] = "*"response.headers["access-control-allow-credentials"] = "true"return response

检查代理服务器配置

如果你使用的是 Nginx 作为代理服务器，确保你的配置没有过滤或修改这些头信息。以下是一个示例配置：

server {listen 80;server_name your_domain.com;location / {proxy_pass http://127.0.0.1:8000;proxy_set_header Host $host;proxy_set_header X-Real-IP $remote_addr;proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;proxy_set_header X-Forwarded-Proto $scheme;# 确保这些头信息被传递proxy_set_header Access-Control-Allow-Origin "*";proxy_set_header Access-Control-Allow-Credentials "true";}
}

总结

通过在 FastAPI 中设置 CORS 中间件和确保代理服务器正确传递头信息，可以解决使用 SwitchyOmega 代理时头信息丢失的问题。确保你的代理配置没有过滤或修改这些头信息，并使用 FastAPI 的 CORS 中间件来统一设置 CORS 头。