Updated role hierarchy cache rebuilder to handle adds to .children as well as .parents

This commit is contained in:
Akita Noek 2016-02-05 16:46:00 -05:00
parent 5ed766ed35
commit 4540eb0079

View File

@ -115,8 +115,12 @@ def store_initial_active_state(sender, **kwargs):
else:
instance._saved_active_state = True
def rebuild_role_hierarchy_cache(sender, **kwargs):
kwargs['instance'].rebuild_role_hierarchy_cache()
def rebuild_role_hierarchy_cache(sender, reverse, model, pk_set, **kwargs):
if reverse:
for id in pk_set:
model.objects.get(id=id).rebuild_role_hierarchy_cache()
else:
kwargs['instance'].rebuild_role_hierarchy_cache()
pre_save.connect(store_initial_active_state, sender=Host)
post_save.connect(emit_update_inventory_on_created_or_deleted, sender=Host)