Início / coding / Perguntas / 78340514
Accepted
l22hs
Asked: 2024-04-17 19:06:34 +0800 CST

Autorização de múltiplas contas na interface swagger do FastAPI

  • 772

Criei as rotas 'usuário' e 'loja' respectivamente. Portanto, há uma API onde a conta do 'usuário' está acessível e há uma API onde a conta da 'loja' está acessível.

# user_route.py...
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/api/user/login")
router = APIRouter(
    prefix="/api/user",
)

# shop_route.py...
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/api/shop/login")
router = APIRouter(
    prefix="/api/shop",
)

No entanto, apenas um tipo de conta pode ser conectado no swagger ui. Então pensei em fazer uma interface swagger para roteador 'usuário' e 'loja', respectivamente, mas apenas 'app1' funciona

# main.py...
app1 = FastAPI(
    title="user Auth",
    docs_url="/docs-user"
)

app2 = FastAPI(
    title="user Auth",
    docs_url="/docs-shop"
)
...
app1.include_router(user_router.router)
app2.include_router(shop_router.router)

Existe uma maneira de usar os dois tipos de contas na interface do usuário do Swagger?

oauth-2.0

1 respostas

  1. Best Answer

    Se eu acertar, você tenta alcançar este comportamento:

    insira a descrição da imagem aqui

    Você pode usar este código para conseguir isso:

    from typing import Annotated

from fastapi import APIRouter, Depends, FastAPI
from fastapi.security import OAuth2PasswordBearer



application = FastAPI(
    title="TEST APP"
)


class ShopOAuth2PasswordBearer(OAuth2PasswordBearer):
    pass


class UserOAuth2PasswordBearer(OAuth2PasswordBearer):
    pass


# user_route.py...
oauth2_scheme_user = ShopOAuth2PasswordBearer(tokenUrl="/api/user/login")
router_user = APIRouter(
    prefix="/api/user",
)

# shop_route.py...
oauth2_scheme_shop = UserOAuth2PasswordBearer(tokenUrl="/api/shop/login")
router_shop = APIRouter(
    prefix="/api/shop",
)


@router_user.post("/login")
async def login_user(token: Annotated[str, Depends(oauth2_scheme_user)]):
    return {"token": "user_token"}


@router_shop.post("/login")
async def login_shop(token: Annotated[str, Depends(oauth2_scheme_shop)]):
    return {"token": "shop_token"}

application.include_router(router_user)
application.include_router(router_shop)
    • 1

relate perguntas