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

get coverage up to 94%

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-14 09:04:06 -07:00
parent e1ce20f5e3
commit b4a2f37ca9
3 changed files with 44 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
app/dao/provider_details_dao.py
View File

@@ -1,4 +1,4 @@
from datetime import datetime, timedelta
from datetime import datetime
from flask import current_app
from sqlalchemy import asc, desc, func
@@ -80,25 +80,26 @@ def dao_reduce_sms_provider_priority(identifier, *, time_threshold):
Will reduce a chosen sms provider's priority, and increase the other provider's priority by 10 points each.
If either provider has been updated in the last `time_threshold`, then it won't take any action.
"""
amount_to_reduce_by = 10
# amount_to_reduce_by = 10
providers_list = _get_sms_providers_for_update(time_threshold)
if len(providers_list) < 2:
current_app.logger.info("Not adjusting providers, number of active providers is less than 2.")
return
providers = {provider.identifier: provider for provider in providers_list}
other_identifier = get_alternative_sms_provider(identifier)
reduced_provider = providers[identifier]
increased_provider = providers[other_identifier]
# always keep values between 0 and 100
reduced_provider_priority = max(0, reduced_provider.priority - amount_to_reduce_by)
increased_provider_priority = min(100, increased_provider.priority + amount_to_reduce_by)
_adjust_provider_priority(reduced_provider, reduced_provider_priority)
_adjust_provider_priority(increased_provider, increased_provider_priority)
# TODO right now we only have one provider. Remove this?
# providers = {provider.identifier: provider for provider in providers_list}
# other_identifier = get_alternative_sms_provider(identifier)
#
# reduced_provider = providers[identifier]
# increased_provider = providers[other_identifier]
#
# # always keep values between 0 and 100
# reduced_provider_priority = max(0, reduced_provider.priority - amount_to_reduce_by)
# increased_provider_priority = min(100, increased_provider.priority + amount_to_reduce_by)
#
# _adjust_provider_priority(reduced_provider, reduced_provider_priority)
# _adjust_provider_priority(increased_provider, increased_provider_priority)
@autocommit
@@ -107,22 +108,23 @@ def dao_adjust_provider_priority_back_to_resting_points():
Provided that neither SMS provider has been modified in the last hour, move both providers by 10 percentage points
each towards their defined resting points (set in SMS_PROVIDER_RESTING_POINTS in config.py).
"""
amount_to_reduce_by = 10
time_threshold = timedelta(hours=1)
providers = _get_sms_providers_for_update(time_threshold)
for provider in providers:
target = current_app.config['SMS_PROVIDER_RESTING_POINTS'][provider.identifier]
current = provider.priority
if current != target:
if current > target:
new_priority = max(target, provider.priority - amount_to_reduce_by)
else:
new_priority = min(target, provider.priority + amount_to_reduce_by)
_adjust_provider_priority(provider, new_priority)
# TODO we only have one provider. Remove?
# amount_to_reduce_by = 10
# time_threshold = timedelta(hours=1)
#
# providers = _get_sms_providers_for_update(time_threshold)
#
# for provider in providers:
# target = current_app.config['SMS_PROVIDER_RESTING_POINTS'][provider.identifier]
# current = provider.priority
#
# if current != target:
# if current > target:
# new_priority = max(target, provider.priority - amount_to_reduce_by)
# else:
# new_priority = min(target, provider.priority + amount_to_reduce_by)
#
# _adjust_provider_priority(provider, new_priority)
def get_provider_details_by_notification_type(notification_type, supports_international=False):