Monthly billing - part 1

This is still a work in progress but it would be good to get some eyes on it.
This commit includes creating and updating a row in the monthly billing table and a method to fetch the results.
There is a command to populate the monthly billing for a service and month so we can try it out.
The total cost at the moment are wrong, they do not take into account the free allowance - see notes below about adding that to the table.
Left to do:
create a nightly task to run to update the monthly totals.
create an endpoint to return the yearly billing, the current day will need to be calculated on the fly and added to the totals.
Add the free allowance into the total costs.

app/dao/date_util.py

@@ -16,3 +16,16 @@ def get_april_fools(year):
     """
     return pytz.timezone('Europe/London').localize(datetime(year, 4, 1, 0, 0, 0)).astimezone(pytz.UTC).replace(
         tzinfo=None)
+
+
+def get_month_start_end_date(month_year):
+    """
+     This function return the start and date of the month_year as UTC,
+     :param month_year: the datetime to calculate the start and end date for that month
+     :return: start_date, end_date, month
+    """
+    import calendar
+    _, num_days = calendar.monthrange(month_year.year, month_year.month)
+    first_day = datetime(month_year.year, month_year.month, 1, 0, 0, 0)
+    last_day = datetime(month_year.year, month_year.month, num_days, 23, 59, 59, 99999)
+    return first_day, last_day

app/dao/monthly_billing_dao.py

@@ -1,6 +1,44 @@
+from datetime import datetime
+
 from app import db
+from app.dao.notification_usage_dao import get_billing_data_for_month
+from app.models import MonthlyBilling, SMS_TYPE
 
 
-def update_monthly_billing(monthly_billing):
-    db.session.add(monthly_billing)
+def create_or_update_monthly_billing_sms(service_id, billing_month):
+    monthly = get_billing_data_for_month(service_id=service_id, billing_month=billing_month)
+    # update monthly
+    monthly_totals = _monthly_billing_data_to_json(monthly)
+    row = MonthlyBilling.query.filter_by(year=billing_month.year,
+                                         month=datetime.strftime(billing_month, "%B"),
+                                         notification_type='sms').first()
+    if row:
+        row.monthly_totals = monthly_totals
+    else:
+        row = MonthlyBilling(service_id=service_id,
+                             notification_type=SMS_TYPE,
+                             year=billing_month.year,
+                             month=datetime.strftime(billing_month, "%B"),
+                             monthly_totals=monthly_totals)
+    db.session.add(row)
     db.session.commit()
+
+
+def get_monthly_billing_sms(service_id, billing_month):
+    monthly = MonthlyBilling.query.filter_by(service_id=service_id,
+                                             year=billing_month.year,
+                                             month=datetime.strftime(billing_month, "%B"),
+                                             notification_type=SMS_TYPE).first()
+    return monthly
+
+
+def _monthly_billing_data_to_json(monthly):
+    # ('April', 6, 1, False, 'sms', 0.014)
+    #  (month, billing_units, rate_multiplier, international, notification_type, rate)
+    # total cost must take into account the free allowance.
+    # might be a good idea to capture free allowance in this table
+    return [{"billing_units": x[1],
+             "rate_multiplier": x[2],
+             "international": x[3],
+             "rate": x[5],
+             "total_cost": (x[1] * x[2]) * x[5]} for x in monthly]

app/dao/notification_usage_dao.py

@@ -6,7 +6,7 @@ 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.dao.date_util import get_financial_year, get_month_start_end_date
 from app.models import (NotificationHistory,
                         Rate,
                         NOTIFICATION_STATUS_TYPES_BILLABLE,
@@ -35,6 +35,21 @@ def get_yearly_billing_data(service_id, year):
     return sum(result, [])
 
 
+@statsd(namespace="dao")
+def get_billing_data_for_month(service_id, billing_month):
+    start_date, end_date = get_month_start_end_date(billing_month)
+    rates = get_rates_for_year(start_date, end_date, SMS_TYPE)
+    result = []
+    # 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:]):
+        result.extend(sms_billing_data_per_month_query(r.rate, service_id, max(r.valid_from, start_date),
+                                                       min(n.valid_from, end_date)))
+    result.extend(
+        sms_billing_data_per_month_query(rates[-1].rate, service_id, max(rates[-1].valid_from, start_date), end_date))
+
+    return result
+
+
 @statsd(namespace="dao")
 def get_monthly_billing_data(service_id, year):
     start_date, end_date = get_financial_year(year)

app/dao/provider_rates_dao.py

@@ -9,3 +9,8 @@ def create_provider_rates(provider_identifier, valid_from, rate):
 
     provider_rates = ProviderRates(provider_id=provider.id, valid_from=valid_from, rate=rate)
     db.session.add(provider_rates)
+
+
+@transactional
+def create_sms_rate(rate):
+    db.session.add(rate)