geMoldInsight/src/inventory/api/customer_routes.py

"""
客户管理路由模块

提供客户信息的管理功能，包括：
- 客户列表查询（支持分页、搜索）
- 创建新客户（自动生成客户编码）
- 更新客户信息
- 删除客户（软删除）

路由前缀: /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.flush()
    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.flush()
    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.flush()
    return {"message": "客户已删除"}