初始化

2026-04-13 14:22:31 +08:00
commit 7cf0a75603
78 changed files with 10702 additions and 0 deletions
--- a/backend/app/routers/categories.py
+++ b/backend/app/routers/categories.py
@@ -0,0 +1,559 @@
+from typing import Optional, List, Dict, Any
+from fastapi import APIRouter, Depends, HTTPException, Query, status
+from sqlalchemy.orm import Session
+from pydantic import BaseModel, Field
+from datetime import datetime
+from decimal import Decimal
+import json
+
+from app.database import get_db
+from app.models import User, ProductCategory, OperationLog
+from app.routers.auth import get_current_user, get_current_admin
+
+router = APIRouter(prefix="/categories", tags=["产品分类管理"])
+
+
+# Pydantic模型
+class CategoryBase(BaseModel):
+    name: str
+    code: Optional[str] = None
+    parent_id: Optional[int] = None
+    commission_rate: Decimal = Field(default=0.03)
+    monthly_rebate: Decimal = Field(default=0.10)
+    quarterly_rebate: Optional[Decimal] = None
+    is_main_product: int = Field(default=0)
+    sort_order: int = Field(default=0)
+
+
+class CategoryCreate(CategoryBase):
+    pass
+
+
+class CategoryUpdate(BaseModel):
+    name: Optional[str] = None
+    code: Optional[str] = None
+    parent_id: Optional[int] = None
+    commission_rate: Optional[Decimal] = None
+    monthly_rebate: Optional[Decimal] = None
+    quarterly_rebate: Optional[Decimal] = None
+    is_main_product: Optional[int] = None
+    sort_order: Optional[int] = None
+    status: Optional[int] = None
+
+
+class CategoryImportItem(BaseModel):
+    name: str
+    code: str
+    parent_code: Optional[str] = None
+    commission_rate: Decimal = Field(default=0.03)
+    monthly_rebate: Decimal = Field(default=0.10)
+    quarterly_rebate: Optional[Decimal] = None
+    is_main_product: int = Field(default=0)
+
+
+class CategoryImportData(BaseModel):
+    items: List[CategoryImportItem]
+    source_file: str
+
+
+# 导入预览缓存（实际生产环境应使用Redis等）
+_import_preview_cache: Dict[str, Any] = {}
+
+
+def log_operation(db: Session, user_id: int, action: str, target_type: str, target_id: Optional[int] = None, 
+                  old_value: Optional[str] = None, new_value: Optional[str] = None):
+    """记录操作日志"""
+    log = OperationLog(
+        user_id=user_id,
+        action=action,
+        target_type=target_type,
+        target_id=target_id,
+        old_value=old_value,
+        new_value=new_value
+    )
+    db.add(log)
+    db.commit()
+
+
+def build_category_tree(categories: List[ProductCategory], parent_id: Optional[int] = None) -> List[Dict]:
+    """构建分类树形结构"""
+    tree = []
+    for cat in categories:
+        if cat.parent_id == parent_id:
+            children = build_category_tree(categories, cat.id)
+            node = {
+                "id": cat.id,
+                "name": cat.name,
+                "code": cat.code,
+                "parent_id": cat.parent_id,
+                "commission_rate": str(cat.commission_rate),
+                "monthly_rebate": str(cat.monthly_rebate),
+                "quarterly_rebate": str(cat.quarterly_rebate) if cat.quarterly_rebate else None,
+                "is_main_product": cat.is_main_product,
+                "sort_order": cat.sort_order,
+                "status": cat.status,
+                "source_file": cat.source_file,
+                "import_batch": cat.import_batch,
+                "last_import_at": cat.last_import_at.isoformat() if cat.last_import_at else None,
+                "created_at": cat.created_at.isoformat() if cat.created_at else None,
+                "updated_at": cat.updated_at.isoformat() if cat.updated_at else None,
+                "children": children if children else []
+            }
+            tree.append(node)
+    return sorted(tree, key=lambda x: x["sort_order"])
+
+
+@router.get("", response_model=dict)
+def get_categories(
+    tree: bool = Query(True, description="是否返回树形结构"),
+    status: Optional[int] = None,
+    db: Session = Depends(get_db),
+    current_user: User = Depends(get_current_user)
+):
+    """获取分类列表（树形结构）"""
+    query = db.query(ProductCategory)
+    
+    # 状态筛选
+    if status is not None:
+        query = query.filter(ProductCategory.status == status)
+    
+    categories = query.order_by(ProductCategory.sort_order, ProductCategory.id).all()
+    
+    if tree:
+        data = build_category_tree(categories)
+    else:
+        data = []
+        for cat in categories:
+            data.append({
+                "id": cat.id,
+                "name": cat.name,
+                "code": cat.code,
+                "parent_id": cat.parent_id,
+                "commission_rate": str(cat.commission_rate),
+                "monthly_rebate": str(cat.monthly_rebate),
+                "quarterly_rebate": str(cat.quarterly_rebate) if cat.quarterly_rebate else None,
+                "is_main_product": cat.is_main_product,
+                "sort_order": cat.sort_order,
+                "status": cat.status,
+                "source_file": cat.source_file,
+                "import_batch": cat.import_batch,
+                "last_import_at": cat.last_import_at.isoformat() if cat.last_import_at else None,
+                "created_at": cat.created_at.isoformat() if cat.created_at else None,
+                "updated_at": cat.updated_at.isoformat() if cat.updated_at else None
+            })
+    
+    return {
+        "code": 200,
+        "message": "success",
+        "data": data
+    }
+
+
+@router.get("/{category_id}", response_model=dict)
+def get_category(
+    category_id: int,
+    db: Session = Depends(get_db),
+    current_user: User = Depends(get_current_user)
+):
+    """获取分类详情"""
+    category = db.query(ProductCategory).filter(ProductCategory.id == category_id).first()
+    
+    if not category:
+        raise HTTPException(status_code=404, detail="分类不存在")
+    
+    # 获取父级信息
+    parent_name = None
+    if category.parent_id:
+        parent = db.query(ProductCategory).filter(ProductCategory.id == category.parent_id).first()
+        if parent:
+            parent_name = parent.name
+    
+    # 获取子分类
+    children = db.query(ProductCategory).filter(ProductCategory.parent_id == category_id).all()
+    children_data = [{"id": c.id, "name": c.name, "code": c.code} for c in children]
+    
+    return {
+        "code": 200,
+        "message": "success",
+        "data": {
+            "id": category.id,
+            "name": category.name,
+            "code": category.code,
+            "parent_id": category.parent_id,
+            "parent_name": parent_name,
+            "commission_rate": str(category.commission_rate),
+            "monthly_rebate": str(category.monthly_rebate),
+            "quarterly_rebate": str(category.quarterly_rebate) if category.quarterly_rebate else None,
+            "is_main_product": category.is_main_product,
+            "sort_order": category.sort_order,
+            "status": category.status,
+            "source_file": category.source_file,
+            "import_batch": category.import_batch,
+            "last_import_at": category.last_import_at.isoformat() if category.last_import_at else None,
+            "children": children_data,
+            "created_at": category.created_at.isoformat() if category.created_at else None,
+            "updated_at": category.updated_at.isoformat() if category.updated_at else None
+        }
+    }
+
+
+@router.post("", response_model=dict)
+def create_category(
+    data: CategoryCreate,
+    db: Session = Depends(get_db),
+    current_admin: User = Depends(get_current_admin)
+):
+    """创建分类"""
+    # 检查code是否已存在
+    if data.code:
+        existing = db.query(ProductCategory).filter(ProductCategory.code == data.code).first()
+        if existing:
+            raise HTTPException(status_code=400, detail="分类编码已存在")
+    
+    # 检查父级是否存在
+    if data.parent_id:
+        parent = db.query(ProductCategory).filter(ProductCategory.id == data.parent_id).first()
+        if not parent:
+            raise HTTPException(status_code=400, detail="父级分类不存在")
+    
+    category = ProductCategory(
+        name=data.name,
+        code=data.code,
+        parent_id=data.parent_id,
+        commission_rate=data.commission_rate,
+        monthly_rebate=data.monthly_rebate,
+        quarterly_rebate=data.quarterly_rebate,
+        is_main_product=data.is_main_product,
+        sort_order=data.sort_order,
+        status=1
+    )
+    db.add(category)
+    db.commit()
+    db.refresh(category)
+    
+    # 记录操作日志
+    log_operation(db, current_admin.id, "CREATE_CATEGORY", "category", category.id,
+                  new_value=f"创建分类: {data.name}, code={data.code}")
+    
+    return {
+        "code": 200,
+        "message": "分类创建成功",
+        "data": {
+            "id": category.id,
+            "name": category.name,
+            "code": category.code,
+            "parent_id": category.parent_id,
+            "commission_rate": str(category.commission_rate),
+            "monthly_rebate": str(category.monthly_rebate),
+            "quarterly_rebate": str(category.quarterly_rebate) if category.quarterly_rebate else None,
+            "is_main_product": category.is_main_product,
+            "sort_order": category.sort_order,
+            "status": category.status,
+            "created_at": category.created_at.isoformat() if category.created_at else None
+        }
+    }
+
+
+@router.put("/{category_id}", response_model=dict)
+def update_category(
+    category_id: int,
+    data: CategoryUpdate,
+    db: Session = Depends(get_db),
+    current_admin: User = Depends(get_current_admin)
+):
+    """更新分类"""
+    category = db.query(ProductCategory).filter(ProductCategory.id == category_id).first()
+    
+    if not category:
+        raise HTTPException(status_code=404, detail="分类不存在")
+    
+    # 如果更换父级，检查是否存在且不会形成循环
+    if data.parent_id is not None and data.parent_id != category.parent_id:
+        if data.parent_id == category_id:
+            raise HTTPException(status_code=400, detail="不能将自己设为父级")
+        parent = db.query(ProductCategory).filter(ProductCategory.id == data.parent_id).first()
+        if not parent:
+            raise HTTPException(status_code=400, detail="父级分类不存在")
+        # 检查是否会成为子分类的子级（循环引用）
+        def check_circular(parent_id: int, target_id: int) -> bool:
+            if parent_id == target_id:
+                return True
+            children = db.query(ProductCategory).filter(ProductCategory.parent_id == parent_id).all()
+            for child in children:
+                if check_circular(child.id, target_id):
+                    return True
+            return False
+        
+        if check_circular(category_id, data.parent_id):
+            raise HTTPException(status_code=400, detail="不能将分类设为其子分类的子级")
+    
+    # 检查code是否已存在
+    if data.code and data.code != category.code:
+        existing = db.query(ProductCategory).filter(ProductCategory.code == data.code).first()
+        if existing:
+            raise HTTPException(status_code=400, detail="分类编码已存在")
+    
+    # 记录旧值
+    old_value = f"name={category.name}, code={category.code}, commission_rate={category.commission_rate}"
+    
+    # 更新分类信息
+    if data.name is not None:
+        category.name = data.name
+    if data.code is not None:
+        category.code = data.code
+    if data.parent_id is not None:
+        category.parent_id = data.parent_id
+    if data.commission_rate is not None:
+        category.commission_rate = data.commission_rate
+    if data.monthly_rebate is not None:
+        category.monthly_rebate = data.monthly_rebate
+    if data.quarterly_rebate is not None:
+        category.quarterly_rebate = data.quarterly_rebate
+    if data.is_main_product is not None:
+        category.is_main_product = data.is_main_product
+    if data.sort_order is not None:
+        category.sort_order = data.sort_order
+    if data.status is not None:
+        category.status = data.status
+    
+    db.commit()
+    db.refresh(category)
+    
+    # 记录操作日志
+    new_value = f"name={category.name}, code={category.code}, commission_rate={category.commission_rate}"
+    log_operation(db, current_admin.id, "UPDATE_CATEGORY", "category", category_id,
+                  old_value=old_value, new_value=new_value)
+    
+    return {
+        "code": 200,
+        "message": "分类更新成功",
+        "data": {
+            "id": category.id,
+            "name": category.name,
+            "code": category.code,
+            "parent_id": category.parent_id,
+            "commission_rate": str(category.commission_rate),
+            "monthly_rebate": str(category.monthly_rebate),
+            "quarterly_rebate": str(category.quarterly_rebate) if category.quarterly_rebate else None,
+            "is_main_product": category.is_main_product,
+            "sort_order": category.sort_order,
+            "status": category.status,
+            "updated_at": category.updated_at.isoformat() if category.updated_at else None
+        }
+    }
+
+
+@router.delete("/{category_id}", response_model=dict)
+def delete_category(
+    category_id: int,
+    db: Session = Depends(get_db),
+    current_admin: User = Depends(get_current_admin)
+):
+    """删除分类"""
+    category = db.query(ProductCategory).filter(ProductCategory.id == category_id).first()
+    
+    if not category:
+        raise HTTPException(status_code=404, detail="分类不存在")
+    
+    # 检查是否有子分类
+    children = db.query(ProductCategory).filter(ProductCategory.parent_id == category_id).first()
+    if children:
+        raise HTTPException(status_code=400, detail="该分类下有子分类，请先删除子分类")
+    
+    # 检查是否有关联的业绩记录
+    from app.models import PerformanceRecord
+    records = db.query(PerformanceRecord).filter(PerformanceRecord.category_id == category_id).first()
+    if records:
+        raise HTTPException(status_code=400, detail="该分类已关联业绩记录，无法删除")
+    
+    name = category.name
+    
+    # 删除分类
+    db.delete(category)
+    db.commit()
+    
+    # 记录操作日志
+    log_operation(db, current_admin.id, "DELETE_CATEGORY", "category", category_id,
+                  old_value=f"删除分类: {name}")
+    
+    return {
+        "code": 200,
+        "message": "分类删除成功",
+        "data": None
+    }
+
+
+@router.post("/import", response_model=dict)
+def preview_import_categories(
+    data: CategoryImportData,
+    db: Session = Depends(get_db),
+    current_admin: User = Depends(get_current_admin)
+):
+    """火山引擎清单导入（预览）"""
+    import uuid
+    
+    batch_id = str(uuid.uuid4())[:8]
+    preview_items = []
+    errors = []
+    
+    # 收集现有分类编码
+    existing_codes = {c.code: c for c in db.query(ProductCategory).all() if c.code}
+    
+    # 构建编码到ID的映射（用于父级关联）
+    code_to_id = {}
+    new_code_to_index = {}
+    
+    for idx, item in enumerate(data.items):
+        if not item.code:
+            errors.append({"index": idx, "message": "分类编码不能为空"})
+            continue
+        
+        if not item.name:
+            errors.append({"index": idx, "message": "分类名称不能为空"})
+            continue
+        
+        # 检查编码是否重复（在当前导入数据中）
+        if item.code in new_code_to_index:
+            errors.append({"index": idx, "message": f"编码 '{item.code}' 在导入数据中重复"})
+            continue
+        
+        new_code_to_index[item.code] = idx
+        
+        # 判断是新增还是更新
+        action = "update" if item.code in existing_codes else "create"
+        existing = existing_codes.get(item.code)
+        
+        preview_item = {
+            "index": idx,
+            "code": item.code,
+            "name": item.name,
+            "parent_code": item.parent_code,
+            "commission_rate": str(item.commission_rate),
+            "monthly_rebate": str(item.monthly_rebate),
+            "quarterly_rebate": str(item.quarterly_rebate) if item.quarterly_rebate else None,
+            "is_main_product": item.is_main_product,
+            "action": action,
+            "existing_id": existing.id if existing else None,
+            "existing_name": existing.name if existing else None
+        }
+        preview_items.append(preview_item)
+    
+    # 检查父级编码是否存在
+    for item in preview_items:
+        if item["parent_code"]:
+            if item["parent_code"] not in existing_codes and item["parent_code"] not in new_code_to_index:
+                errors.append({"index": item["index"], "message": f"父级编码 '{item['parent_code']}' 不存在"})
+    
+    # 保存预览数据到缓存
+    preview_data = {
+        "batch_id": batch_id,
+        "source_file": data.source_file,
+        "items": data.items,
+        "preview": preview_items,
+        "errors": errors,
+        "created_at": datetime.now().isoformat()
+    }
+    _import_preview_cache[batch_id] = preview_data
+    
+    return {
+        "code": 200,
+        "message": "预览生成成功",
+        "data": {
+            "batch_id": batch_id,
+            "total": len(data.items),
+            "create_count": len([p for p in preview_items if p["action"] == "create"]),
+            "update_count": len([p for p in preview_items if p["action"] == "update"]),
+            "error_count": len(errors),
+            "preview": preview_items,
+            "errors": errors
+        }
+    }
+
+
+@router.post("/import/confirm", response_model=dict)
+def confirm_import_categories(
+    batch_id: str,
+    db: Session = Depends(get_db),
+    current_admin: User = Depends(get_current_admin)
+):
+    """确认导入"""
+    if batch_id not in _import_preview_cache:
+        raise HTTPException(status_code=404, detail="导入批次不存在或已过期")
+    
+    preview_data = _import_preview_cache[batch_id]
+    
+    if preview_data["errors"]:
+        raise HTTPException(status_code=400, detail="导入数据存在错误，无法确认导入")
+    
+    items = preview_data["items"]
+    source_file = preview_data["source_file"]
+    
+    # 收集现有分类
+    existing_codes = {c.code: c for c in db.query(ProductCategory).all() if c.code}
+    
+    # 第一步：创建/更新所有分类（不处理parent_id）
+    created_items = {}
+    updated_count = 0
+    created_count = 0
+    
+    for item in items:
+        if item.code in existing_codes:
+            # 更新现有分类
+            category = existing_codes[item.code]
+            category.name = item.name
+            category.commission_rate = item.commission_rate
+            category.monthly_rebate = item.monthly_rebate
+            category.quarterly_rebate = item.quarterly_rebate
+            category.is_main_product = item.is_main_product
+            category.source_file = source_file
+            category.import_batch = batch_id
+            category.last_import_at = datetime.now()
+            updated_count += 1
+        else:
+            # 创建新分类
+            category = ProductCategory(
+                name=item.name,
+                code=item.code,
+                commission_rate=item.commission_rate,
+                monthly_rebate=item.monthly_rebate,
+                quarterly_rebate=item.quarterly_rebate,
+                is_main_product=item.is_main_product,
+                source_file=source_file,
+                import_batch=batch_id,
+                last_import_at=datetime.now(),
+                status=1
+            )
+            db.add(category)
+            db.flush()  # 获取ID
+            created_count += 1
+        
+        created_items[item.code] = category
+    
+    # 第二步：处理父级关联
+    for item in items:
+        if item.parent_code:
+            category = created_items[item.code]
+            if item.parent_code in created_items:
+                category.parent_id = created_items[item.parent_code].id
+            elif item.parent_code in existing_codes:
+                category.parent_id = existing_codes[item.parent_code].id
+    
+    db.commit()
+    
+    # 记录操作日志
+    log_operation(db, current_admin.id, "IMPORT_CATEGORIES", "category", None,
+                  new_value=f"导入分类: 创建{created_count}个, 更新{updated_count}个, 来源:{source_file}")
+    
+    # 清理缓存
+    del _import_preview_cache[batch_id]
+    
+    return {
+        "code": 200,
+        "message": "导入成功",
+        "data": {
+            "batch_id": batch_id,
+            "created": created_count,
+            "updated": updated_count,
+            "total": len(items)
+        }
+    }