diff --git a/rbac/init.py b/rbac/init.py index 2e5de48..39e5a6c 100644 --- a/rbac/init.py +++ b/rbac/init.py @@ -42,4 +42,4 @@ def load_rbac(): env.get_org_users = get_org_users env.sor_get_org_users = sor_get_org_users env.get_owner_orgid = get_owner_orgid - + env.sor_add_user_roles = sor_add_user_roles diff --git a/wwwroot/add_adminuser.dspy b/wwwroot/add_adminuser.dspy index eb73d91..3cbee0e 100644 --- a/wwwroot/add_adminuser.dspy +++ b/wwwroot/add_adminuser.dspy @@ -12,7 +12,7 @@ debug(f'{dbname=}') async with db.sqlorContext(dbname) as sor: await sor.C('users',ns.copy()) uid = ns['id'] - await sor_add_user_roles(ns['id'], ['*.admin']) + await sor_add_user_roles(sor, ns['id'], ['*.admin']) return UiMessage(title='Success', message='admin user added') return UiError(title='Error', message='Error happened when add admin user')