Python FastAPI: 타입 선언의 중요성과 활용 방법

파이썬 FastAPI에서 타입 선언은 코드의 가독성을 높이고 데이터의 유효성을 검사하는 데에 중요한 역할을 합니다. 이번 글에서는 타입 선언의 중요성과 활용 방법에 대해 알아보겠습니다.

예제 1: 경로 매개변수의 타입 선언

from fastapi import FastAPI

app = FastAPI()

@app.get("/items/{item_id}")
async def read_item(item_id: int):
    return {"item_id": item_id}

위 예제에서는 경로 매개변수 item_id의 타입을 int로 선언하고 있습니다. 이를 통해 FastAPI는 해당 매개변수가 정수여야 한다는 것을 알고, 필요한 경우 자동으로 형 변환을 수행합니다.

예제 2: 쿼리 매개변수의 타입 선언

from fastapi import FastAPI

app = FastAPI()

@app.get("/items/")
async def read_item(q: str):
    return {"q": q}

이 예제에서는 쿼리 매개변수 q의 타입을 str로 선언하고 있습니다. 이를 통해 FastAPI는 해당 매개변수가 문자열이어야 한다는 것을 알고, 필요한 경우 자동으로 형 변환을 수행합니다.

예제 3: 응답 모델의 타입 선언

from fastapi import FastAPI
from pydantic import BaseModel

app = FastAPI()

class Item(BaseModel):
    name: str
    description: str = None
    price: float

@app.post("/items/")
async def create_item(item: Item):
    return item

이 예제에서는 응답 모델로 사용될 Item 클래스를 정의하고 있습니다. 이를 통해 FastAPI는 요청 본문의 데이터를 해당 모델에 맞게 검증하고, 필요한 경우 형 변환을 수행한 후 응답으로 반환합니다.