qa

Author: francesconazzaro

alembic/versions/a4e8be715296_add_deleted_as_new_status.py

@@ -5,25 +5,50 @@
 Create Date: 2024-07-25 13:13:11.955119
 
 """
+
 from alembic import op
 import sqlalchemy as sa
 from sqlalchemy.dialects.postgresql import ENUM
 
 
 # revision identifiers, used by Alembic.
-revision = 'a4e8be715296'
-down_revision = 'd5d4afc97d40'
+revision = "a4e8be715296"
+down_revision = "d5d4afc97d40"
 branch_labels = None
 depends_on = None
 
 
-old_status_enum = ENUM('pending', 'in_progress', 'completed', 'failed', name='status', create_type=False)
-new_status_enum = ENUM('pending', 'in_progress', 'completed', 'failed', 'deleted', name='status', create_type=False)
+old_status_enum = ENUM(
+    "pending", "in_progress", "completed", "failed", name="status", create_type=False
+)
+new_status_enum = ENUM(
+    "pending",
+    "in_progress",
+    "completed",
+    "failed",
+    "deleted",
+    name="status",
+    create_type=False,
+)
+
 
 def upgrade() -> None:
     # Add the new status to the enum
-    op.alter_column('system_requests', 'status', existing_type=old_status_enum, type_=new_status_enum, existing_nullable=False)
+    op.alter_column(
+        "system_requests",
+        "status",
+        existing_type=old_status_enum,
+        type_=new_status_enum,
+        existing_nullable=False,
+    )
+
 
 def downgrade() -> None:
     # Remove the new status from the enum
-    op.alter_column('system_requests', 'status', existing_type=new_status_enum, type_=old_status_enum, existing_nullable=False)
+    op.alter_column(
+        "system_requests",
+        "status",
+        existing_type=new_status_enum,
+        type_=old_status_enum,
+        existing_nullable=False,
+    )