한양과학기술고등학교

# app/schemas.py
from datetime import datetime
from pydantic import BaseModel, ConfigDict

class PostCreate(BaseModel):
    title: str
    body: str = ""
    author_id: int

class PostUpdate(BaseModel):
    title: str | None = None
    body: str | None = None

class PostRead(BaseModel):
    model_config = ConfigDict(from_attributes=True)   # ORM -> 응답
    id: int
    title: str
    body: str
    author_id: int
    created_at: datetime

from sqlalchemy import select
from sqlalchemy.exc import IntegrityError

@router.post("", response_model=PostRead, status_code=201)
def create_post(payload: PostCreate, db: DbDep):
    post = Post(**payload.model_dump())
    db.add(post)
    try:
        db.commit()
    except IntegrityError:
        db.rollback()
        raise HTTPException(409, "제약 조건 위반 (존재하지 않는 author_id 등)")
    db.refresh(post)
    return post

@router.get("", response_model=list[PostRead])
def list_posts(db: DbDep, skip: int = 0, limit: int = 20, q: str | None = None):
    stmt = select(Post)
    if q:
        stmt = stmt.where(Post.title.ilike(f"%{q}%"))
    stmt = stmt.order_by(Post.id.desc()).offset(skip).limit(min(limit, 100))
    return db.scalars(stmt).all()

@router.get("/{post_id}", response_model=PostRead)
def get_post(post_id: int, db: DbDep):
    post = db.get(Post, post_id)
    if post is None:
        raise HTTPException(404, "게시글 없음")
    return post

@router.patch("/{post_id}", response_model=PostRead)
def update_post(post_id: int, payload: PostUpdate, db: DbDep):
    post = db.get(Post, post_id)
    if post is None:
        raise HTTPException(404)
    data = payload.model_dump(exclude_unset=True)   # 보낸 필드만
    for key, value in data.items():
        setattr(post, key, value)
    db.commit()
    db.refresh(post)
    return post

@router.delete("/{post_id}", status_code=204)
def delete_post(post_id: int, db: DbDep):
    post = db.get(Post, post_id)
    if post is None:
        raise HTTPException(404)
    db.delete(post)
    db.commit()

def get_or_404(db: Session, model, pk):
    obj = db.get(model, pk)
    if obj is None:
        raise HTTPException(404, f"{model.__name__} not found")
    return obj