239 lines
7.7 KiB
Python
239 lines
7.7 KiB
Python
from datetime import datetime, timedelta
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from sqlalchemy import select, and_
|
||
import pytz
|
||
|
||
from backend.database import Task, Event, WeeklyTaskException, EventException
|
||
from backend.models import ScheduleItem
|
||
|
||
TZ = pytz.timezone("Europe/Moscow")
|
||
|
||
|
||
def get_weekday_from_date(date_str: str) -> int:
|
||
"""Получить день недели (0=Monday, 6=Sunday) из даты."""
|
||
dt = datetime.strptime(date_str, "%Y-%m-%d")
|
||
return dt.weekday()
|
||
|
||
|
||
def get_week_range(date_str: str = None):
|
||
"""Получить диапазон дат недели (понедельник-воскресенье)."""
|
||
if date_str:
|
||
base_date = datetime.strptime(date_str, "%Y-%m-%d").date()
|
||
else:
|
||
base_date = datetime.now(TZ).date()
|
||
|
||
days_since_monday = base_date.weekday()
|
||
monday = base_date - timedelta(days=days_since_monday)
|
||
sunday = monday + timedelta(days=6)
|
||
|
||
return monday.strftime("%Y-%m-%d"), sunday.strftime("%Y-%m-%d")
|
||
|
||
|
||
def format_date_russian(date_str: str) -> str:
|
||
"""Форматировать дату в русском формате: 'Вторник, 2 декабря'."""
|
||
dt = datetime.strptime(date_str, "%Y-%m-%d")
|
||
|
||
weekdays = ["Понедельник", "Вторник", "Среда", "Четверг", "Пятница", "Суббота", "Воскресенье"]
|
||
months = [
|
||
"января", "февраля", "марта", "апреля", "мая", "июня",
|
||
"июля", "августа", "сентября", "октября", "ноября", "декабря"
|
||
]
|
||
|
||
weekday = weekdays[dt.weekday()]
|
||
month = months[dt.month - 1]
|
||
|
||
return f"{weekday}, {dt.day} {month}"
|
||
|
||
|
||
def _time_to_minutes(value: str) -> int:
|
||
hour, minute = map(int, value.split(":"))
|
||
return hour * 60 + minute
|
||
|
||
|
||
async def check_event_overlap(
|
||
db: AsyncSession,
|
||
date: str,
|
||
start_time: str,
|
||
duration_min: int,
|
||
exclude_id: int = None,
|
||
occurrence_date: str = None
|
||
) -> bool:
|
||
"""Проверить пересечение по времени с разовыми и weekly events."""
|
||
start_minutes = _time_to_minutes(start_time)
|
||
end_minutes = start_minutes + duration_min
|
||
|
||
events = await materialize_events(db, date, date)
|
||
|
||
for event in events:
|
||
if exclude_id and event.id == exclude_id and (occurrence_date is None or event.date == occurrence_date):
|
||
continue
|
||
|
||
ev_start_minutes = _time_to_minutes(event.start_time)
|
||
ev_end_minutes = ev_start_minutes + (event.duration_min or 0)
|
||
if not (end_minutes <= ev_start_minutes or start_minutes >= ev_end_minutes):
|
||
return True
|
||
|
||
return False
|
||
|
||
|
||
async def materialize_events(
|
||
db: AsyncSession,
|
||
from_date: str,
|
||
to_date: str
|
||
) -> list[ScheduleItem]:
|
||
items = []
|
||
|
||
result = await db.execute(
|
||
select(Event).where(
|
||
and_(
|
||
Event.date >= from_date,
|
||
Event.date <= to_date,
|
||
Event.repeat_weekly == False
|
||
)
|
||
)
|
||
)
|
||
one_off_events = result.scalars().all()
|
||
for event in one_off_events:
|
||
items.append(ScheduleItem(
|
||
kind="event",
|
||
id=event.id,
|
||
date=event.date,
|
||
source_date=event.date,
|
||
title=event.title,
|
||
start_time=event.start_time,
|
||
duration_min=event.duration_min,
|
||
repeat_weekly=False
|
||
))
|
||
|
||
result = await db.execute(select(Event).where(Event.repeat_weekly == True))
|
||
weekly_events = result.scalars().all()
|
||
if not weekly_events:
|
||
return items
|
||
|
||
result = await db.execute(
|
||
select(EventException).where(
|
||
and_(
|
||
EventException.date >= from_date,
|
||
EventException.date <= to_date
|
||
)
|
||
)
|
||
)
|
||
exceptions = result.scalars().all()
|
||
exception_dict = {(ex.event_id, ex.date): ex for ex in exceptions}
|
||
|
||
from_dt = datetime.strptime(from_date, "%Y-%m-%d").date()
|
||
to_dt = datetime.strptime(to_date, "%Y-%m-%d").date()
|
||
current_date = from_dt
|
||
|
||
while current_date <= to_dt:
|
||
date_str = current_date.strftime("%Y-%m-%d")
|
||
weekday = current_date.weekday()
|
||
|
||
for event in weekly_events:
|
||
base_date = datetime.strptime(event.date, "%Y-%m-%d").date()
|
||
if current_date < base_date or event.weekday != weekday:
|
||
continue
|
||
|
||
exception = exception_dict.get((event.id, date_str))
|
||
if exception:
|
||
if exception.action == "delete":
|
||
continue
|
||
items.append(ScheduleItem(
|
||
kind="event",
|
||
id=event.id,
|
||
date=exception.replacement_date or date_str,
|
||
source_date=date_str,
|
||
title=exception.replacement_title or event.title,
|
||
start_time=exception.replacement_start_time or event.start_time,
|
||
duration_min=exception.replacement_duration_min or event.duration_min,
|
||
repeat_weekly=True
|
||
))
|
||
continue
|
||
|
||
items.append(ScheduleItem(
|
||
kind="event",
|
||
id=event.id,
|
||
date=date_str,
|
||
source_date=date_str,
|
||
title=event.title,
|
||
start_time=event.start_time,
|
||
duration_min=event.duration_min,
|
||
repeat_weekly=True
|
||
))
|
||
|
||
current_date += timedelta(days=1)
|
||
|
||
return items
|
||
|
||
|
||
async def materialize_weekly_tasks(
|
||
db: AsyncSession,
|
||
from_date: str,
|
||
to_date: str
|
||
) -> list[ScheduleItem]:
|
||
"""Материализовать weekly tasks в диапазоне дат."""
|
||
items = []
|
||
|
||
result = await db.execute(select(Task).where(Task.repeat_weekly == True))
|
||
weekly_tasks = result.scalars().all()
|
||
if not weekly_tasks:
|
||
return items
|
||
|
||
result = await db.execute(
|
||
select(WeeklyTaskException).where(
|
||
and_(
|
||
WeeklyTaskException.date >= from_date,
|
||
WeeklyTaskException.date <= to_date
|
||
)
|
||
)
|
||
)
|
||
exceptions = result.scalars().all()
|
||
exception_dict = {}
|
||
for ex in exceptions:
|
||
if ex.task_id is not None:
|
||
exception_dict[(ex.task_id, ex.date)] = ex
|
||
else:
|
||
exception_dict[(None, ex.date, ex.weekday)] = ex
|
||
|
||
from_dt = datetime.strptime(from_date, "%Y-%m-%d").date()
|
||
to_dt = datetime.strptime(to_date, "%Y-%m-%d").date()
|
||
|
||
current_date = from_dt
|
||
while current_date <= to_dt:
|
||
date_str = current_date.strftime("%Y-%m-%d")
|
||
weekday = current_date.weekday()
|
||
|
||
for task in weekly_tasks:
|
||
base_date = datetime.strptime(task.date, "%Y-%m-%d").date()
|
||
if current_date < base_date or task.weekday != weekday:
|
||
continue
|
||
|
||
exception = exception_dict.get((task.id, date_str))
|
||
if not exception:
|
||
exception = exception_dict.get((None, date_str, weekday))
|
||
|
||
if exception:
|
||
if exception.action == "delete":
|
||
continue
|
||
items.append(ScheduleItem(
|
||
kind="task",
|
||
id=task.id,
|
||
date=exception.replacement_date or date_str,
|
||
source_date=date_str,
|
||
title=exception.replacement_title or task.title,
|
||
repeat_weekly=True
|
||
))
|
||
continue
|
||
|
||
items.append(ScheduleItem(
|
||
kind="task",
|
||
id=task.id,
|
||
date=date_str,
|
||
source_date=date_str,
|
||
title=task.title,
|
||
repeat_weekly=True
|
||
))
|
||
|
||
current_date += timedelta(days=1)
|
||
|
||
return items
|