External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-05-17 22:37:41 -02:30
Code Issues Packages Projects Releases Wiki Activity

Updated notifications.py

Browse Source 
Fixed the customized notification returning incorrect values for host_status_counts

Update notifications.py

Removed if condition

Added exception handling

A nitpick
This commit is contained in:
Saurabh Thakre
2021-02-18 12:24:14 +00:00
parent 753749d2b4
commit 6aca9d80bb
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
awx/main/models/notifications.py
View File

@@ -383,12 +383,17 @@ class JobNotificationMixin(object):
and a url to the job run.""" and a url to the job run."""
job_context = {'host_status_counts': {}} job_context = {'host_status_counts': {}}
summary = None summary = None
if hasattr(self, 'job_host_summaries'): try:
summary = self.job_host_summaries.first() has_event_property = any([f for f in self.event_class._meta.fields if f.name == 'event'])
if summary: except NotImplementedError:
from awx.api.serializers import JobHostSummarySerializer has_event_property = False
summary_data = JobHostSummarySerializer(summary).to_representation(summary) if has_event_property:
job_context['host_status_counts'] = summary_data qs = self.get_event_queryset()
if qs:
event = qs.only('event_data').filter(event='playbook_on_stats').first()
if event:
summary = event.get_host_status_counts()
job_context['host_status_counts'] = summary
context = { context = {
'job': job_context, 'job': job_context,
'job_friendly_name': self.get_notification_friendly_name(), 'job_friendly_name': self.get_notification_friendly_name(),