在 FastAPI 中设置 CORS 头

在 FastAPI 中设置 CORS 头

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

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

app = 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_url

    response.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 头。