优化权限控制
This commit is contained in:
Binary file not shown.
Binary file not shown.
@@ -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),
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -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,
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user