Kenik
66 lines
2.8 KiB
Python
66 lines
2.8 KiB
Python
"""Расчёт дат для напоминалок админов: парсинг ввода и следующее срабатывание.
|
|
|
|
Не зависит от bot.py/database.py — чистая логика дат, легко тестируется отдельно.
|
|
"""
|
|
import calendar
|
|
from datetime import datetime, timedelta
|
|
from typing import Optional
|
|
|
|
DT_FORMAT = "%Y-%m-%d %H:%M:%S" # формат хранения в БД
|
|
INPUT_FORMAT = "%d.%m.%Y %H:%M" # формат ввода админом
|
|
|
|
RECURRENCES = ("once", "weekly", "monthly", "yearly")
|
|
|
|
|
|
def parse_reminder_datetime(text: str) -> Optional[datetime]:
|
|
"""Парсит дату/время в формате ДД.ММ.ГГГГ ЧЧ:ММ. None, если формат неверный."""
|
|
text = text.strip()
|
|
try:
|
|
return datetime.strptime(text, INPUT_FORMAT)
|
|
except ValueError:
|
|
return None
|
|
|
|
|
|
def _clamp_day(year: int, month: int, day: int) -> int:
|
|
"""День месяца, не превышающий число дней в этом месяце
|
|
(31.01 -> 28/29.02, 31.03 и т.п. — без ползучего дрейфа, см. compute_next_occurrence)."""
|
|
last_day = calendar.monthrange(year, month)[1]
|
|
return min(day, last_day)
|
|
|
|
|
|
def compute_next_occurrence(anchor: datetime, recurrence: str, after: datetime) -> datetime:
|
|
"""Следующее срабатывание строго после `after`, отталкиваясь от якорной даты `anchor`
|
|
(а не от предыдущего срабатывания) — так периодичность не «дрейфует» из-за клампинга
|
|
коротких месяцев. Если пропущено несколько периодов (бот был выключен) — промежуточные
|
|
пропускаются, возвращается ближайшее будущее относительно `after`.
|
|
"""
|
|
if recurrence not in RECURRENCES:
|
|
raise ValueError(f"Неизвестная периодичность: {recurrence}")
|
|
|
|
if recurrence == "weekly":
|
|
candidate = anchor
|
|
while candidate <= after:
|
|
candidate += timedelta(days=7)
|
|
return candidate
|
|
|
|
if recurrence == "monthly":
|
|
year, month = anchor.year, anchor.month
|
|
candidate = anchor
|
|
while candidate <= after:
|
|
month += 1
|
|
if month > 12:
|
|
month = 1
|
|
year += 1
|
|
day = _clamp_day(year, month, anchor.day)
|
|
candidate = candidate.replace(year=year, month=month, day=day)
|
|
return candidate
|
|
|
|
# yearly
|
|
year = anchor.year
|
|
candidate = anchor
|
|
while candidate <= after:
|
|
year += 1
|
|
day = _clamp_day(year, anchor.month, anchor.day)
|
|
candidate = candidate.replace(year=year, day=day)
|
|
return candidate
|