mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 23:41:17 -05:00
142 lines
4.7 KiB
Python
142 lines
4.7 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
from sqlalchemy import Float, Integer
|
|
from sqlalchemy import func, case, cast
|
|
from sqlalchemy import literal_column
|
|
|
|
from app import db
|
|
from app.dao.date_util import get_financial_year
|
|
from app.models import (
|
|
NotificationHistory,
|
|
Rate,
|
|
NOTIFICATION_STATUS_TYPES_BILLABLE,
|
|
KEY_TYPE_TEST,
|
|
SMS_TYPE,
|
|
EMAIL_TYPE
|
|
)
|
|
from app.statsd_decorators import statsd
|
|
from app.utils import get_london_month_from_utc_column
|
|
|
|
|
|
@statsd(namespace="dao")
|
|
def get_billing_data_for_month(service_id, start_date, end_date, notification_type):
|
|
results = []
|
|
|
|
if notification_type == EMAIL_TYPE:
|
|
return billing_data_per_month_query(0, service_id, start_date, end_date, EMAIL_TYPE)
|
|
|
|
elif notification_type == SMS_TYPE:
|
|
rates = get_rates_for_daterange(start_date, end_date, SMS_TYPE)
|
|
|
|
if not rates:
|
|
return []
|
|
|
|
# so the start end date in the query are the valid from the rate, not the month
|
|
# - this is going to take some thought
|
|
for r, n in zip(rates, rates[1:]):
|
|
results.extend(
|
|
billing_data_per_month_query(
|
|
r.rate, service_id, max(r.valid_from, start_date),
|
|
min(n.valid_from, end_date), SMS_TYPE
|
|
)
|
|
)
|
|
results.extend(
|
|
billing_data_per_month_query(
|
|
rates[-1].rate, service_id, max(rates[-1].valid_from, start_date),
|
|
end_date, SMS_TYPE
|
|
)
|
|
)
|
|
|
|
return results
|
|
|
|
|
|
@statsd(namespace="dao")
|
|
def get_monthly_billing_data(service_id, year):
|
|
start_date, end_date = get_financial_year(year)
|
|
rates = get_rates_for_daterange(start_date, end_date, SMS_TYPE)
|
|
|
|
if not rates:
|
|
return []
|
|
|
|
result = []
|
|
for r, n in zip(rates, rates[1:]):
|
|
result.extend(billing_data_per_month_query(r.rate, service_id, r.valid_from, n.valid_from, SMS_TYPE))
|
|
result.extend(billing_data_per_month_query(rates[-1].rate, service_id, rates[-1].valid_from, end_date, SMS_TYPE))
|
|
|
|
return [(datetime.strftime(x[0], "%B"), x[1], x[2], x[3], x[4], x[5]) for x in result]
|
|
|
|
|
|
def billing_data_filter(notification_type, start_date, end_date, service_id):
|
|
return [
|
|
NotificationHistory.notification_type == notification_type,
|
|
NotificationHistory.created_at.between(start_date, end_date),
|
|
NotificationHistory.service_id == service_id,
|
|
NotificationHistory.status.in_(NOTIFICATION_STATUS_TYPES_BILLABLE),
|
|
NotificationHistory.key_type != KEY_TYPE_TEST
|
|
]
|
|
|
|
|
|
def get_rates_for_daterange(start_date, end_date, notification_type):
|
|
rates = Rate.query.filter(Rate.notification_type == notification_type).order_by(Rate.valid_from).all()
|
|
|
|
if not rates:
|
|
return []
|
|
|
|
results = []
|
|
for current_rate, current_rate_expiry_date in zip(rates, rates[1:]):
|
|
if is_between(current_rate.valid_from, start_date, end_date) or \
|
|
is_between(current_rate_expiry_date.valid_from - timedelta(microseconds=1), start_date, end_date):
|
|
results.append(current_rate)
|
|
|
|
if is_between(rates[-1].valid_from, start_date, end_date):
|
|
results.append(rates[-1])
|
|
|
|
if not results:
|
|
for x in reversed(rates):
|
|
if start_date >= x.valid_from:
|
|
results.append(x)
|
|
break
|
|
|
|
return results
|
|
|
|
|
|
def is_between(date, start_date, end_date):
|
|
return start_date <= date <= end_date
|
|
|
|
|
|
@statsd(namespace="dao")
|
|
def billing_data_per_month_query(rate, service_id, start_date, end_date, notification_type):
|
|
month = get_london_month_from_utc_column(NotificationHistory.created_at)
|
|
if notification_type == SMS_TYPE:
|
|
filter_subq = func.sum(NotificationHistory.billable_units).label('billing_units')
|
|
elif notification_type == EMAIL_TYPE:
|
|
filter_subq = func.count(NotificationHistory.billable_units).label('billing_units')
|
|
|
|
results = db.session.query(
|
|
month.label('month'),
|
|
filter_subq,
|
|
rate_multiplier().label('rate_multiplier'),
|
|
NotificationHistory.international,
|
|
NotificationHistory.notification_type,
|
|
cast(rate, Float()).label('rate')
|
|
).filter(
|
|
*billing_data_filter(notification_type, start_date, end_date, service_id)
|
|
).group_by(
|
|
NotificationHistory.notification_type,
|
|
month,
|
|
NotificationHistory.rate_multiplier,
|
|
NotificationHistory.international
|
|
).order_by(
|
|
month,
|
|
rate_multiplier()
|
|
).all()
|
|
|
|
return results
|
|
|
|
|
|
def rate_multiplier():
|
|
return cast(case([
|
|
(NotificationHistory.rate_multiplier == None, literal_column("'1'")), # noqa
|
|
(NotificationHistory.rate_multiplier != None, NotificationHistory.rate_multiplier), # noqa
|
|
]), Integer())
|