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

rename which_financial_year to get_financial_year_for_datetime

Browse Source 
also fix bug on the very second of rollover, march 31st 23:00:00, and
add tests
This commit is contained in:
Leo Hemsted
2019-08-28 14:27:08 +01:00
parent 2abcda47a3
commit b7e8f1baa2
3 changed files with 24 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/dao/date_util.py
View File

@@ -63,8 +63,9 @@ def get_current_financial_year_start_year():
return financial_year_start
def which_financial_year(start_date):
if start_date <= get_april_fools(int(start_date.strftime('%Y'))):
return int(start_date.strftime('%Y')) - 1
def get_financial_year_for_datetime(start_date):
year = int(start_date.strftime('%Y'))
if start_date < get_april_fools(year):
return year - 1
else:
return int(start_date.strftime('%Y'))
return year

6
app/dao/fact_billing_dao.py
View File

@@ -9,7 +9,7 @@ from app import db
from app.dao.date_util import (
get_april_fools as financial_year_start,
get_financial_year,
which_financial_year
get_financial_year_for_datetime
)
from app.models import (
@@ -33,7 +33,7 @@ from app.utils import get_london_midnight_in_utc
def fetch_sms_free_allowance_remainder(start_date):
# ASSUMPTION: AnnualBilling has been populated for year.
billing_year = which_financial_year(start_date)
billing_year = get_financial_year_for_datetime(start_date)
print(billing_year)
start_of_year = convert_utc_to_bst(financial_year_start(billing_year))
query = db.session.query(
@@ -60,7 +60,7 @@ def fetch_sms_free_allowance_remainder(start_date):
def fetch_sms_billing_for_all_services(start_date, end_date):
# ASSUMPTION: AnnualBilling has been populated for year.
billing_year = which_financial_year(start_date)
billing_year = get_financial_year_for_datetime(start_date)
free_allowance_remainder = fetch_sms_free_allowance_remainder(start_date).subquery()
sms_billable_units = func.sum(FactBilling.billable_units * FactBilling.rate_multiplier)
sms_remainder = func.coalesce(free_allowance_remainder.c.sms_remainder, AnnualBilling.free_sms_fragment_limit)