From 0913807714dcf62b4ff12ee4396b3e45ed7cb52f Mon Sep 17 00:00:00 2001
From: VineetBala-AOT <vineet.balachandran@aot-technologies.com>
Date: Mon, 29 Jan 2024 10:40:23 -0800
Subject: [PATCH] Updating migration file for engagement metadata

---
 .../45f89f245e3d_engagement_metadata.py       |  1 +
 ...ec0128056a33_rework_engagement_metadata.py | 30 ++++++++++++++-----
 2 files changed, 23 insertions(+), 8 deletions(-)

diff --git a/met-api/migrations/versions/45f89f245e3d_engagement_metadata.py b/met-api/migrations/versions/45f89f245e3d_engagement_metadata.py
index bd216de6f..a9e71b9f0 100644
--- a/met-api/migrations/versions/45f89f245e3d_engagement_metadata.py
+++ b/met-api/migrations/versions/45f89f245e3d_engagement_metadata.py
@@ -22,6 +22,7 @@ def upgrade():
     sa.Column('created_date', sa.DateTime(), nullable=False),
     sa.Column('updated_date', sa.DateTime(), nullable=True),
     sa.Column('engagement_id', sa.Integer(), nullable=False),
+    sa.Column('project_metadata', postgresql.JSONB(astext_type=sa.Text()), nullable=True),
     sa.Column('created_by', sa.String(length=50), nullable=True),
     sa.Column('updated_by', sa.String(length=50), nullable=True),
     sa.ForeignKeyConstraint(['engagement_id'], ['engagement.id'], ondelete='CASCADE'),
diff --git a/met-api/migrations/versions/ec0128056a33_rework_engagement_metadata.py b/met-api/migrations/versions/ec0128056a33_rework_engagement_metadata.py
index 25d0a79de..fec84a620 100644
--- a/met-api/migrations/versions/ec0128056a33_rework_engagement_metadata.py
+++ b/met-api/migrations/versions/ec0128056a33_rework_engagement_metadata.py
@@ -41,23 +41,37 @@ def upgrade():
     sa.PrimaryKeyConstraint('id', name='pk_engagement_metadata_taxa'),
     sa.UniqueConstraint('id', name='uq_engagement_metadata_taxa_id')
     )
-    # remove old primary key constraint from engagement_metadata.engagement_id
+    # remove old data from engagement_metadata table
+    op.execute('DELETE FROM engagement_metadata')
+
+    # Drop the existing primary key constraint
     op.drop_constraint('engagement_metadata_pkey', 'engagement_metadata', type_='primary')
+
+    # Create a new index on engagement_metadata_taxa
     op.create_index(op.f('ix_engagement_metadata_taxa_position'), 'engagement_metadata_taxa', ['position'], unique=False)
+
+    # Add new columns to engagement_metadata
     op.add_column('engagement_metadata', sa.Column('id', sa.Integer(), nullable=False))
-    # add new primary key constraint on engagement_metadata.id
+    op.add_column('engagement_metadata', sa.Column('taxon_id', sa.Integer(), nullable=False))
+    op.add_column('engagement_metadata', sa.Column('value', sa.Text(), nullable=False))
+
+    # Create a new primary key constraint on the 'id' column
     op.create_primary_key('pk_engagement_metadata', 'engagement_metadata', ['id'])
-    # add autoincrement to engagement_metadata.id by creating a sequence
+
+    # Create a new sequence and set it as the default for the 'id' column
     op.execute('CREATE SEQUENCE engagement_metadata_id_seq START 1')
     op.execute('ALTER TABLE engagement_metadata ALTER COLUMN id SET DEFAULT nextval(\'engagement_metadata_id_seq\')')
     op.execute('ALTER SEQUENCE engagement_metadata_id_seq OWNED BY engagement_metadata.id')
-    # remove not-null constraint from engagement_metadata.engagement_id
+
+    # Remove the not-null constraint from 'engagement_id'
     op.alter_column('engagement_metadata', 'engagement_id', existing_type=sa.INTEGER(), nullable=True)
 
-    op.add_column('engagement_metadata', sa.Column('taxon_id', sa.Integer(), nullable=False))
-    op.add_column('engagement_metadata', sa.Column('value', sa.Text(), nullable=False))
+    # Create a foreign key constraint
     op.create_foreign_key('fk_engagement_meta_taxon', 'engagement_metadata', 'engagement_metadata_taxa', ['taxon_id'], ['id'], ondelete='CASCADE')
+
+    # Remove the 'project_tracking_id' column
     op.drop_column('engagement_metadata', 'project_tracking_id')
+
     # add default taxa for default tenant
     default_short_name = current_app.config.get('DEFAULT_TENANT_SHORT_NAME')
     tenant_id = TenantModel.find_by_short_name(default_short_name).id
@@ -136,10 +150,10 @@ def downgrade():
     op.add_column('engagement_metadata', sa.Column('project_tracking_id', sa.VARCHAR(length=100), autoincrement=False, nullable=True))
     op.alter_column('engagement_metadata', 'engagement_id', existing_type=sa.INTEGER(), nullable=False)
     op.drop_constraint('fk_engagement_meta_taxon', 'engagement_metadata', type_='foreignkey')
-    op.drop_column('engagement_metadata', 'value')
-    op.drop_column('engagement_metadata', 'taxon_id')
     # remove primary key constraint from engagement_metadata.id
     op.drop_constraint('pk_engagement_metadata', 'engagement_metadata', type_='primary')
+    op.drop_column('engagement_metadata', 'value')
+    op.drop_column('engagement_metadata', 'taxon_id')
     op.drop_column('engagement_metadata', 'id')
     op.drop_index(op.f('ix_engagement_metadata_taxa_position'), table_name='engagement_metadata_taxa')
     # add primary key constraint to engagement_metadata.engagement_id