增加小区楼栋的接口

app/api/endpoints/community_building.py

@@ -0,0 +1,109 @@
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+from typing import List
+from app.models.community_building import (
+    CommunityBuildingDB,
+    CommunityBuildingCreate,
+    CommunityBuildingUpdate,
+    CommunityBuildingInfo
+)
+from app.models.database import get_db
+from app.api.deps import get_admin_user
+from app.models.user import UserDB
+from app.core.response import success_response, error_response, ResponseModel
+
+router = APIRouter()
+
+@router.get("/list/{community_id}", response_model=ResponseModel)
+async def get_community_buildings(
+    community_id: int,
+    skip: int = 0,
+    limit: int = 20,
+    db: Session = Depends(get_db)
+):
+    """获取小区的楼栋列表"""
+    buildings = db.query(CommunityBuildingDB).filter(
+        CommunityBuildingDB.community_id == community_id
+    ).order_by(
+        CommunityBuildingDB.building_number
+    ).offset(skip).limit(limit).all()
+    
+    return success_response(data=[
+        CommunityBuildingInfo.model_validate(b) for b in buildings
+    ])
+
+@router.post("", response_model=ResponseModel)
+async def create_building(
+    building: CommunityBuildingCreate,
+    db: Session = Depends(get_db),
+    admin: UserDB = Depends(get_admin_user)
+):
+    """创建楼栋（管理员）"""
+    # 检查是否已存在相同编号的楼栋
+    exists = db.query(CommunityBuildingDB).filter(
+        CommunityBuildingDB.community_id == building.community_id,
+        CommunityBuildingDB.building_number == building.building_number
+    ).first()
+    
+    if exists:
+        return error_response(code=400, message="该楼栋编号已存在")
+    
+    db_building = CommunityBuildingDB(**building.model_dump())
+    db.add(db_building)
+    
+    try:
+        db.commit()
+        db.refresh(db_building)
+        return success_response(data=CommunityBuildingInfo.model_validate(db_building))
+    except Exception as e:
+        db.rollback()
+        return error_response(code=500, message=f"创建失败: {str(e)}")
+
+@router.put("/{building_id}", response_model=ResponseModel)
+async def update_building(
+    building_id: int,
+    building: CommunityBuildingUpdate,
+    db: Session = Depends(get_db),
+    admin: UserDB = Depends(get_admin_user)
+):
+    """更新楼栋信息（管理员）"""
+    db_building = db.query(CommunityBuildingDB).filter(
+        CommunityBuildingDB.id == building_id
+    ).first()
+    
+    if not db_building:
+        return error_response(code=404, message="楼栋不存在")
+    
+    update_data = building.model_dump(exclude_unset=True)
+    for key, value in update_data.items():
+        setattr(db_building, key, value)
+    
+    try:
+        db.commit()
+        db.refresh(db_building)
+        return success_response(data=CommunityBuildingInfo.model_validate(db_building))
+    except Exception as e:
+        db.rollback()
+        return error_response(code=500, message=f"更新失败: {str(e)}")
+
+@router.delete("/{building_id}", response_model=ResponseModel)
+async def delete_building(
+    building_id: int,
+    db: Session = Depends(get_db),
+    admin: UserDB = Depends(get_admin_user)
+):
+    """删除楼栋（管理员）"""
+    db_building = db.query(CommunityBuildingDB).filter(
+        CommunityBuildingDB.id == building_id
+    ).first()
+    
+    if not db_building:
+        return error_response(code=404, message="楼栋不存在")
+    
+    try:
+        db.delete(db_building)
+        db.commit()
+        return success_response(message="删除成功")
+    except Exception as e:
+        db.rollback()
+        return error_response(code=500, message=f"删除失败: {str(e)}") 

app/main.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from app.api.endpoints import user, address, community, station, order, coupon
+from app.api.endpoints import user, address, community, station, order, coupon, community_building
 from app.models.database import Base, engine
 from fastapi.exceptions import RequestValidationError
 from fastapi.responses import JSONResponse
@@ -29,6 +29,7 @@ app.add_middleware(
 app.include_router(user.router, prefix="/api/user", tags=["用户"])
 app.include_router(address.router, prefix="/api/address", tags=["配送地址"])
 app.include_router(community.router, prefix="/api/community", tags=["社区"])
+app.include_router(community_building.router, prefix="/api/community/building", tags=["社区楼栋"])
 app.include_router(station.router, prefix="/api/station", tags=["驿站"])
 app.include_router(order.router, prefix="/api/order", tags=["订单"])
 app.include_router(coupon.router, prefix="/api/coupon", tags=["优惠券"])

app/models/community_building.py

@@ -0,0 +1,40 @@
+from sqlalchemy import Column, Integer, String, ForeignKey, DateTime
+from sqlalchemy.sql import func
+from pydantic import BaseModel, Field
+from typing import Optional
+from datetime import datetime
+from .database import Base
+
+# 数据库模型
+class CommunityBuildingDB(Base):
+    __tablename__ = "community_buildings"
+
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    community_id = Column(Integer, ForeignKey("communities.id"), index=True)
+    building_name = Column(String(50), nullable=False)  # 楼栋名称，如"1号楼"
+    building_number = Column(String(20), nullable=False)  # 楼栋编号，如"A1"
+    create_time = Column(DateTime(timezone=True), server_default=func.now())
+    update_time = Column(DateTime(timezone=True), onupdate=func.now())
+
+    class Config:
+        unique_together = [("community_id", "building_number")]
+
+# Pydantic 模型
+class CommunityBuildingCreate(BaseModel):
+    community_id: int
+    building_name: str = Field(..., max_length=50)
+    building_number: str = Field(..., max_length=20)
+
+class CommunityBuildingUpdate(BaseModel):
+    building_name: Optional[str] = Field(None, max_length=50)
+    building_number: Optional[str] = Field(None, max_length=20)
+
+class CommunityBuildingInfo(BaseModel):
+    id: int
+    community_id: int
+    building_name: str
+    building_number: str
+    create_time: datetime
+    
+    class Config:
+        from_attributes = True