adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Update user controller

Browse Source
This commit is contained in:
wxy
2026-03-27 13:58:31 +08:00
parent f30260939a
commit 14838dc064
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/app/controllers/user_controller.py
View File

@@ -112,7 +112,7 @@ def get_current_user_info(
api_logger.info(f"当前用户信息获取成功: {result.username}, 角色: {result_schema.role}, 工作空间: {result_schema.current_workspace_name}") api_logger.info(f"当前用户信息获取成功: {result.username}, 角色: {result_schema.role}, 工作空间: {result_schema.current_workspace_name}")
# 设置权限：如果用户来自 SSO Source，则使用该 Source 的 permissions；否则返回全部权限 # 设置权限：如果用户来自 SSO Source，则使用该 Source 的 permissions；否则返回 "all" 表示拥有所有权限
if current_user.external_source: if current_user.external_source:
from premium.sso.models import SSOSource from premium.sso.models import SSOSource
source = db.query(SSOSource).filter(SSOSource.source_code == current_user.external_source).first() source = db.query(SSOSource).filter(SSOSource.source_code == current_user.external_source).first()
@@ -121,7 +121,8 @@ def get_current_user_info(
else: else:
result_schema.permissions = [] result_schema.permissions = []
else: else:
result_schema.permissions = ["pricing", "user"] from premium.sso.base import SSOPermission
result_schema.permissions = [SSOPermission.ALL.value]
return success(data=result_schema, msg=t("users.info.get_success")) return success(data=result_schema, msg=t("users.info.get_success"))
@@ -191,7 +192,8 @@ def get_tenant_users(
source = db.query(SSOSource).filter(SSOSource.source_code == user.external_source).first() source = db.query(SSOSource).filter(SSOSource.source_code == user.external_source).first()
u_schema.permissions = source.permissions if source and source.permissions else [] u_schema.permissions = source.permissions if source and source.permissions else []
else: else:
u_schema.permissions = ["pricing", "user"] from premium.sso.base import SSOPermission
u_schema.permissions = [SSOPermission.ALL.value]
return success( return success(
data={ data={