track/backend/app/api/v1/endpoints/users.py

"""用户列表 — 对接 MOM sys_user"""
from fastapi import APIRouter, Query, HTTPException, status
from pydantic import BaseModel
from app.core.mom_database import MomSessionLocal
from sqlalchemy import text

router = APIRouter(prefix="/users", tags=["用户"])


class UserOption(BaseModel):
    id: str
    username: str
    full_name: str
    department: str = ""

    model_config = {"from_attributes": True}


@router.get("/", response_model=list[UserOption])
def list_users(
    keyword: str = Query("", description="按用户名/姓名模糊搜索"),
    limit: int = Query(100, ge=1, le=500),
    dept: str = Query("IRIS", description="部门过滤"),
):
    """获取 MOM 系统用户列表，默认只返回 IRIS 部门"""
    db = MomSessionLocal()
    try:
        base_sql = """
            SELECT id, username,
                   SPLIT_PART(username, '/', 1) AS full_name,
                   COALESCE(department, '') AS department
            FROM sys_user
            WHERE department = :dept
        """
        params = {"dept": dept, "lim": limit}
        if keyword.strip():
            sql = text(base_sql + " AND username ILIKE :kw ORDER BY username LIMIT :lim")
            params["kw"] = f"%{keyword.strip()}%"
        else:
            sql = text(base_sql + " ORDER BY username LIMIT :lim")
        rows = db.execute(sql, params).fetchall()

        return [
            UserOption(
                id=str(row.id),
                username=row.username.split("/")[-1] if "/" in row.username else row.username,
                full_name=row.full_name,
                department=row.department or "",
            )
            for row in rows
        ]
    except Exception as e:
        raise HTTPException(
            status_code=status.HTTP_502_BAD_GATEWAY,
            detail=f"MOM 用户查询失败: {str(e)}",
        )
    finally:
        db.close()