fix: load_path.py改为直接操作数据库，不依赖sage set_role_perm
This commit is contained in:
Hermes Agent
parent
2d3c74b6ad
commit
ca66a19a74
168
load_path.py
168
load_path.py
@ -2,137 +2,89 @@
|
|||||||
"""
|
"""
|
||||||
Portal CMS CRUD RBAC 权限注册脚本
|
Portal CMS CRUD RBAC 权限注册脚本
|
||||||
|
|
||||||
注册CMS管理后台的所有CRUD路径权限:
|
直接操作 ocai_cms 数据库的 permission/rolepermission 表，
|
||||||
- superuser (owner.superuser): CMS管理页面和API
|
注册CMS管理后台的CRUD路径权限。
|
||||||
- any: 公开API（搜索下拉等）
|
|
||||||
|
|
||||||
使用方法:
|
用法: cd ~/portal && py3/bin/python load_path.py
|
||||||
cd ~/repos/sage
|
|
||||||
./py3/bin/python ~/repos/portal/load_path.py
|
|
||||||
"""
|
"""
|
||||||
|
import asyncio
|
||||||
import subprocess
|
|
||||||
import os
|
|
||||||
import sys
|
import sys
|
||||||
|
import os
|
||||||
|
|
||||||
|
sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)), 'pkgs'))
|
||||||
|
|
||||||
def find_sage_root():
|
from appPublic.jsonConfig import getConfig
|
||||||
candidates = [
|
from sqlor.dbpools import DBPools
|
||||||
os.path.expanduser("~/repos/sage"),
|
from appPublic.uniqueID import getID
|
||||||
os.path.expanduser("~/sage"),
|
|
||||||
]
|
|
||||||
for c in candidates:
|
|
||||||
if os.path.isdir(os.path.join(c, "py3")) and os.path.isdir(os.path.join(c, "wwwroot")):
|
|
||||||
return c
|
|
||||||
return None
|
|
||||||
|
|
||||||
|
# CMS管理页面路径 — superuser可访问
|
||||||
SAGE_ROOT = find_sage_root()
|
|
||||||
if not SAGE_ROOT:
|
|
||||||
print("ERROR: Cannot find Sage root directory")
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
PYTHON = os.path.join(SAGE_ROOT, "py3", "bin", "python")
|
|
||||||
SET_PERM_SCRIPT = os.path.join(SAGE_ROOT, "set_role_perm.py")
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# 权限路径定义
|
|
||||||
# ============================================================
|
|
||||||
|
|
||||||
# any — 无需登录（公开API: 搜索下拉、内容类型列表）
|
|
||||||
PATHS_ANY = [
|
|
||||||
"/api/get_search_cms_categories.dspy",
|
|
||||||
"/api/get_search_content_type.dspy",
|
|
||||||
]
|
|
||||||
|
|
||||||
# owner.superuser — CMS管理CRUD页面和API
|
|
||||||
PATHS_SUPERUSER = [
|
PATHS_SUPERUSER = [
|
||||||
|
"/admin.ui",
|
||||||
# CMS Content CRUD
|
# CMS Content CRUD
|
||||||
"/cms_content_list", "/cms_content_list/%",
|
"/cms_content_list", "/cms_content_list/%",
|
||||||
"/api/cms_content_create.dspy",
|
"/api/cms_content_create.dspy", "/api/cms_content_update.dspy", "/api/cms_content_delete.dspy",
|
||||||
"/api/cms_content_update.dspy",
|
|
||||||
"/api/cms_content_delete.dspy",
|
|
||||||
"/api/cms_content_list.dspy",
|
|
||||||
"/cms_content_list/get_cms_content.dspy",
|
|
||||||
"/cms_content_list/add_cms_content.dspy",
|
|
||||||
"/cms_content_list/update_cms_content.dspy",
|
|
||||||
"/cms_content_list/delete_cms_content.dspy",
|
|
||||||
|
|
||||||
# CMS Sections CRUD
|
# CMS Sections CRUD
|
||||||
"/cms_sections_list", "/cms_sections_list/%",
|
"/cms_sections_list", "/cms_sections_list/%",
|
||||||
"/api/cms_sections_create.dspy",
|
"/api/cms_sections_create.dspy", "/api/cms_sections_update.dspy", "/api/cms_sections_delete.dspy",
|
||||||
"/api/cms_sections_update.dspy",
|
|
||||||
"/api/cms_sections_delete.dspy",
|
|
||||||
"/api/cms_sections_list.dspy",
|
|
||||||
"/cms_sections_list/get_cms_sections.dspy",
|
|
||||||
"/cms_sections_list/add_cms_sections.dspy",
|
|
||||||
"/cms_sections_list/update_cms_sections.dspy",
|
|
||||||
"/cms_sections_list/delete_cms_sections.dspy",
|
|
||||||
|
|
||||||
# CMS Categories CRUD
|
# CMS Categories CRUD
|
||||||
"/cms_categories_list", "/cms_categories_list/%",
|
"/cms_categories_list", "/cms_categories_list/%",
|
||||||
"/api/cms_categories_create.dspy",
|
"/api/cms_categories_create.dspy", "/api/cms_categories_update.dspy", "/api/cms_categories_delete.dspy",
|
||||||
"/api/cms_categories_update.dspy",
|
|
||||||
"/api/cms_categories_delete.dspy",
|
|
||||||
"/api/cms_categories_list.dspy",
|
|
||||||
"/cms_categories_list/get_cms_categories.dspy",
|
|
||||||
"/cms_categories_list/add_cms_categories.dspy",
|
|
||||||
"/cms_categories_list/update_cms_categories.dspy",
|
|
||||||
"/cms_categories_list/delete_cms_categories.dspy",
|
|
||||||
|
|
||||||
# CMS Leads CRUD
|
# CMS Leads CRUD
|
||||||
"/cms_leads_list", "/cms_leads_list/%",
|
"/cms_leads_list", "/cms_leads_list/%",
|
||||||
"/api/cms_leads_create.dspy",
|
"/api/cms_leads_create.dspy", "/api/cms_leads_update.dspy", "/api/cms_leads_delete.dspy",
|
||||||
"/api/cms_leads_update.dspy",
|
|
||||||
"/api/cms_leads_delete.dspy",
|
|
||||||
"/api/cms_leads_list.dspy",
|
|
||||||
"/cms_leads_list/get_cms_leads.dspy",
|
|
||||||
"/cms_leads_list/add_cms_leads.dspy",
|
|
||||||
"/cms_leads_list/update_cms_leads.dspy",
|
|
||||||
"/cms_leads_list/delete_cms_leads.dspy",
|
|
||||||
|
|
||||||
# CMS Site Config CRUD
|
# CMS Site Config CRUD
|
||||||
"/cms_site_config_list", "/cms_site_config_list/%",
|
"/cms_site_config_list", "/cms_site_config_list/%",
|
||||||
"/api/cms_site_config_create.dspy",
|
"/api/cms_site_config_create.dspy", "/api/cms_site_config_update.dspy", "/api/cms_site_config_delete.dspy",
|
||||||
"/api/cms_site_config_update.dspy",
|
|
||||||
"/api/cms_site_config_delete.dspy",
|
|
||||||
"/api/cms_site_config_list.dspy",
|
|
||||||
"/cms_site_config_list/get_cms_site_config.dspy",
|
|
||||||
"/cms_site_config_list/add_cms_site_config.dspy",
|
|
||||||
"/cms_site_config_list/update_cms_site_config.dspy",
|
|
||||||
"/cms_site_config_list/delete_cms_site_config.dspy",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
# ============================================================
|
SUPERUSER_ROLE_ID = 'r0ZHXa9vjGUHqkd4m_66w'
|
||||||
# 执行注册
|
|
||||||
# ============================================================
|
|
||||||
|
|
||||||
def run_set_perm(role, path):
|
|
||||||
env = os.environ.copy()
|
|
||||||
env['SAGE_RBAC_DB'] = 'ocai_cms'
|
|
||||||
cmd = [PYTHON, SET_PERM_SCRIPT, role, path]
|
|
||||||
result = subprocess.run(cmd, capture_output=True, text=True, env=env)
|
|
||||||
return result.returncode == 0
|
|
||||||
|
|
||||||
|
|
||||||
def register_role_paths(role, paths):
|
async def register_permissions():
|
||||||
count = 0
|
config = getConfig('.')
|
||||||
for p in paths:
|
db = DBPools(config.databases)
|
||||||
if run_set_perm(role, p):
|
|
||||||
count += 1
|
|
||||||
print(f" {role}: {count}/{len(paths)} paths registered")
|
|
||||||
return count
|
|
||||||
|
|
||||||
|
async with db.sqlorContext('ocai_cms') as sor:
|
||||||
|
cur = await sor.conn.cursor()
|
||||||
|
|
||||||
def main():
|
# Get existing permissions
|
||||||
print(f"Sage root: {SAGE_ROOT}")
|
await cur.execute("SELECT path FROM permission")
|
||||||
print(f"RBAC DB: ocai_cms")
|
existing = {r[0] for r in await cur.fetchall()}
|
||||||
total = 0
|
|
||||||
total += register_role_paths("any", PATHS_ANY)
|
count = 0
|
||||||
total += register_role_paths("owner.superuser", PATHS_SUPERUSER)
|
for path in PATHS_SUPERUSER:
|
||||||
print(f"\nDone. Total {total} permission entries registered.")
|
if path not in existing:
|
||||||
print("NOTE: Restart Sage after permission changes to reload RBAC cache.")
|
perm_id = getID()
|
||||||
|
await cur.execute(
|
||||||
|
"INSERT INTO permission (id, name, path) VALUES (%s, %s, %s)",
|
||||||
|
(perm_id, None, path)
|
||||||
|
)
|
||||||
|
existing.add(path)
|
||||||
|
print(f" + permission: {path}")
|
||||||
|
|
||||||
|
# Get permission ID and link to superuser role
|
||||||
|
await cur.execute("SELECT id FROM permission WHERE path = %s", (path,))
|
||||||
|
row = await cur.fetchone()
|
||||||
|
if row:
|
||||||
|
perm_id = row[0]
|
||||||
|
# Check if rolepermission already exists
|
||||||
|
await cur.execute(
|
||||||
|
"SELECT id FROM rolepermission WHERE roleid = %s AND permid = %s",
|
||||||
|
(SUPERUSER_ROLE_ID, perm_id)
|
||||||
|
)
|
||||||
|
if not await cur.fetchone():
|
||||||
|
rp_id = getID()
|
||||||
|
await cur.execute(
|
||||||
|
"INSERT INTO rolepermission (id, roleid, permid) VALUES (%s, %s, %s)",
|
||||||
|
(rp_id, SUPERUSER_ROLE_ID, perm_id)
|
||||||
|
)
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
await sor.conn.commit()
|
||||||
|
await cur.close()
|
||||||
|
print(f"\nDone: {count} new rolepermission entries for owner.superuser")
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
print("=== Portal CMS RBAC 权限注册 ===")
|
||||||
|
asyncio.run(register_permissions())
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user