mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-16 10:12:32 -05:00
6f7793d761
- moved get_current_financial_year_start_year from service.utils to dao.date_utils - Moved logic for data persistence from rest to dao when updating records in db
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
import pytz
|
|
|
|
from app.utils import convert_bst_to_utc
|
|
|
|
|
|
def get_months_for_financial_year(year):
|
|
return [
|
|
convert_bst_to_utc(month) for month in (
|
|
get_months_for_year(4, 13, year) +
|
|
get_months_for_year(1, 4, year + 1)
|
|
)
|
|
if month < datetime.now()
|
|
]
|
|
|
|
|
|
def get_months_for_year(start, end, year):
|
|
return [datetime(year, month, 1) for month in range(start, end)]
|
|
|
|
|
|
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 because 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_and_end_date_in_utc(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 convert_bst_to_utc(first_day), convert_bst_to_utc(last_day)
|
|
|
|
|
|
def get_current_financial_year_start_year():
|
|
now = datetime.now()
|
|
financial_year_start = now.year
|
|
start_date, end_date = get_financial_year(now.year)
|
|
if now < start_date:
|
|
financial_year_start = financial_year_start - 1
|
|
return financial_year_start
|