90 lines
3.3 KiB
Python
90 lines
3.3 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy.orm import Session
|
|
from typing import List
|
|
|
|
from app.database import get_db
|
|
from app.models.dress import Dress
|
|
from app.schemas.dress import DressCreate, DressUpdate, DressResponse, DressListResponse
|
|
from app.utils.response import APIResponse
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("/", response_model=DressResponse)
|
|
def create_dress(dress: DressCreate, db: Session = Depends(get_db)):
|
|
"""创建服装记录"""
|
|
try:
|
|
db_dress = Dress(**dress.dict())
|
|
db.add(db_dress)
|
|
db.commit()
|
|
db.refresh(db_dress)
|
|
return APIResponse.created(data=db_dress, message="服装创建成功")
|
|
except Exception as e:
|
|
db.rollback()
|
|
return APIResponse.error(message=f"服装创建失败: {str(e)}", code=500)
|
|
|
|
@router.get("/", response_model=DressListResponse)
|
|
def get_all_dresses(
|
|
skip: int = 0,
|
|
limit: int = 100,
|
|
db: Session = Depends(get_db)
|
|
):
|
|
"""获取所有服装记录"""
|
|
try:
|
|
dresses = db.query(Dress).offset(skip).limit(limit).all()
|
|
total = db.query(Dress).count()
|
|
return APIResponse.ok(
|
|
data={
|
|
"items": dresses,
|
|
"total": total,
|
|
"page": skip // limit + 1 if limit > 0 else 1,
|
|
"size": limit
|
|
},
|
|
message="服装列表获取成功"
|
|
)
|
|
except Exception as e:
|
|
return APIResponse.error(message=f"获取服装列表失败: {str(e)}", code=500)
|
|
|
|
@router.get("/{dress_id}", response_model=DressResponse)
|
|
def get_dress(dress_id: int, db: Session = Depends(get_db)):
|
|
"""获取单个服装记录"""
|
|
try:
|
|
dress = db.query(Dress).filter(Dress.id == dress_id).first()
|
|
if dress is None:
|
|
return APIResponse.not_found(message=f"未找到ID为{dress_id}的服装")
|
|
return APIResponse.ok(data=dress, message="服装获取成功")
|
|
except Exception as e:
|
|
return APIResponse.error(message=f"获取服装失败: {str(e)}", code=500)
|
|
|
|
@router.put("/{dress_id}", response_model=DressResponse)
|
|
def update_dress(dress_id: int, dress: DressUpdate, db: Session = Depends(get_db)):
|
|
"""更新服装记录"""
|
|
try:
|
|
db_dress = db.query(Dress).filter(Dress.id == dress_id).first()
|
|
if db_dress is None:
|
|
return APIResponse.not_found(message=f"未找到ID为{dress_id}的服装")
|
|
|
|
# 更新服装字段
|
|
for field, value in dress.dict(exclude_unset=True).items():
|
|
setattr(db_dress, field, value)
|
|
|
|
db.commit()
|
|
db.refresh(db_dress)
|
|
return APIResponse.ok(data=db_dress, message="服装更新成功")
|
|
except Exception as e:
|
|
db.rollback()
|
|
return APIResponse.error(message=f"更新服装失败: {str(e)}", code=500)
|
|
|
|
@router.delete("/{dress_id}")
|
|
def delete_dress(dress_id: int, db: Session = Depends(get_db)):
|
|
"""删除服装记录"""
|
|
try:
|
|
db_dress = db.query(Dress).filter(Dress.id == dress_id).first()
|
|
if db_dress is None:
|
|
return APIResponse.not_found(message=f"未找到ID为{dress_id}的服装")
|
|
|
|
db.delete(db_dress)
|
|
db.commit()
|
|
return APIResponse.ok(message="服装删除成功")
|
|
except Exception as e:
|
|
db.rollback()
|
|
return APIResponse.error(message=f"删除服装失败: {str(e)}", code=500) |