diff --git a/set_role_perm.py b/set_role_perm.py index 796b287..6f1aa0c 100644 --- a/set_role_perm.py +++ b/set_role_perm.py @@ -43,24 +43,23 @@ async def main(): for p in perms: await add_roleperm(sor, role, p.id) return - """ - if role == ‘customer.*' # 客户企业的任意角色 - if role == '*.admin' # 任意类型的其他的管理员 - """ - orgtypeid, name = role.split('.') - ns = { - 'orgtypeid': orgtypeid, - 'name': name - } - roles = await sor.R('role', ns.copy()) - if not roles: - ns['id'] = getID() - await sor.C('role', ns.copy()) + if role in ['anonymous', 'any', 'logined']: + await add_roleperm(sor, role, p.id) else: - ns['id'] = roles[0].id - for p in perms: - await add_roleperm(sor, ns['id'], p.id) - await delete_anonymous_perm(sor, p.id) + orgtypeid, name = role.split('.') + ns = { + 'orgtypeid': orgtypeid, + 'name': name + } + roles = await sor.R('role', ns.copy()) + if not roles: + ns['id'] = getID() + await sor.C('role', ns.copy()) + else: + ns['id'] = roles[0].id + for p in perms: + await add_roleperm(sor, ns['id'], p.id) + await delete_anonymous_perm(sor, p.id) if __name__ == '__main__':