Сделать события-группы многие ко многим

[dyakovri]

Иное поведение дублирует пары в расписании преподавателей и не дает возможности на карточке предмета посмотреть все смежные группы

[dyakovri]

Код для миграции существующих событий

import sqlalchemy as sa
e = sa.create_engine(pwd)

res = e.execute("""
    with inner_q as (
        select 
            name, 
            start_ts, 
            end_ts, 
            array_agg(id) as old_ids, 
            array_agg(group_id) as group_ids, 
            count(*) as groups_incl
        from api_timetable."event" e 
        where not is_deleted
        group by name, start_ts, end_ts
    )
    select old_ids, group_ids
    from inner_q
""").all()

def merge_groups(res):
    to_delete = []
    pairs = []
    for old_ids, group_ids in res:
        save = old_ids.pop()
        to_delete.extend(old_ids)
        pairs.extend([(save, x) for x in group_ids])
    return to_delete, pairs

def test_merge_groups():
    res = [
        [[1,2,3], [10,11,12]],
        [[4,5,6], [10,11]],
        [[7], [10]]
    ]
    to_delete, pairs = merge_groups(res)
    assert to_delete == [1,2,4,5], to_delete
    assert pairs == [
        (3,10),
        (3,11),
        (3,12),
        (6,10),
        (6,11),
        (7,10),
    ], pairs
    
test_merge_groups()

to_delete, pairs = merge_groups(res)
delete_old_events_query = f"""UPDATE events SET is_deleted = True WHERE id IN [{', '.join([str(x) for x in to_delete])}]"""
create_new_links_query = f"""INSERT INTO events_groups(event_id, group_id) VALUES ({'), ('.join([str(x) + ', ' + str(y) for x, y in pairs])})"""

Общая идея:

1. По названию пары, времени ее начала и окончания объединяем события, получаем массивы события-группы
2. Добавляем к удалению события, кроме последнего из списка, получаем список событий к удалению
3. Последнее событие множим на все группы, получаем список пар id события-группа
4. Генерим SQL код

Остается только выполнить SQL

[grigoriev-semyon]

done in #95