55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
"""
|
|
客户管理路由模块
|
|
|
|
提供客户信息的管理功能，包括：
|
|
- 客户列表查询（支持分页、搜索）
|
|
- 创建新客户（自动生成客户编码）
|
|
|
|
路由前缀: /api/customers
|
|
"""
|
|
from fastapi import APIRouter, Depends, Query
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy import select
|
|
from typing import Optional, List
|
|
from datetime import datetime
|
|
|
|
from database.database import get_db_session
|
|
from services.auth_service import get_current_active_user
|
|
from 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)
|