From 15a0d4b8bbd61bb1625cf80a08a60487eeeebbf3 Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Wed, 3 Jun 2026 17:11:55 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E6=9D=83=E9=99=90=E5=88=9D=E5=A7=8B?=
 =?UTF-8?q?=E5=8C=96=E8=84=9A=E6=9C=AC=E6=8C=87=E5=AE=9A=20SAGE=5FRBAC=5FD?=
 =?UTF-8?q?B=3Docai=5Fcms=20=E9=81=BF=E5=85=8D=E6=95=B0=E6=8D=AE=E5=BA=93?=
 =?UTF-8?q?=E4=B8=8D=E5=8C=B9=E9=85=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 init_any_permissions.py       | 5 ++++-
 init_superuser_permissions.py | 4 +++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/init_any_permissions.py b/init_any_permissions.py
index eb5b2c2..c1626e8 100644
--- a/init_any_permissions.py
+++ b/init_any_permissions.py
@@ -92,12 +92,15 @@ def scan_bricks(bricks_dir):
 def set_any_perms(paths):
     """为路径列表设置any权限"""
     count = 0
+    env = os.environ.copy()
+    env['SAGE_RBAC_DB'] = 'ocai_cms'
     for p in paths:
         result = subprocess.run(
             [py, sp, "any", p],
             cwd=sage_root,
             capture_output=True,
-            text=True
+            text=True,
+            env=env
         )
         status = "✓" if result.returncode == 0 else "✗"
         print(f"  {status} any  {p}")
diff --git a/init_superuser_permissions.py b/init_superuser_permissions.py
index ff0af97..43cae82 100644
--- a/init_superuser_permissions.py
+++ b/init_superuser_permissions.py
@@ -29,9 +29,11 @@ if not sp:
 
 def run(role, paths):
     assert sp is not None, "set_role_perm.py not found"
+    env = os.environ.copy()
+    env['SAGE_RBAC_DB'] = 'ocai_cms'
     for p in paths:
         print(f"  {role:30s} {p}")
-        subprocess.run([py, sp, role, p], cwd=sage_root, capture_output=True)
+        subprocess.run([py, sp, role, p], cwd=sage_root, capture_output=True, env=env)
 
 # ─── superuser — 所有权限 ───
 superuser_paths = [