mirror of
https://github.com/GSA/notifications-api.git
synced 2026-01-03 00:00:44 -05:00
9400988d72
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.
32 lines
1.3 KiB
Python
32 lines
1.3 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
import pytz
|
|
|
|
|
|
def get_financial_year(year):
|
|
return get_april_fools(year), get_april_fools(year + 1) - timedelta(microseconds=1)
|
|
|
|
|
|
def get_april_fools(year):
|
|
"""
|
|
This function converts the start of the financial year April 1, 00:00 as BST (British Standard Time) to UTC,
|
|
the tzinfo is lastly removed from the datetime becasue the database stores the timestamps without timezone.
|
|
:param year: the year to calculate the April 1, 00:00 BST for
|
|
:return: the datetime of April 1 for the given year, for example 2016 = 2016-03-31 23:00:00
|
|
"""
|
|
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
|