diff --git a/epictrack-api/src/api/reports/thirty_sixty_ninety_report.py b/epictrack-api/src/api/reports/thirty_sixty_ninety_report.py index b046fd37b..5dd110d0c 100644 --- a/epictrack-api/src/api/reports/thirty_sixty_ninety_report.py +++ b/epictrack-api/src/api/reports/thirty_sixty_ninety_report.py @@ -505,6 +505,18 @@ def _format_table_data_events(self, events, style): ]) return data + def _get_event_date_source(self, data): + """Return the date source""" + date_sources = { + "decision_referral": "Decision", + "work_issue": "Issue", + "pcp": "Comment Period", + "other": "Milestone", + } + if data["event_type_id"] == EventTypeEnum.REFERRAL.value: + return "Referral" + return date_sources[data["event_type"]] + def _format_table_data(self, period_data, row_index, style): """Generates styled table rows for the given period data""" # Define a bold style for labels @@ -518,6 +530,7 @@ def _format_table_data(self, period_data, row_index, style): events = group["items"] events_data = self._format_table_data_events(events, style) work = events[0] + date_source_label = self._get_event_date_source(work) + " Date" data.append( [ [ @@ -526,7 +539,7 @@ def _format_table_data(self, period_data, row_index, style): style_bold, ), Paragraph( - f"{work['event_date']: %B %d, %Y}", + f"{date_source_label}: {work['event_date']: %B %d, %Y}", style, ), ],