darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-09-07 18:18:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add report options 1/3/5-day

Browse Source
This commit is contained in:
Andrew Shumway
2024-06-10 10:15:22 -06:00
parent 8e6b3fa39e
commit a401562735
7 changed files with 63 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
app/models/service.py
View File

@@ -390,7 +390,7 @@ class Service(JSONModel, SortByNameMixin):
def get_data_retention_item(self, id):
return next((dr for dr in self.data_retention if dr["id"] == id), None)
def get_days_of_retention(self, notification_type):
def get_days_of_retention(self, notification_type, number_of_days):
return next(
(
dr
@@ -398,7 +398,10 @@ class Service(JSONModel, SortByNameMixin):
if dr["notification_type"] == notification_type
),
{},
).get("days_of_retention", current_app.config["ACTIVITY_STATS_LIMIT_DAYS"])
).get(
"days_of_retention",
current_app.config["ACTIVITY_STATS_LIMIT_DAYS"].get(number_of_days),
)
@cached_property
def organization(self):