Add method to populate all notification count fields during seeding

cds-snc · Nov 7, 2024 · c73808e · c73808e
1 parent 089738d
commit c73808e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/notifications_utils/clients/redis/annual_limit.py b/notifications_utils/clients/redis/annual_limit.py
@@ -102,6 +102,9 @@ def reset_all_notification_counts(self, service_ids=None):
 
         self._redis_client.delete_hash_fields(hashes=hashes)
 
+    def seed_annual_limit_notifications(self, service_id: str, mapping: dict):
+        self._redis_client.bulk_set_hash_fields(key=annual_limit_notifications_key(service_id), mapping=mapping)
+
     def was_seeded_today(self, service_id):
         last_seeded_time = self.get_seeded_at(service_id)
         return last_seeded_time == datetime.utcnow().strftime("%Y-%m-%d") if last_seeded_time else False