고급 입력 유효성 검사 및 오류 처리

FastAPI로 AI 프로덕션 배포하기

Matt Eckerle

Software and Data Engineering Leader

고급 입력이 필요한 이유

레스토랑 주문 API
가변 항목 수

class Order(BaseModel):
    item1: str
    item2: str
    item3: str

레스토랑

중첩 Pydantic 모델

from pydantic import BaseModel

class Foo(BaseModel):
    count: int

class Bar(BaseModel):
    foo: Foo

>>> m = Bar(foo={'count': 4})
>>> print(m)
foo=Foo(count=4)

from pydantic import BaseModel
from typing import List


class OrderItem(BaseModel):
    name: str
    quantity: int


class RestaurantOrder(BaseModel):
    customer_name: str
    items: List[OrderItem]

사용자 정의 모델 유효성 검사기

from fastapi import FastAPI
from fastapi.exceptions import (
    RequestValidationError
)
from pydantic import (
    BaseModel,
    model_validator,
)

from typing import List

class OrderItem(BaseModel):
    name: str
    quantity: int

class RestaurantOrder(BaseModel):
    customer_name: str
    items: List[OrderItem]


    @model_validator(mode="after")

    def validate_after(self):
        if len(self.items) == 0:
           raise RequestValidationError(
               "No items in order!"
           )
        return self

{"detail":"No items in order!"}

전역 예외 핸들러

from fastapi import FastAPI
from fastapi.exceptions import RequestValidationError
from fastapi.responses import PlainTextResponse


app = FastAPI()

@app.exception_handler(RequestValidationError)

async def validation_exception_handler(request, exc):
    msg = "Input validation error. See the documentation: http://127.0.0.1:8000/docs"
    return PlainTextResponse(msg, status_code=422)

Input validation error. See the documentation: http://127.0.0.1:8000/docs

연습해 봅시다!

FastAPI로 AI 프로덕션 배포하기