zware/awx
1
0
Fork 0
mirror of https://github.com/ZwareBear/awx.git synced 2026-05-17 07:48:39 -05:00
Code Issues Packages Projects Releases Wiki Activity

Switched to a nicer contextmanager implemenation for role hierarchy rebuild batching

Browse Source 
#1206
This commit is contained in:
Akita Noek
2016-03-15 15:30:43 -04:00
parent e45982b011
commit ce669b03ad
3 changed files with 258 additions and 256 deletions
Download Patch File Download Diff File Expand all files Collapse all files
awx/main/fields.py
+6 -7
View File
@@ -23,7 +23,7 @@ from django.db.transaction import TransactionManagementError
# AWX
from awx.main.models.rbac import RolePermission, Role
from awx.main.models.rbac import RolePermission, Role, batch_role_ancestor_rebuilding
__all__ = ['AutoOneToOneField', 'ImplicitRoleField']
@@ -280,12 +280,11 @@ class ImplicitRoleField(models.ForeignKey):
for parent in parents:
new_parents.add(parent)
Role.pause_role_ancestor_rebuilding()
for role in original_parents - new_parents:
this_role.parents.remove(role)
for role in new_parents - original_parents:
this_role.parents.add(role)
Role.unpause_role_ancestor_rebuilding()
with batch_role_ancestor_rebuilding():
for role in original_parents - new_parents:
this_role.parents.remove(role)
for role in new_parents - original_parents:
this_role.parents.add(role)
setattr(self, '__original_parent_roles', new_parents)