您通常可以使用FastAPI的依赖关系系统来组合依赖关系，并将逻辑移动到更小的可组合依赖关系中。
要同时接受一个API键作为查询参数和头文件，创建一个允许其中一个的依赖项，两者都是可选的，然后确保至少有一个存在（如果两者都存在，则决定您希望优先给予哪一个）：

from fastapi import FastAPI, Depends, Query, Header
from fastapi.exceptions import HTTPException
from typing import Annotated, Optional

app = FastAPI()

users = {
    'random_token': {'id': 1}
}


def token_from_header_or_qs(api_key: str | None = Query(None), x_api_key: str | None = Header(None)) -> str:
    token = api_key or x_api_key
    
    if not token:
        raise HTTPException(status_code=400)
        
    return token
    

def get_current_user(token: Annotated[str, Depends(token_from_header_or_qs)]):
    user = users.get(token)
    
    if not user:
        raise HTTPException(status_code=401)
        
    return user
    
    
@app.get("/")
def protected_endpoint(user: Annotated[dict, Depends(get_current_user)]):
    return user

token_from_header_or_qs函数允许来自任何一个源的令牌，而get_current_user函数只关心验证令牌和获取用户。控制器函数唯一需要关心的是获取有效的用户（如果您希望它们存在于所有或一组路由中，您也可以将这些注册为APIRouter或FastAPI示例本身的依赖项。