mirror of
https://github.com/GSA/notifications-api.git
synced 2026-07-16 03:10:48 -04:00
120 lines
3.5 KiB
Python
120 lines
3.5 KiB
Python
import calendar
|
|
import pytz
|
|
|
|
from datetime import date, datetime, time, timedelta
|
|
|
|
from app.utils import utc_now
|
|
|
|
|
|
def get_months_for_financial_year(year):
|
|
return [
|
|
month for month in (get_months_for_year(1, 13, year)) 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_calendar_year(year):
|
|
return get_new_years(year), get_new_years(year + 1) - timedelta(microseconds=1)
|
|
|
|
|
|
def get_calendar_year_dates(year):
|
|
year_start_datetime, year_end_datetime = get_calendar_year(year)
|
|
|
|
return (year_start_datetime.date(), year_end_datetime.date())
|
|
|
|
|
|
def get_current_calendar_year():
|
|
now = utc_now()
|
|
current_year = int(now.strftime("%Y"))
|
|
year = current_year
|
|
return get_calendar_year(year)
|
|
|
|
|
|
def get_new_years(year):
|
|
return datetime(year, 1, 1, 0, 0, 0)
|
|
|
|
|
|
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 first_day, last_day
|
|
|
|
|
|
def get_current_calendar_year_start_year():
|
|
now = datetime.now()
|
|
financial_year_start = now.year
|
|
start_date, end_date = get_calendar_year(now.year)
|
|
if now < start_date:
|
|
financial_year_start = financial_year_start - 1
|
|
return financial_year_start
|
|
|
|
|
|
def get_calendar_year_for_datetime(start_date):
|
|
if isinstance(start_date, date):
|
|
start_date = datetime.combine(start_date, time.min)
|
|
|
|
year = int(start_date.strftime("%Y"))
|
|
if start_date < get_new_years(year):
|
|
return year - 1
|
|
else:
|
|
return year
|
|
|
|
|
|
def get_number_of_days_for_month(year, month):
|
|
return calendar.monthrange(year, month)[1]
|
|
|
|
|
|
def generate_date_range(start_date, end_date=None, days=0):
|
|
if end_date:
|
|
current_date = start_date
|
|
while current_date <= end_date:
|
|
try:
|
|
yield current_date.date()
|
|
except ValueError:
|
|
pass
|
|
current_date += timedelta(days=1)
|
|
elif days > 0:
|
|
end_date = start_date + timedelta(days=days)
|
|
current_date = start_date
|
|
while current_date < end_date:
|
|
try:
|
|
yield current_date.date()
|
|
except ValueError:
|
|
pass
|
|
current_date += timedelta(days=1)
|
|
else:
|
|
return "An end_date or number of days must be specified"
|
|
|
|
|
|
def build_local_and_utc_date_range(
|
|
start_date_str: str,
|
|
days: int = 7,
|
|
timezone: str = "UTC"
|
|
):
|
|
"""
|
|
Convert date to local range based on timezone
|
|
"""
|
|
|
|
user_timezone = pytz.timezone(timezone)
|
|
local_end_date = datetime.strptime(start_date_str, "%Y-%m-%d").replace(tzinfo=user_timezone)
|
|
# Subtract (days - 1) so the entire final day is included in the range
|
|
local_start_date = local_end_date - timedelta(days=days)
|
|
|
|
|
|
# Convert to UTC for database queries
|
|
utc_start_date = local_start_date.astimezone(pytz.utc).replace(hour=0, minute=0, second=0)
|
|
utc_end_date = local_end_date.astimezone(pytz.utc).replace(hour=23, minute=59, second=59)
|
|
|
|
return (local_start_date, utc_start_date, utc_end_date)
|