KPI通用化: bsc_layer_config表+API+企业选择器+企业CRUD

backend/app/api/bsc_layers.py

@@ -0,0 +1,41 @@
+"""BSC四层配置 API"""
+from fastapi import APIRouter, Depends, Query
+from sqlalchemy.orm import Session
+from typing import List
+
+from app.database import get_db
+from app.auth_middleware import require_auth
+from app.models import BscLayerConfig, Entity
+
+router = APIRouter(prefix="/api/cma/bsc-layers", tags=["BSC层配置"],
+    dependencies=[Depends(require_auth)],
+)
+
+
+@router.get("")
+def list_bsc_layers(entity_id: int = Query(1, description="企业ID"), db: Session = Depends(get_db)):
+    """获取某企业的BSC四层权重配置"""
+    # 验证企业存在
+    entity = db.query(Entity).filter(Entity.id == entity_id).first()
+    if not entity:
+        from fastapi.responses import JSONResponse
+        return JSONResponse(status_code=404, content={"detail": "企业不存在"})
+
+    layers = db.query(BscLayerConfig).filter(
+        BscLayerConfig.entity_id == entity_id
+    ).order_by(BscLayerConfig.id).all()
+
+    return {
+        "entity_id": entity_id,
+        "entity_name": entity.short_name or entity.name,
+        "layers": [
+            {
+                "id": l.id,
+                "layer": l.layer,
+                "weight": float(l.weight),
+                "kpi_count_min": l.kpi_count_min,
+                "kpi_count_max": l.kpi_count_max,
+            }
+            for l in layers
+        ]
+    }

backend/app/api/entities.py

@@ -1,7 +1,8 @@
 """企业实体 API"""
 from fastapi import APIRouter, Depends
 from sqlalchemy.orm import Session
-from typing import List
+from typing import List, Optional
+from pydantic import BaseModel
 
 from app.database import get_db
 from app.auth_middleware import require_auth
@@ -12,6 +13,20 @@ router = APIRouter(prefix="/api/cma/entities", tags=["企业实体"],
 )
 
 
+class EntityCreate(BaseModel):
+    name: str
+    short_name: Optional[str] = None
+    industry: Optional[str] = None
+    status: Optional[str] = "active"
+
+
+class EntityUpdate(BaseModel):
+    name: Optional[str] = None
+    short_name: Optional[str] = None
+    industry: Optional[str] = None
+    status: Optional[str] = None
+
+
 @router.get("")
 def list_entities(db: Session = Depends(get_db)):
     """获取企业列表"""
@@ -27,3 +42,52 @@ def list_entities(db: Session = Depends(get_db)):
             for e in entities
         ]
     }
+
+
+@router.post("")
+def create_entity(data: EntityCreate, db: Session = Depends(get_db)):
+    """创建企业"""
+    entity = Entity(
+        name=data.name,
+        short_name=data.short_name,
+        industry=data.industry,
+        status=data.status or "active",
+    )
+    db.add(entity)
+    db.commit()
+    db.refresh(entity)
+    return {
+        "id": entity.id,
+        "name": entity.name,
+        "short_name": entity.short_name,
+        "industry": entity.industry,
+        "message": "企业创建成功",
+    }
+
+
+@router.put("/{entity_id}")
+def update_entity(entity_id: int, data: EntityUpdate, db: Session = Depends(get_db)):
+    """更新企业信息"""
+    entity = db.query(Entity).filter(Entity.id == entity_id).first()
+    if not entity:
+        from fastapi.responses import JSONResponse
+        return JSONResponse(status_code=404, content={"detail": "企业不存在"})
+
+    if data.name is not None:
+        entity.name = data.name
+    if data.short_name is not None:
+        entity.short_name = data.short_name
+    if data.industry is not None:
+        entity.industry = data.industry
+    if data.status is not None:
+        entity.status = data.status
+
+    db.commit()
+    db.refresh(entity)
+    return {
+        "id": entity.id,
+        "name": entity.name,
+        "short_name": entity.short_name,
+        "industry": entity.industry,
+        "message": "企业更新成功",
+    }