diff --git a/set_role_perm.py b/set_role_perm.py index 2ac5633..9265315 100644 --- a/set_role_perm.py +++ b/set_role_perm.py @@ -1,3 +1,4 @@ +import os import sys import asyncio from sqlor.dbpools import DBPools @@ -34,7 +35,8 @@ async def main(): role = sys.argv[1] path = sys.argv[2] - async with db.sqlorContext('sage') as sor: + dbname = os.environ.get('SAGE_RBAC_DB', 'sage') + async with db.sqlorContext(dbname) as sor: perms = None if '%' in path: perms = await sor.sqlExe("select * from permission where path like ${path}$", {'path': path})