plugins devel

bot/utils/date_utils.py

@@ -0,0 +1,53 @@
+# 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