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

Initial front end count steps

Browse Source
This commit is contained in:
Andrew Shumway
2023-10-10 10:59:46 -06:00
committed by Carlo Costino
parent 63a4a92657
commit 2fbda1cd0c
2 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/__init__.py
View File

@@ -287,8 +287,9 @@ def init_app(application):
remaining_global_messages = 0
if current_app:
service_id = session.get("service_id")
global_limit = current_app.config["GLOBAL_SERVICE_MESSAGE_LIMIT"]
global_messages_count = service_api_client.get_global_notification_count()
global_messages_count = service_api_client.get_global_notification_count(service_id)
remaining_global_messages = global_limit - global_messages_count
return {"daily_global_messages_remaining": remaining_global_messages}

7
app/notify_client/service_api_client.py
View File

@@ -497,11 +497,8 @@ class ServiceAPIClient(NotifyAdminAPIClient):
return int(count)
def get_global_notification_count(self):
# if cache is not set, or not enabled, return 0
count = redis_client.get(daily_total_cache_key()) or 0
return int(count)
def get_global_notification_count(self, service_id):
return self.get("/service/{}/notification-count".format(service_id))
service_api_client = ServiceAPIClient()