54 lines
2.6 KiB
Python
54 lines
2.6 KiB
Python
# bot/utils/date_utils.py
|
||
from datetime import datetime, timedelta
|
||
import pytz
|
||
|
||
def ensure_datetime(value):
|
||
"""
|
||
Приводит значение к объекту datetime с учетом временной зоны.
|
||
|
||
:param value: Значение даты (строка или datetime).
|
||
:return: Объект datetime.
|
||
:raises: TypeError, если передан некорректный тип.
|
||
"""
|
||
if isinstance(value, str):
|
||
try:
|
||
# Если строка соответствует формату ISO 8601
|
||
return datetime.fromisoformat(value)
|
||
except ValueError:
|
||
raise ValueError(f"Некорректный формат даты: {value}")
|
||
elif isinstance(value, datetime):
|
||
return value
|
||
else:
|
||
raise TypeError(f"Ожидался тип str или datetime, получено: {type(value)}")
|
||
|
||
def get_period_dates(period, now=None):
|
||
"""
|
||
Возвращает диапазон дат (start_date, end_date) для заданного периода.
|
||
|
||
:param period: Период (строка: 'today', 'yesterday', 'last_week', 'last_month').
|
||
:param now: Текущая дата/время (опционально).
|
||
:return: Кортеж (start_date, end_date).
|
||
:raises: ValueError, если период не поддерживается.
|
||
"""
|
||
if now is None:
|
||
now = datetime.now(pytz.UTC)
|
||
|
||
if period == "today":
|
||
start_date = now.replace(hour=0, minute=0, second=0, microsecond=0)
|
||
end_date = now.replace(hour=23, minute=59, second=59, microsecond=999999)
|
||
elif period == "yesterday":
|
||
start_date = (now - timedelta(days=1)).replace(hour=0, minute=0, second=0, microsecond=0)
|
||
end_date = (now - timedelta(days=1)).replace(hour=23, minute=59, second=59, microsecond=999999)
|
||
elif period == "last_week":
|
||
start_date = (now - timedelta(days=now.weekday() + 7)).replace(hour=0, minute=0, second=0, microsecond=0)
|
||
end_date = (start_date + timedelta(days=6)).replace(hour=23, minute=59, second=59, microsecond=999999)
|
||
elif period == "last_month":
|
||
first_day_of_current_month = now.replace(day=1, hour=0, minute=0, second=0, microsecond=0)
|
||
last_day_of_previous_month = first_day_of_current_month - timedelta(days=1)
|
||
start_date = last_day_of_previous_month.replace(day=1)
|
||
end_date = last_day_of_previous_month.replace(hour=23, minute=59, second=59, microsecond=999999)
|
||
else:
|
||
raise ValueError(f"Неподдерживаемый период: {period}")
|
||
|
||
return start_date, end_date
|