FastAPI에서의 디펜던시 인젝션(Dependency Injection)

# 예시 코드 출처:https://blog.hops.pub/flask-to-fast-api

from fastapi import Header

async def verify_token(x_token: str = Header(...)):
    if x_token != 'fake-super-secret-token':
        raise HTTPException(status_code=401)
    return x_token

async def verify_token_get_user(
    token: str = Depends(verify_token),
    session: Session = Depends(session),
):
    return get_user_from_token(token, session)

auth_api = APIRouter(
    dependencies=[Depends(verify_token)],
)

@auth_api.get('/authorized-ping/')
async def authorized_ping():
    return 'pong'

@auth_api.get('/me/')
async def me(
    u: User = Depends(verify_token_get_user),
):
    return {'name': u.name}

from fastapi import FastAPI, Header, HTTPException, Depends, APIRouter
from typing import Optional

# 비동기 함수로 토큰을 검증하는 디펜던시를 정의합니다. 이 함수는 요청 헤더에서 'x_token'을 추출합니다.
async def verify_token(x_token: str = Header(None)):
    # 토큰이 유효하지 않으면 401 상태 코드를 반환하는 HTTPException을 발생시킵니다.
    if x_token != 'fake-super-secret-token':
        raise HTTPException(status_code=401)
    # 토큰이 유효하면 토큰을 반환합니다.
    return x_token

# 이 함수는 'verify_token' 디펜던시를 사용하여 토큰을 검증하고, 
# 그 결과로 얻은 토큰을 사용하여 유저 정보를 가져오는 로직을 구현합니다.
# 여기서 'session'은 예시 코드이며, 실제로는 데이터베이스 세션을 생성하고 관리하는 로직이 필요합니다.
async def verify_token_get_user(
    token: str = Depends(verify_token),  # 'verify_token' 디펜던시를 통해 토큰을 검증합니다.
    session: Optional[Session] = Depends(None),  # 데이터베이스 세션 관리를 위한 디펜던시(여기서는 단순화를 위해 None으로 설정).
):
    # 'get_user_from_token' 함수를 사용하여 토큰으로부터 유저 정보를 가져옵니다.
    # 이 함수는 예시이며, 실제 구현이 필요합니다.
    return get_user_from_token(token, session)

# APIRouter 인스턴스를 생성하고, 'verify_token' 디펜던시를 이 라우터의 모든 경로에 자동으로 적용합니다.
auth_api = APIRouter(
    dependencies=[Depends(verify_token)],  # 모든 경로에 'verify_token' 디펜던시를 적용합니다.
)

# 'authorized_ping' 엔드포인트를 정의합니다. 이 엔드포인트는 'verify_token' 디펜던시를 통해 토큰 검증이 자동으로 이루어집니다.
@auth_api.get('/authorized-ping/')
async def authorized_ping():
    return 'pong'

# 'me' 엔드포인트를 정의합니다. 이 엔드포인트는 'verify_token_get_user' 디펜던시를 사용하여
# 토큰을 검증하고 유저 정보를 가져옵니다.
@auth_api.get('/me/')
async def me(
    u: User = Depends(verify_token_get_user),  # 'verify_token_get_user' 디펜던시를 통해 유저 정보를 가져옵니다.
):
    # 유저 정보에서 이름을 추출하여 반환합니다.
    return {'name': u.name}