darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-09 06:32:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

reduce code

Browse Source
This commit is contained in:
Kenneth Kehl
2024-12-18 08:12:44 -08:00
parent 2c2ee41542
commit 9d9ce137c4
2 changed files with 10011 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
app/clients/cloudwatch/aws_cloudwatch.py
View File

@@ -155,27 +155,18 @@ class AwsCloudwatchClient(Client):
def check_delivery_receipts(self, start, end): def check_delivery_receipts(self, start, end):
region = cloud_config.sns_region region = cloud_config.sns_region
account_number = self._extract_account_number(cloud_config.ses_domain_arn) account_number = self._extract_account_number(cloud_config.ses_domain_arn)
delivered_event_set = set()
log_group_name = f"sns/{region}/{account_number[4]}/DirectPublishToPhoneNumber" log_group_name = f"sns/{region}/{account_number[4]}/DirectPublishToPhoneNumber"
all_delivered_events = self._get_log(log_group_name, start, end) delivered_event_set = self._get_receipts(log_group_name, start, end)
current_app.logger.info(
f"Delivered count {len(all_delivered_events)} over range {start} to {end}"
)
for event in all_delivered_events:
actual_event = self.event_to_db_format(event["message"])
delivered_event_set.add(json.dumps(actual_event))
failed_event_set = set()
log_group_name = ( log_group_name = (
f"sns/{region}/{account_number[4]}/DirectPublishToPhoneNumber/Failure" f"sns/{region}/{account_number[4]}/DirectPublishToPhoneNumber/Failure"
) )
all_failed_events = self._get_log(log_group_name, start, end) failed_event_set = self._get_receipts(log_group_name, start, end)
current_app.logger.info(
f"Failed count {len(all_delivered_events)} over range {start} to {end}"
)
for event in all_failed_events:
actual_event = self.event_to_db_format(event["message"])
failed_event_set.add(json.dumps(actual_event))
return delivered_event_set, failed_event_set return delivered_event_set, failed_event_set
def _get_receipts(self, log_group_name, start, end):
event_set = set()
all_events = self._get_log(log_group_name, start, end)
for event in all_events:
actual_event = self.event_to_db_format(event["message"])
event_set.add(json.dumps(actual_event))
return event_set

10001
loadtest_10k.csv Normal file
View File

File diff suppressed because it is too large Load Diff