Format migration code

Author: lulunac27a

migrations/versions/28e53a4e60fe_add_combo_multipliers.py

@@ -5,60 +5,81 @@
 Create Date: 2025-01-14 22:48:14.773004
 
 """
+
 from alembic import op
 import sqlalchemy as sa
 
 
 # revision identifiers, used by Alembic.
-revision = '28e53a4e60fe'
-down_revision = 'fa5ff0ea7eab'
+revision = "28e53a4e60fe"
+down_revision = "fa5ff0ea7eab"
 branch_labels = None
 depends_on = None
 
 
 def upgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('task', schema=None) as batch_op:
-        batch_op.alter_column('streak',
-               existing_type=sa.INTEGER(),
-               nullable=False,
-               existing_server_default=sa.text('0'))
-        batch_op.alter_column('completed',
-               existing_type=sa.INTEGER(),
-               type_=sa.Boolean(),
-               existing_nullable=False,
-               existing_server_default=sa.text('0'))
+    with op.batch_alter_table("task", schema=None) as batch_op:
+        batch_op.alter_column(
+            "streak",
+            existing_type=sa.INTEGER(),
+            nullable=False,
+            existing_server_default=sa.text("0"),
+        )
+        batch_op.alter_column(
+            "completed",
+            existing_type=sa.INTEGER(),
+            type_=sa.Boolean(),
+            existing_nullable=False,
+            existing_server_default=sa.text("0"),
+        )
 
-    with op.batch_alter_table('user', schema=None) as batch_op:
-        batch_op.add_column(sa.Column('combo_multiplier', sa.Integer(), nullable=False, server_default='0'))
-        batch_op.add_column(sa.Column('last_task_completed', sa.Integer(), nullable=False, server_default='-1'))
-        batch_op.alter_column('tasks_completed',
-               existing_type=sa.INTEGER(),
-               nullable=False,
-               existing_server_default=sa.text('0'))
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "combo_multiplier", sa.Integer(), nullable=False, server_default="0"
+            )
+        )
+        batch_op.add_column(
+            sa.Column(
+                "last_task_completed", sa.Integer(), nullable=False, server_default="-1"
+            )
+        )
+        batch_op.alter_column(
+            "tasks_completed",
+            existing_type=sa.INTEGER(),
+            nullable=False,
+            existing_server_default=sa.text("0"),
+        )
 
     # ### end Alembic commands ###
 
 
 def downgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('user', schema=None) as batch_op:
-        batch_op.alter_column('tasks_completed',
-               existing_type=sa.INTEGER(),
-               nullable=True,
-               existing_server_default=sa.text('0'))
-        batch_op.drop_column('last_task_completed')
-        batch_op.drop_column('combo_multiplier')
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.alter_column(
+            "tasks_completed",
+            existing_type=sa.INTEGER(),
+            nullable=True,
+            existing_server_default=sa.text("0"),
+        )
+        batch_op.drop_column("last_task_completed")
+        batch_op.drop_column("combo_multiplier")
 
-    with op.batch_alter_table('task', schema=None) as batch_op:
-        batch_op.alter_column('completed',
-               existing_type=sa.Boolean(),
-               type_=sa.INTEGER(),
-               existing_nullable=False,
-               existing_server_default=sa.text('0'))
-        batch_op.alter_column('streak',
-               existing_type=sa.INTEGER(),
-               nullable=True,
-               existing_server_default=sa.text('0'))
+    with op.batch_alter_table("task", schema=None) as batch_op:
+        batch_op.alter_column(
+            "completed",
+            existing_type=sa.Boolean(),
+            type_=sa.INTEGER(),
+            existing_nullable=False,
+            existing_server_default=sa.text("0"),
+        )
+        batch_op.alter_column(
+            "streak",
+            existing_type=sa.INTEGER(),
+            nullable=True,
+            existing_server_default=sa.text("0"),
+        )
 
     # ### end Alembic commands ###