mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-23 20:01:01 -05:00
8122fa1d5d
Makes these assumptions: - we don’t care about history before 29th November 2019 at 11am (this is when priority started to mean a proportion rather than a ranking) - the priority of the second provider will always be the inverse of the first provider Which means the code is a lot simpler/actually does what you’d expect.
124 lines
4.2 KiB
Python
124 lines
4.2 KiB
Python
from collections import defaultdict
|
|
from datetime import datetime
|
|
from operator import itemgetter
|
|
|
|
from flask import abort, render_template, url_for
|
|
from werkzeug.utils import redirect
|
|
|
|
from app import format_date_numeric, provider_client
|
|
from app.main import main
|
|
from app.main.forms import ProviderForm, ProviderRatioForm
|
|
from app.utils import user_is_platform_admin
|
|
|
|
PROVIDER_PRIORITY_MEANING_SWITCHOVER = datetime(2019, 11, 29, 11, 0).isoformat()
|
|
|
|
|
|
@main.route("/providers")
|
|
@user_is_platform_admin
|
|
def view_providers():
|
|
providers = provider_client.get_all_providers()['provider_details']
|
|
domestic_email_providers, domestic_sms_providers, intl_sms_providers = [], [], []
|
|
for provider in providers:
|
|
if provider['notification_type'] == 'sms':
|
|
domestic_sms_providers.append(provider)
|
|
if provider.get('supports_international', None):
|
|
intl_sms_providers.append(provider)
|
|
elif provider['notification_type'] == 'email':
|
|
domestic_email_providers.append(provider)
|
|
|
|
add_monthly_traffic(domestic_sms_providers)
|
|
|
|
return render_template(
|
|
'views/providers/providers.html',
|
|
email_providers=domestic_email_providers,
|
|
domestic_sms_providers=domestic_sms_providers,
|
|
intl_sms_providers=intl_sms_providers
|
|
)
|
|
|
|
|
|
def add_monthly_traffic(domestic_sms_providers):
|
|
total_sms_sent = sum(provider['current_month_billable_sms'] for provider in domestic_sms_providers)
|
|
|
|
for provider in domestic_sms_providers:
|
|
percentage = (provider['current_month_billable_sms'] / total_sms_sent * 100) if total_sms_sent else 0
|
|
provider['monthly_traffic'] = round(percentage)
|
|
|
|
|
|
@main.route("/provider/<uuid:provider_id>/edit", methods=['GET', 'POST'])
|
|
@user_is_platform_admin
|
|
def edit_provider(provider_id):
|
|
provider = provider_client.get_provider_by_id(provider_id)['provider_details']
|
|
form = ProviderForm(active=provider['active'], priority=provider['priority'])
|
|
|
|
if form.validate_on_submit():
|
|
provider_client.update_provider(provider_id, form.priority.data)
|
|
return redirect(url_for('.view_providers'))
|
|
|
|
return render_template('views/providers/edit-provider.html', form=form, provider=provider)
|
|
|
|
|
|
@main.route("/provider/edit-sms-provider-ratio", methods=['GET', 'POST'])
|
|
@user_is_platform_admin
|
|
def edit_sms_provider_ratio():
|
|
|
|
providers = sorted([
|
|
provider
|
|
for provider in provider_client.get_all_providers()['provider_details']
|
|
if provider['notification_type'] == 'sms'
|
|
], key=itemgetter('identifier'), reverse=True)
|
|
|
|
form = ProviderRatioForm(ratio=providers[0]['priority'])
|
|
|
|
if len(providers) < 2:
|
|
abort(400)
|
|
|
|
primary_provider, secondary_provider = providers[0:2]
|
|
|
|
if form.validate_on_submit():
|
|
provider_client.update_provider(primary_provider['id'], form.percentage_left)
|
|
provider_client.update_provider(secondary_provider['id'], form.percentage_right)
|
|
return redirect(url_for('.edit_sms_provider_ratio'))
|
|
|
|
return render_template(
|
|
'views/providers/edit-sms-provider-ratio.html',
|
|
versions=_chunk_versions_by_day(_get_versions_since_switchover(primary_provider['id'])),
|
|
form=form,
|
|
primary_provider=providers[0]['display_name'],
|
|
secondary_provider=providers[1]['display_name'],
|
|
)
|
|
|
|
|
|
def _get_versions_since_switchover(provider_id):
|
|
|
|
for version in sorted(
|
|
provider_client.get_provider_versions(provider_id)['data'],
|
|
key=lambda version: version['updated_at'] or ''
|
|
):
|
|
|
|
if not version['updated_at']:
|
|
continue
|
|
|
|
if version['updated_at'] < PROVIDER_PRIORITY_MEANING_SWITCHOVER:
|
|
continue
|
|
|
|
yield version
|
|
|
|
|
|
def _chunk_versions_by_day(versions):
|
|
|
|
days = defaultdict(list)
|
|
|
|
for version in sorted(versions, key=lambda version: version['updated_at'] or '', reverse=True):
|
|
days[
|
|
format_date_numeric(version['updated_at'])
|
|
].append(version)
|
|
|
|
return sorted(days.items(), reverse=True)
|
|
|
|
|
|
@main.route("/provider/<uuid:provider_id>")
|
|
@user_is_platform_admin
|
|
def view_provider(provider_id):
|
|
versions = provider_client.get_provider_versions(provider_id)
|
|
return render_template('views/providers/provider.html', provider_versions=versions['data'])
|