feat: clinic setup update

fix: migration issue
2025-05-13 17:01:48 +05:30
parent dabc7dd308
commit 13023e5913
12 changed files with 280 additions and 74 deletions
@@ -0,0 +1,127 @@
+"""fix enum clinic doctor
+
+Revision ID: 0ce7107c1910
+Revises: a3a9b7d17bdd
+Create Date: 2025-05-13 12:20:49.384154
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+from sqlalchemy import text
+
+# revision identifiers, used by Alembic.
+revision: str = '0ce7107c1910'
+down_revision: Union[str, None] = 'a3a9b7d17bdd'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    
+    # Check if the enum type exists before creating it
+    conn = op.get_bind()
+    result = conn.execute(text("SELECT EXISTS(SELECT 1 FROM pg_type WHERE typname = 'clinicdoctortype')"))
+    enum_exists = result.scalar()
+    
+    if not enum_exists:
+        # Create the new enum type if it doesn't exist
+        op.execute(text("CREATE TYPE clinicdoctortype AS ENUM ('DOCTOR', 'NURSE')"))
+    
+    # Check if the new_role column already exists
+    inspector = sa.inspect(conn)
+    columns = inspector.get_columns('clinic_doctors')
+    column_names = [column['name'] for column in columns]
+    
+    if 'new_role' not in column_names:
+        # Add a temporary column with the new type
+        op.add_column('clinic_doctors', sa.Column('new_role', postgresql.ENUM('DOCTOR', 'NURSE', name='clinicdoctortype'), nullable=True))
+        
+        # Update the temporary column with values based on the old column
+        op.execute(text("UPDATE clinic_doctors SET new_role = 'DOCTOR' WHERE role = 'DIRECTOR'"))
+        op.execute(text("UPDATE clinic_doctors SET new_role = 'NURSE' WHERE role = 'PRACTICE_MANAGER'"))
+        
+        # Drop the old column and rename the new one
+        op.drop_column('clinic_doctors', 'role')
+        op.alter_column('clinic_doctors', 'new_role', new_column_name='role')
+    
+    # We need to handle the users table that depends on the clinicuserroles enum
+    # First check if the users table has a clinicRole column that uses the enum
+    has_clinic_role = False
+    try:
+        user_columns = inspector.get_columns('users')
+        for column in user_columns:
+            if column['name'] == 'clinicRole':
+                has_clinic_role = True
+                break
+    except:
+        # Table might not exist
+        pass
+    
+    if has_clinic_role:
+        # We need to update the users table to not use the enum before dropping it
+        # First, create a temporary column with a string type
+        op.add_column('users', sa.Column('temp_clinic_role', sa.String(), nullable=True))
+        
+        # Copy the values - use double quotes to preserve case sensitivity in PostgreSQL
+        op.execute(text('UPDATE users SET temp_clinic_role = "clinicRole"::text'))
+        
+        # Drop the old column and rename the new one
+        op.drop_column('users', 'clinicRole')
+        op.alter_column('users', 'temp_clinic_role', new_column_name='clinicRole')
+    
+    # Now we can safely drop the old enum type
+    op.execute(text("DROP TYPE IF EXISTS clinicuserroles"))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    
+    # Create the old enum type
+    op.execute(text("CREATE TYPE clinicuserroles AS ENUM ('DIRECTOR', 'PRACTICE_MANAGER')"))
+    
+    # Add a temporary column with the old type
+    op.add_column('clinic_doctors', sa.Column('old_role', postgresql.ENUM('DIRECTOR', 'PRACTICE_MANAGER', name='clinicuserroles'), nullable=True))
+    
+    # Update the temporary column with values based on the new column
+    op.execute(text("UPDATE clinic_doctors SET old_role = 'DIRECTOR' WHERE role = 'DOCTOR'"))
+    op.execute(text("UPDATE clinic_doctors SET old_role = 'PRACTICE_MANAGER' WHERE role = 'NURSE'"))
+    
+    # Drop the new column and rename the old one
+    op.drop_column('clinic_doctors', 'role')
+    op.alter_column('clinic_doctors', 'old_role', new_column_name='role')
+    
+    # If we modified the users table in the upgrade, we need to restore it
+    conn = op.get_bind()
+    inspector = sa.inspect(conn)
+    has_clinic_role = False
+    try:
+        user_columns = inspector.get_columns('users')
+        for column in user_columns:
+            if column['name'] == 'clinicRole':
+                has_clinic_role = True
+                break
+    except:
+        pass
+    
+    if has_clinic_role:
+        # Create a temporary column with the enum type
+        op.add_column('users', sa.Column('temp_clinic_role', postgresql.ENUM('DIRECTOR', 'PRACTICE_MANAGER', name='clinicuserroles'), nullable=True))
+        
+        # Copy the values (with appropriate conversion) - use double quotes to preserve case sensitivity
+        op.execute(text('UPDATE users SET temp_clinic_role = "clinicRole"::clinicuserroles'))
+        
+        # Drop the old column and rename the new one
+        op.drop_column('users', 'clinicRole')
+        op.alter_column('users', 'temp_clinic_role', new_column_name='clinicRole')
+    
+    # Drop the new enum type
+    op.execute(text("DROP TYPE IF EXISTS clinicdoctortype"))
+    # ### end Alembic commands ###
+    # ### end Alembic commands ###