后端模块拆分

src/inventory/api/customer_routes.py

@@ -0,0 +1,92 @@
+"""
+客户管理路由模块
+
+提供客户信息的管理功能，包括：
+- 客户列表查询（支持分页、搜索）
+- 创建新客户（自动生成客户编码）
+- 更新客户信息
+- 删除客户（软删除）
+
+路由前缀: /api/customers
+"""
+from fastapi import APIRouter, Depends, Query, HTTPException
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy import select
+from typing import Optional, List
+from datetime import datetime
+
+from shared.database.database import get_db_session
+from shared.services.auth_service import get_current_active_user, get_current_admin_user
+from shared.models.database import User, Customer
+from .schemas import CustomerCreate, CustomerResponse
+
+router = APIRouter(prefix="/customers", tags=["客户管理"])
+
+
+@router.get("", response_model=List[CustomerResponse])
+async def list_customers(
+    skip: int = Query(0, ge=0),
+    limit: int = Query(20, ge=1, le=100),
+    search: Optional[str] = None,
+    db_session: AsyncSession = Depends(get_db_session),
+    current_user: User = Depends(get_current_active_user)
+):
+    query = select(Customer).where(Customer.is_active == True)
+    if search:
+        query = query.where(Customer.name.ilike(f"%{search}%"))
+    query = query.offset(skip).limit(limit).order_by(Customer.created_at.desc())
+    result = await db_session.execute(query)
+    return [CustomerResponse.from_orm(c) for c in result.scalars().all()]
+
+
+@router.post("", response_model=CustomerResponse, status_code=201)
+async def create_customer(
+    customer_data: CustomerCreate,
+    db_session: AsyncSession = Depends(get_db_session),
+    current_user: User = Depends(get_current_active_user)
+):
+    data = customer_data.dict()
+    if not data.get("code"):
+        data["code"] = f"C{datetime.now().strftime('%Y%m%d%H%M%S')}"
+    
+    customer = Customer(**data)
+    db_session.add(customer)
+    await db_session.commit()
+    await db_session.refresh(customer)
+    return CustomerResponse.from_orm(customer)
+
+
+@router.put("/{customer_id}", response_model=CustomerResponse)
+async def update_customer(
+    customer_id: int,
+    customer_data: CustomerCreate,
+    db_session: AsyncSession = Depends(get_db_session),
+    current_user: User = Depends(get_current_active_user)
+):
+    result = await db_session.execute(select(Customer).where(Customer.id == customer_id))
+    customer = result.scalar_one_or_none()
+    if not customer:
+        raise HTTPException(status_code=404, detail="客户不存在")
+    
+    for key, value in customer_data.dict().items():
+        setattr(customer, key, value)
+    
+    await db_session.commit()
+    await db_session.refresh(customer)
+    return CustomerResponse.from_orm(customer)
+
+
+@router.delete("/{customer_id}")
+async def delete_customer(
+    customer_id: int,
+    db_session: AsyncSession = Depends(get_db_session),
+    current_user: User = Depends(get_current_admin_user)
+):
+    result = await db_session.execute(select(Customer).where(Customer.id == customer_id))
+    customer = result.scalar_one_or_none()
+    if not customer:
+        raise HTTPException(status_code=404, detail="客户不存在")
+    
+    customer.is_active = False
+    await db_session.commit()
+    return {"message": "客户已删除"}