feat: add load_path.py for RBAC permission registration

This commit is contained in:
yumoqing 2026-05-27 16:28:37 +08:00
parent 3e93f61594
commit e0296882fe

56
scripts/load_path.py Normal file
View File

@ -0,0 +1,56 @@
"""Generate RBAC permissions for uapi module paths.
Run from Sage root with Sage venv:
cd ~/repos/sage && ./py3/bin/python ../uapi/scripts/load_path.py
"""
import os
import sys
import asyncio
sage_root = os.environ.get('SAGE_ROOT')
if sage_root and sage_root not in sys.path:
sys.path.insert(0, sage_root)
from sqlor.dbpools import DBPools
from appPublic.jsonConfig import getConfig
from appPublic.dictObject import DictObject
from appPublic.uniqueID import getID
paths = [
("/uapi", "logined"),
("/uapi/jump_in.dspy", "logined"),
("/uapi/minimax_callback.dspy", "any"),
("/uapi/uptask_callback.dspy", "any"),
("/uapi/viducallback", "any"),
]
async def main():
config = getConfig('.')
DBPools(config.databases)
dbname = 'sage'
async with DBPools().sqlorContext(dbname) as sor:
cnt = 0
for path, role in paths:
r = await sor.sqlExe(
'select * from permission where permcode = ${permcode}$',
{'permcode': path}
)
if len(r) == 0:
await sor.sqlExe(
'''insert into permission (id, permcode, permname, permtype)
values (${id}$, ${permcode}$, ${permname}$, ${permtype}$)''',
{
'id': getID(),
'permcode': path,
'permname': path,
'permtype': role,
}
)
cnt += 1
print(f'{cnt} path(s) inserted for uapi')
if __name__ == '__main__':
asyncio.run(main())