soc/iris-web/source/app/alembic/versions/c29ef01617f5...

"""Migrate notes directories

Revision ID: c29ef01617f5
Revises: e0df2de997cc
Create Date: 2023-12-30 17:24:36.430292

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy import text, Table, MetaData

from app.alembic.alembic_utils import _table_has_column

# revision identifiers, used by Alembic.
revision = 'c29ef01617f5'
down_revision = 'e0df2de997cc'
branch_labels = None
depends_on = None


def upgrade():
    if not _table_has_column('notes', 'directory_id'):
        metadata = MetaData()

        notes_group = Table('notes_group', metadata, autoload_with=op.get_bind())

        result = op.get_bind().execute(sa.select(sa.func.max(notes_group.c.group_id)))
        max_group_id = result.scalar() or 0

        # Set the starting value for the primary key in the note_directory table
        op.execute(text(f"SELECT setval('note_directory_id_seq', {max_group_id + 1})"))

        op.execute(text("""
            INSERT INTO note_directory (id, name, parent_id, case_id)
            SELECT group_id, group_title, NULL, group_case_id
            FROM notes_group
        """))

        op.add_column('notes', sa.Column('directory_id', sa.BigInteger, sa.ForeignKey('note_directory.id')))

        op.execute(text("""
             UPDATE notes
             SET directory_id = (
                 SELECT group_id
                 FROM notes_group_link
                 WHERE notes_group_link.note_id = notes.note_id
             )
         """))

        # Commit the transaction
        op.execute(text("COMMIT"))

    return


def downgrade():
    pass