feat: 跨域访问从硬编码改为权限码crossDomain动态控制, 新增SQL迁移脚本

This commit is contained in:
yueli
2026-07-15 09:22:58 +08:00
parent d75c301556
commit 9cae1cc44c
2 changed files with 79 additions and 9 deletions

View File

@ -167,14 +167,33 @@ def permission_required(permission_code):
return decorator
return wrapper
def _has_cross_domain_permission(user_role, user_company=''):
"""
检查指定角色是否拥有「全局跨域访问」权限。
复用 AuthService.get_user_permissions 统一查询。
"""
if not user_role:
return False
try:
from app.services.auth_service import AuthService
perm_dict = AuthService.get_user_permissions(user_role, company_name=user_company)
all_perms = perm_dict.get('menus', []) + perm_dict.get('elements', [])
return 'crossDomain' in all_perms
except Exception:
return False
def get_current_company_filter():
"""
多租户数据权限隔离工具函数。
根据当前 JWT 中的 role 和 company_name 决定 SQL 查询应过滤到哪个公司。
权限逻辑(基于权限码 crossDomain支持动态分配:
- SUPER_ADMIN → None全量跨域
- 拥有 crossDomain 权限 → None全量跨域
- 普通用户 → 强制隔离到 JWT 中记录的公司
返回值:
None → 不限制公司(超级管理员全局视角,或显式传了 'ALL'
None → 不限制公司
str → 仅查询该公司数据
使用示例:
@ -194,17 +213,21 @@ def get_current_company_filter():
# 从请求参数获取前端指定的公司(兼容 company_name 和 company 两个参数名)
req_company = request.args.get('company_name', '') or request.args.get('company', '')
# 超级管理员 → 全局跨域
if user_role == 'SUPER_ADMIN':
# 超级管理员:允许通过参数指定公司过滤
if req_company and req_company.strip().upper() != 'ALL':
return req_company.strip()
# 未指定或指定了 ALL → 返回 None全量不限制
return None
else:
# 其他角色SUPERVISOR / FINANCE / WAREHOUSE_MGR 等):
# 强制隔离到 JWT 中记录的公司,无视前端参数
user_company = claims.get('company_name', '')
return user_company if user_company else None
# 动态权限码检查:拥有 crossDomain 权限 → 全局跨域
user_company = claims.get('company_name', '')
if user_role and _has_cross_domain_permission(user_role, user_company):
if req_company and req_company.strip().upper() != 'ALL':
return req_company.strip()
return None
# 普通用户 → 强制隔离到本公司
return user_company if user_company else None
def audit_log(module: str = None, action: str = None, get_target_id_fn=None, get_target_name_fn=None, get_details_fn=None):

View File

@ -0,0 +1,47 @@
-- ============================================================
-- migrations_add_cross_domain_permission.sql
-- 新增「全局跨域访问」权限节点,支持基于权限码的动态跨公司数据访问控制
--
-- 逻辑:
-- - 拥有 crossDomain 权限的角色 → 可查看所有公司数据
-- - 无此权限的角色 → 只能看本公司数据
--
-- 执行方式:
-- psql -h <host> -U postgres -d inventory_system -f migrations_add_cross_domain_permission.sql
-- ============================================================
BEGIN;
-- 1. 向 sys_element 插入跨域权限码(挂 system 菜单下)
INSERT INTO sys_element (menu_code, name, code, element_type)
SELECT 'system_mgmt', '全局跨域访问', 'crossDomain', 'button'
WHERE NOT EXISTS (
SELECT 1 FROM sys_element WHERE code = 'crossDomain'
);
-- 2. 向 sys_menu 插入菜单节点(如果 system 菜单下还缺此菜单项)
-- 注意: 这是一个能力型权限,不一定需要前端菜单入口,
-- 如果需要在权限分配页面中可见,保留此 INSERT
-- 如果纯后端控制,此行可注释。
INSERT INTO sys_menu (code, name, parent_code, is_visible)
SELECT 'system_cross_domain', '全局跨域访问', 'system_mgmt', true
WHERE NOT EXISTS (
SELECT 1 FROM sys_menu WHERE code = 'system_cross_domain'
);
-- 3. 默认分配给 SUPER_ADMIN 角色
INSERT INTO sys_role_permission (role_code, target_code, type)
SELECT 'SUPER_ADMIN', 'crossDomain', 'element'
WHERE NOT EXISTS (
SELECT 1 FROM sys_role_permission
WHERE role_code = 'SUPER_ADMIN' AND target_code = 'crossDomain'
);
-- 4. 验证
SELECT code, name, menu_code, element_type
FROM sys_element WHERE code = 'crossDomain';
SELECT role_code, target_code, type
FROM sys_role_permission WHERE target_code = 'crossDomain';
COMMIT;