76 lines
2.6 KiB
Python
76 lines
2.6 KiB
Python
"""用户列表 — 对接 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:
|
|
# 尝试按部门过滤；若 sys_user 无 department 列则降级全量查询
|
|
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()
|
|
except Exception:
|
|
# 降级：不使用 department 列过滤
|
|
fallback_sql = """
|
|
SELECT id, username,
|
|
SPLIT_PART(username, '/', 1) AS full_name,
|
|
'' AS department
|
|
FROM sys_user
|
|
"""
|
|
params = {"lim": limit}
|
|
if keyword.strip():
|
|
sql = text(fallback_sql + " WHERE username ILIKE :kw ORDER BY username LIMIT :lim")
|
|
params["kw"] = f"%{keyword.strip()}%"
|
|
else:
|
|
sql = text(fallback_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()
|