From cf6d0b35ef5a9c0f15c7b7cac30a2099108a6f3a Mon Sep 17 00:00:00 2001 From: Automated Action Date: Tue, 17 Jun 2025 02:56:06 +0000 Subject: [PATCH] Fix user role migration script to add column before updating values --- migrations/versions/002_add_user_roles.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/migrations/versions/002_add_user_roles.py b/migrations/versions/002_add_user_roles.py index 454d843..947f1b2 100644 --- a/migrations/versions/002_add_user_roles.py +++ b/migrations/versions/002_add_user_roles.py @@ -20,7 +20,12 @@ def upgrade(): # Add role column to users table with op.batch_alter_table('users', schema=None) as batch_op: batch_op.add_column(sa.Column('role', sa.Enum('admin', 'user', name='userrole'), nullable=True)) - batch_op.execute("UPDATE users SET role = 'user' WHERE role IS NULL") + + # Update existing users to have the 'user' role + op.execute("UPDATE users SET role = 'user' WHERE role IS NULL") + + # Make the column non-nullable with a default value + with op.batch_alter_table('users', schema=None) as batch_op: batch_op.alter_column('role', nullable=False, server_default='user')