mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-19 14:09:20 -04:00
Updated code and added tests
This commit is contained in:
@@ -262,22 +262,57 @@ def get_months_for_year(start, end, year):
|
||||
return [datetime(year, month, 1) for month in range(start, end)]
|
||||
|
||||
|
||||
def get_total_units_for_month(monthly_usage):
|
||||
total = 0
|
||||
for rate_group in monthly_usage:
|
||||
total += rate_group['units'] * rate_group['multiplier']
|
||||
return total
|
||||
|
||||
|
||||
def get_free_paid_breakdown_for_billable_units(year, billable_units):
|
||||
cumulative = 0
|
||||
for month in get_months_for_financial_year(year):
|
||||
previous_cumulative = cumulative
|
||||
monthly_usage = billable_units.get(month, 0)
|
||||
monthly_usage = get_total_units_for_month(billable_units.get(month, []))
|
||||
cumulative += monthly_usage
|
||||
breakdown = get_free_paid_breakdown_for_month(
|
||||
cumulative, previous_cumulative, monthly_usage
|
||||
cumulative, previous_cumulative, billable_units.get(month, [])
|
||||
)
|
||||
yield {
|
||||
'name': month,
|
||||
'paid': breakdown['paid'],
|
||||
'free': breakdown['free']
|
||||
'free': breakdown['free'],
|
||||
'rate_groups': breakdown['rate_groups']
|
||||
}
|
||||
|
||||
|
||||
def get_rate_groups(monthly_usage, free_allowance=0):
|
||||
rate_groups = []
|
||||
|
||||
for rate_group in monthly_usage:
|
||||
free_units = 0
|
||||
units = rate_group["units"]
|
||||
if free_allowance > 0:
|
||||
units = rate_group["units"] * rate_group["multiplier"] - free_allowance
|
||||
if units < 0:
|
||||
free_units = rate_group["units"] * rate_group["multiplier"]
|
||||
free_allowance = abs(units)
|
||||
units = 0
|
||||
else:
|
||||
free_units = free_allowance
|
||||
free_allowance = 0
|
||||
|
||||
rate_group = {
|
||||
"international": rate_group["international"],
|
||||
"free_units": free_units,
|
||||
"units": units,
|
||||
"multiplier": rate_group["multiplier"]
|
||||
}
|
||||
rate_groups.append(rate_group)
|
||||
|
||||
return rate_groups
|
||||
|
||||
|
||||
def get_free_paid_breakdown_for_month(
|
||||
cumulative,
|
||||
previous_cumulative,
|
||||
@@ -285,20 +320,25 @@ def get_free_paid_breakdown_for_month(
|
||||
):
|
||||
allowance = 250000
|
||||
|
||||
total_monthly_units = get_total_units_for_month(monthly_usage)
|
||||
|
||||
if cumulative < allowance:
|
||||
return {
|
||||
'paid': 0,
|
||||
'free': monthly_usage,
|
||||
'free': total_monthly_units,
|
||||
}
|
||||
elif previous_cumulative < allowance:
|
||||
remaining_allowance = allowance - previous_cumulative
|
||||
return {
|
||||
'paid': monthly_usage - (allowance - previous_cumulative),
|
||||
'free': allowance - previous_cumulative
|
||||
'paid': total_monthly_units - remaining_allowance,
|
||||
'free': remaining_allowance,
|
||||
'rate_groups': get_rate_groups(monthly_usage, remaining_allowance)
|
||||
}
|
||||
else:
|
||||
return {
|
||||
'paid': monthly_usage,
|
||||
'free': 0
|
||||
'paid': total_monthly_units,
|
||||
'free': 0,
|
||||
'rate_groups': get_rate_groups(monthly_usage)
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -82,12 +82,22 @@
|
||||
smallest=True
|
||||
) }}
|
||||
<ul>
|
||||
{% if month.free %}
|
||||
<li class="tabular-numbers">{{ "{:,}".format(month.free) }} free text messages</li>
|
||||
{% endif %}
|
||||
{% if month.paid %}
|
||||
<li class="tabular-numbers">{{ "{:,}".format(month.paid) }} text messages at
|
||||
{{- ' {:.2f}p'.format(sms_rate * 100) }}</li>
|
||||
{% if month.rate_groups %}
|
||||
rate_group {{ month.rate_groups|length }}
|
||||
{% for rate in month.rate_groups %}
|
||||
{% if rate.free_units %}
|
||||
<li class="tabular-numbers">{{ "{:,}".format(rate.free_units) }} {% if rate.international %}international{% else %}UK{% endif %} free text messages</li>
|
||||
{% endif %}
|
||||
{% if rate.units %}
|
||||
<li class="tabular-numbers">{{ "{:,}".format(rate.units) }} {% if rate.international %}international{% else %}UK{% endif %} text messages at
|
||||
{{- ' {:.2f}p'.format(rate.multiplier * sms_rate * 100) }}</li>
|
||||
{% endif %}
|
||||
{% endfor %}
|
||||
{% else %}
|
||||
{% if month.paid %}
|
||||
<li class="tabular-numbers">{{ "{:,}".format(month.paid) }} text messages at
|
||||
{{- ' {:.2f}p'.format(sms_rate * 100) }}</li>
|
||||
{% endif %}
|
||||
{% endif %}
|
||||
{% if not (month.free or month.paid) %}
|
||||
<li aria-hidden="true">–</li>
|
||||
|
||||
Reference in New Issue
Block a user