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 # For created checks
if created: if created:
if instance.policy: 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: elif instance.agent:
instance.agent.generate_checks_from_policies() instance.agent.generate_checks_from_policies()
# Checks that are updated except for agent # Checks that are updated except for agent
else: else:
if instance.policy: 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") @receiver(pre_delete, sender="checks.Check")
@ -35,7 +35,7 @@ def pre_delete_check_handler(sender, instance, **kwargs):
# Policy check deleted # Policy check deleted
if instance.policy: 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 # Agent check deleted
elif instance.agent: elif instance.agent: