feat: migrate upgrade when tenant_id is null

langgenius · Jul 8, 2024 · 5cc539a · 5cc539a
1 parent 07044d6
commit 5cc539a
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/api/migrations/versions/161cadc1af8d_add_dataset_permission_tenant_id.py b/api/migrations/versions/161cadc1af8d_add_dataset_permission_tenant_id.py
@@ -35,7 +35,16 @@ def upgrade():
         """
         )
 
-        # Step 3: Alter column to NOT NULL
+        # Step 3: Update the column to set a default value for existing rows
+        op.execute(
+            """
+            UPDATE dataset_permissions
+            SET tenant_id = '00000000-0000-0000-0000-000000000000'
+            WHERE tenant_id IS NULL
+            """
+        )
+
+        # Step 4: Alter column to NOT NULL
         op.alter_column('dataset_permissions', 'tenant_id', nullable=False)
 
     # ### end Alembic commands ###