실습 1 — Pydantic 틀로 여행 조건 확인하기

import pandas as pd

# 요청 글 10개와 사람이 정리한 정답 조건을 읽는다
trips = pd.read_csv("trip_requests.csv")

print(trips.shape)
# 예산은 일행 전체의 예산(만 원), 관심사는 ; 로 이어 적었다
print(trips[["번호", "목적지", "박수", "어른", "아이", "예산", "관심사"]].to_string(index=False))

from pydantic import BaseModel, Field


class TripRequest(BaseModel):
    """손님의 여행 조건. 칸 이름은 영어로, 뜻은 description 에 한국어로 적는다"""
    destination: str = Field(description="여행 목적지. 도시나 섬 이름")
    nights: int = Field(description="몇 박인지. 2박 3일이면 2", ge=1, le=30)
    adults: int = Field(description="어른 수", ge=1)
    children: int = Field(default=0, description="아이 수. 없으면 0", ge=0)
    budget: int = Field(description="일행 전체 예산(만 원). 150만 원이면 150", ge=0)
    interests: list[str] = Field(description="관심사 낱말 목록. 예: 바다, 맛집")


def row_to_trip(row):
    """표의 한 행(한국어 열 이름)을 틀(영어 칸 이름)에 맞춰 넣는다"""
    return TripRequest(
        destination=row["목적지"],
        nights=row["박수"],
        adults=row["어른"],
        children=row["아이"],
        budget=row["예산"],
        interests=row["관심사"].split(";"),    # "바다;맛집" → ["바다", "맛집"]
    )


first = row_to_trip(trips.iloc[0])     # iloc[0]: 첫째 행
print(repr(first))
# model_dump: 틀의 객체를 보통 사전으로 바꾼다. dump 는 "쏟아 내다"라는 뜻
print(first.model_dump())

from pydantic import ValidationError

# 정답 조건 10행이 모두 틀을 통과하는지 본다
passed = [row_to_trip(row) for _, row in trips.iterrows()]
print("통과한 행:", len(passed), "/", len(trips))

# 일부러 틀린 값을 넣은 네 가지 경우
good = first.model_dump()
bad_cases = {
    "박수가 음수": {**good, "nights": -1},
    "어른이 0명": {**good, "adults": 0},
    "예산이 글자": {**good, "budget": "많이"},
    "목적지 빠짐": {k: v for k, v in good.items() if k != "destination"},
}
for name, values in bad_cases.items():
    try:
        TripRequest(**values)          # ** 는 사전을 이름 붙은 인자로 풀어 넘긴다
        print(name, "→ 통과")
    except ValidationError as e:
        err = e.errors()[0]
        # loc: 어느 칸, type: 어긴 조건의 이름
        print(name, "→ 걸림:", err["loc"][0], err["type"])

import json
from pathlib import Path

# 틀을 JSON 스키마로 바꾼다. 실습 2에서 모델에게 건네지는 설계도가 이것이다
schema = TripRequest.model_json_schema()
for name, spec in schema["properties"].items():
    print(f"{name:12} {spec['type']:8} {spec['description']}")
print("꼭 채울 칸:", schema["required"])

Path("out_ch04_trip_schema.json").write_text(json.dumps(schema, ensure_ascii=False, indent=2), encoding="utf-8")
print("저장: out_ch04_trip_schema.json")