优化权限控制

This commit is contained in:
yml2213
2026-06-22 17:07:29 +08:00
parent 480dfa6085
commit 8771d91a30
18 changed files with 222 additions and 16 deletions
Binary file not shown.
+3 -3
View File
@@ -7,7 +7,7 @@ from sqlalchemy.orm import Session
from ..database import get_db
from ..models import User, AuditLog
from ..security import verify_password, create_access_token
from ..permissions import get_role_permissions, ROLE_LABELS
from ..permissions import get_user_permissions, ROLE_LABELS
from ..schemas import LoginRequest, TokenResponse
from ..deps import get_current_user
@@ -23,7 +23,7 @@ def login(req: LoginRequest, db: Session = Depends(get_db)):
raise HTTPException(status_code=403, detail="账号已禁用,请联系管理员")
token = create_access_token({"sub": str(user.id), "role": user.role})
perms = get_role_permissions(user.role)
perms = get_user_permissions(user)
# 审计
db.add(AuditLog(user_id=user.id, username=user.username, action="login", target="auth"))
@@ -46,7 +46,7 @@ def me(current_user: User = Depends(get_current_user)):
"role_label": ROLE_LABELS.get(current_user.role, current_user.role),
"is_active": current_user.is_active,
"remark": current_user.remark,
"permissions": get_role_permissions(current_user.role),
"permissions": get_user_permissions(current_user),
}
+19 -4
View File
@@ -6,7 +6,7 @@ from sqlalchemy.orm import Session
from ..database import get_db
from ..models import User, AuditLog
from ..security import hash_password
from ..permissions import ROLE_LABELS, get_role_permissions
from ..permissions import ROLE_LABELS, PERMISSIONS, get_user_permissions
from ..schemas import UserCreate, UserUpdate, UserInfo
from ..deps import require_permission, get_current_user
@@ -28,7 +28,8 @@ def list_users(
is_active=u.is_active,
remark=u.remark or "",
created_at=u.created_at,
permissions=get_role_permissions(u.role),
permissions=get_user_permissions(u),
custom_permissions=u.custom_permissions,
))
return result
@@ -60,7 +61,8 @@ def create_user(
return UserInfo(
id=user.id, username=user.username, role=user.role,
is_active=user.is_active, remark=user.remark or "",
permissions=get_role_permissions(user.role),
permissions=get_user_permissions(user),
custom_permissions=user.custom_permissions,
)
@@ -83,6 +85,8 @@ def update_user(
user.is_active = req.is_active
if req.remark is not None:
user.remark = req.remark
if req.custom_permissions is not None:
user.custom_permissions = req.custom_permissions
db.commit()
db.refresh(user)
@@ -94,7 +98,8 @@ def update_user(
return UserInfo(
id=user.id, username=user.username, role=user.role,
is_active=user.is_active, remark=user.remark or "",
permissions=get_role_permissions(user.role),
permissions=get_user_permissions(user),
custom_permissions=user.custom_permissions,
)
@@ -115,3 +120,13 @@ def delete_user(
db.delete(user)
db.commit()
return {"message": "已删除", "success": True}
@router.get("/permissions/list")
def list_permissions(_: User = Depends(require_permission("user:assign_permissions"))):
"""返回所有可用权限点及角色默认权限映射。"""
from ..permissions import ROLE_PERMISSIONS
return {
"permissions": PERMISSIONS,
"role_permissions": ROLE_PERMISSIONS,
}