Fix migration script to work with all deployments

MerginMaps · Nov 11, 2024 · 5bdeef7 · 5bdeef7
1 parent 3bdd3dd
commit 5bdeef7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/server/migrations/community/d02961c7416c_add_project_member_table.py b/server/migrations/community/d02961c7416c_add_project_member_table.py
@@ -18,6 +18,9 @@
 
 
 def upgrade():
+    conn = op.get_bind()
+    conn.execute(sa.text("CREATE EXTENSION IF NOT EXISTS intarray;"))
+
     op.create_table(
         "project_member",
         sa.Column("project_id", postgresql.UUID(as_uuid=True), nullable=False),
@@ -50,6 +53,7 @@ def upgrade():
     data_upgrade()
 
     op.drop_table("project_access")
+    conn.execute(sa.text("DROP EXTENSION IF EXISTS intarray;"))
 
 
 def downgrade():