fix policy instance pk

This commit is contained in:
wh1te909 2020-06-15 17:43:46 +00:00
parent 990474216f
commit 0bd5a39744
1 changed files with 3 additions and 3 deletions

View File

@ -16,14 +16,14 @@ def post_save_check_handler(sender, instance, created, **kwargs):
# For created checks
if created:
if instance.policy:
generate_agent_checks_from_policies_task.delay(policypk=instance.pk)
generate_agent_checks_from_policies_task.delay(policypk=instance.policy.pk)
elif instance.agent:
instance.agent.generate_checks_from_policies()
# Checks that are updated except for agent
else:
if instance.policy:
generate_agent_checks_from_policies_task.delay(policypk=instance.pk)
generate_agent_checks_from_policies_task.delay(policypk=instance.policy.pk)
@receiver(pre_delete, sender="checks.Check")
@ -35,7 +35,7 @@ def pre_delete_check_handler(sender, instance, **kwargs):
# Policy check deleted
if instance.policy:
generate_agent_checks_from_policies_task.delay(policypk=instance.pk)
generate_agent_checks_from_policies_task.delay(policypk=instance.policy.pk)
# Agent check deleted
elif instance.agent: