This commit is contained in:
wuaho 2021-09-24 09:37:35 +08:00
parent 01600069d6
commit b97ce1a77a

View File

@ -34,6 +34,7 @@ async def add_role_domain(
role=item.role_id, role=item.role_id,
domain=item.game) domain=item.game)
casbin_enforcer.load_model() casbin_enforcer.load_model()
casbin_enforcer.load_policy()
return schemas.Msg(code=0, msg='添加成功', data=True) return schemas.Msg(code=0, msg='添加成功', data=True)
@ -75,6 +76,7 @@ async def del_role_domain(
role=data_in.role_id, role=data_in.role_id,
domain=data_in.game) domain=data_in.game)
casbin_enforcer.load_model() casbin_enforcer.load_model()
casbin_enforcer.load_policy()
await crud.role.delete_id(db, data_in.role_id) await crud.role.delete_id(db, data_in.role_id)
return schemas.Msg(code=0, msg='ok', data=res) return schemas.Msg(code=0, msg='ok', data=res)
@ -91,7 +93,7 @@ async def add_policy(
res = 0 res = 0
for path in data_id.path_list: for path in data_id.path_list:
res = casbin_enforcer.add_policy(data_id.role_id, data_id.game, path, data_id.act) res = casbin_enforcer.add_policy(data_id.role_id, data_id.game, path, data_id.act)
# casbin_enforcer.load_model() casbin_enforcer.load_model()
casbin_enforcer.load_policy() casbin_enforcer.load_policy()
return schemas.Msg(code=0, msg='ok', data=res) return schemas.Msg(code=0, msg='ok', data=res)
@ -106,7 +108,7 @@ async def remove_policy(
""" """
res = casbin_enforcer.remove_policy(data_id.role_id, data_id.game, data_id.path, data_id.act) res = casbin_enforcer.remove_policy(data_id.role_id, data_id.game, data_id.path, data_id.act)
# casbin_enforcer.load_model() casbin_enforcer.load_model()
casbin_enforcer.load_policy() casbin_enforcer.load_policy()
return schemas.Msg(code=0, msg='ok', data=res) return schemas.Msg(code=0, msg='ok', data=res)
@ -156,6 +158,7 @@ async def del_api(
for item in paths: for item in paths:
casbin_enforcer.remove_filtered_policy(2, item['path']) casbin_enforcer.remove_filtered_policy(2, item['path'])
casbin_enforcer.load_model() casbin_enforcer.load_model()
casbin_enforcer.load_policy()
# 删除保存的记录 # 删除保存的记录
res = await crud.api_list.del_api(db, data_in) res = await crud.api_list.del_api(db, data_in)