Format Python migrations code

Author: lulunac27a

migrations/env.py

@@ -12,32 +12,31 @@
 # Interpret the config file for Python logging.
 # This line sets up loggers basically.
 fileConfig(config.config_file_name)
-logger = logging.getLogger('alembic.env')
+logger = logging.getLogger("alembic.env")
 
 
 def get_engine():
     try:
         # this works with Flask-SQLAlchemy<3 and Alchemical
-        return current_app.extensions['migrate'].db.get_engine()
+        return current_app.extensions["migrate"].db.get_engine()
     except (TypeError, AttributeError):
         # this works with Flask-SQLAlchemy>=3
-        return current_app.extensions['migrate'].db.engine
+        return current_app.extensions["migrate"].db.engine
 
 
 def get_engine_url():
     try:
-        return get_engine().url.render_as_string(hide_password=False).replace(
-            '%', '%%')
+        return get_engine().url.render_as_string(hide_password=False).replace("%", "%%")
     except AttributeError:
-        return str(get_engine().url).replace('%', '%%')
+        return str(get_engine().url).replace("%", "%%")
 
 
 # add your model's MetaData object here
 # for 'autogenerate' support
 # from myapp import mymodel
 # target_metadata = mymodel.Base.metadata
-config.set_main_option('sqlalchemy.url', get_engine_url())
-target_db = current_app.extensions['migrate'].db
+config.set_main_option("sqlalchemy.url", get_engine_url())
+target_db = current_app.extensions["migrate"].db
 
 # other values from the config, defined by the needs of env.py,
 # can be acquired:
@@ -46,7 +45,7 @@ def get_engine_url():
 
 
 def get_metadata():
-    if hasattr(target_db, 'metadatas'):
+    if hasattr(target_db, "metadatas"):
         return target_db.metadatas[None]
     return target_db.metadata
 
@@ -64,9 +63,8 @@ def run_migrations_offline():
 
     """
     url = config.get_main_option("sqlalchemy.url")
-    context.configure(
-        url=url, target_metadata=get_metadata(), literal_binds=True
-    )
+    context.configure(url=url, target_metadata=get_metadata(),
+                      literal_binds=True)
 
     with context.begin_transaction():
         context.run_migrations()
@@ -84,23 +82,21 @@ def run_migrations_online():
     # when there are no changes to the schema
     # reference: http://alembic.zzzcomputing.com/en/latest/cookbook.html
     def process_revision_directives(context, revision, directives):
-        if getattr(config.cmd_opts, 'autogenerate', False):
+        if getattr(config.cmd_opts, "autogenerate", False):
             script = directives[0]
             if script.upgrade_ops.is_empty():
                 directives[:] = []
-                logger.info('No changes in schema detected.')
+                logger.info("No changes in schema detected.")
 
-    conf_args = current_app.extensions['migrate'].configure_args
+    conf_args = current_app.extensions["migrate"].configure_args
     if conf_args.get("process_revision_directives") is None:
         conf_args["process_revision_directives"] = process_revision_directives
 
     connectable = get_engine()
 
     with connectable.connect() as connection:
         context.configure(
-            connection=connection,
-            target_metadata=get_metadata(),
-            **conf_args
+            connection=connection, target_metadata=get_metadata(), **conf_args
         )
 
         with context.begin_transaction():

migrations/versions/5da7b534611d_add_number_of_tasks_completed_in_a_day_.py

@@ -5,28 +5,31 @@
 Create Date: 2025-01-04 18:52:22.404495
 
 """
+
 from alembic import op
 import sqlalchemy as sa
 
 
 # revision identifiers, used by Alembic.
-revision = '5da7b534611d'
-down_revision = 'c0fcd66c3df6'
+revision = "5da7b534611d"
+down_revision = "c0fcd66c3df6"
 branch_labels = None
 depends_on = None
 
 
 def upgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('user', schema=None) as batch_op:
-        batch_op.add_column(sa.Column('daily_tasks_completed', sa.Integer(), nullable=False))
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column("daily_tasks_completed", sa.Integer(), nullable=False)
+        )
 
     # ### 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.drop_column('daily_tasks_completed')
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.drop_column("daily_tasks_completed")
 
     # ### end Alembic commands ###

migrations/versions/5e8982aea1ef_add_number_of_tasks_completed_column.py

@@ -5,28 +5,30 @@
 Create Date: 2025-01-04 18:50:49.963825
 
 """
+
 from alembic import op
 import sqlalchemy as sa
 
 
 # revision identifiers, used by Alembic.
-revision = '5e8982aea1ef'
-down_revision = '6b7325da9331'
+revision = "5e8982aea1ef"
+down_revision = "6b7325da9331"
 branch_labels = None
 depends_on = None
 
 
 def upgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('user', schema=None) as batch_op:
-        batch_op.add_column(sa.Column('tasks_completed', sa.Integer(), nullable=False))
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.add_column(sa.Column("tasks_completed",
+                            sa.Integer(), nullable=False))
 
     # ### 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.drop_column('tasks_completed')
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.drop_column("tasks_completed")
 
     # ### end Alembic commands ###

migrations/versions/6b7325da9331_add_task_completed_column.py

@@ -5,28 +5,30 @@
 Create Date: 2025-01-04 18:49:58.913626
 
 """
+
 from alembic import op
 import sqlalchemy as sa
 
 
 # revision identifiers, used by Alembic.
-revision = '6b7325da9331'
-down_revision = '8f05b938c89f'
+revision = "6b7325da9331"
+down_revision = "8f05b938c89f"
 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.add_column(sa.Column('completed', sa.Boolean(), nullable=False))
+    with op.batch_alter_table("task", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column("completed", sa.Boolean(), nullable=False))
 
     # ### end Alembic commands ###
 
 
 def downgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('task', schema=None) as batch_op:
-        batch_op.drop_column('completed')
+    with op.batch_alter_table("task", schema=None) as batch_op:
+        batch_op.drop_column("completed")
 
     # ### end Alembic commands ###

migrations/versions/868a910249be_create_user_and_task_tables.py

@@ -5,49 +5,55 @@
 Create Date: 2025-01-04 18:44:02.793639
 
 """
+
 from alembic import op
 import sqlalchemy as sa
 
 
 # revision identifiers, used by Alembic.
-revision = '868a910249be'
+revision = "868a910249be"
 down_revision = None
 branch_labels = None
 depends_on = None
 
 
 def upgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table('user',
-    sa.Column('id', sa.Integer(), nullable=False),
-    sa.Column('username', sa.String(length=80), nullable=False),
-    sa.Column('xp', sa.Float(), nullable=False),
-    sa.Column('xp_required', sa.Float(), nullable=False),
-    sa.Column('total_xp', sa.Float(), nullable=False),
-    sa.Column('level', sa.Integer(), nullable=False),
-    sa.PrimaryKeyConstraint('id'),
-    sa.UniqueConstraint('id'),
-    sa.UniqueConstraint('username')
+    op.create_table(
+        "user",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("username", sa.String(length=80), nullable=False),
+        sa.Column("xp", sa.Float(), nullable=False),
+        sa.Column("xp_required", sa.Float(), nullable=False),
+        sa.Column("total_xp", sa.Float(), nullable=False),
+        sa.Column("level", sa.Integer(), nullable=False),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("id"),
+        sa.UniqueConstraint("username"),
     )
-    op.create_table('task',
-    sa.Column('id', sa.Integer(), nullable=False),
-    sa.Column('name', sa.String(length=80), nullable=False),
-    sa.Column('original_due_date', sa.Date(), nullable=False),
-    sa.Column('due_date', sa.Date(), nullable=False),
-    sa.Column('priority', sa.Integer(), nullable=False),
-    sa.Column('difficulty', sa.Integer(), nullable=False),
-    sa.Column('repeat_interval', sa.Integer(), nullable=False),
-    sa.Column('repeat_often', sa.Integer(), nullable=False),
-    sa.Column('user_id', sa.Integer(), nullable=True),
-    sa.ForeignKeyConstraint(['user_id'], ['user.id'], ),
-    sa.PrimaryKeyConstraint('id'),
-    sa.UniqueConstraint('id')
+    op.create_table(
+        "task",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("name", sa.String(length=80), nullable=False),
+        sa.Column("original_due_date", sa.Date(), nullable=False),
+        sa.Column("due_date", sa.Date(), nullable=False),
+        sa.Column("priority", sa.Integer(), nullable=False),
+        sa.Column("difficulty", sa.Integer(), nullable=False),
+        sa.Column("repeat_interval", sa.Integer(), nullable=False),
+        sa.Column("repeat_often", sa.Integer(), nullable=False),
+        sa.Column("user_id", sa.Integer(), nullable=True),
+        sa.ForeignKeyConstraint(
+            ["user_id"],
+            ["user.id"],
+        ),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("id"),
     )
     # ### end Alembic commands ###
 
 
 def downgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_table('task')
-    op.drop_table('user')
+    op.drop_table("task")
+    op.drop_table("user")
     # ### end Alembic commands ###

migrations/versions/8f05b938c89f_add_times_completed_column.py

@@ -5,28 +5,30 @@
 Create Date: 2025-01-04 18:49:08.275615
 
 """
+
 from alembic import op
 import sqlalchemy as sa
 
 
 # revision identifiers, used by Alembic.
-revision = '8f05b938c89f'
-down_revision = 'df154647d001'
+revision = "8f05b938c89f"
+down_revision = "df154647d001"
 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.add_column(sa.Column('times_completed', sa.Integer(), nullable=False))
+    with op.batch_alter_table("task", schema=None) as batch_op:
+        batch_op.add_column(sa.Column("times_completed",
+                            sa.Integer(), nullable=False))
 
     # ### end Alembic commands ###
 
 
 def downgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('task', schema=None) as batch_op:
-        batch_op.drop_column('times_completed')
+    with op.batch_alter_table("task", schema=None) as batch_op:
+        batch_op.drop_column("times_completed")
 
     # ### end Alembic commands ###

migrations/versions/c0fcd66c3df6_add_daily_streak_column.py

@@ -5,30 +5,34 @@
 Create Date: 2025-01-04 18:51:36.383519
 
 """
+
 from alembic import op
 import sqlalchemy as sa
 
 
 # revision identifiers, used by Alembic.
-revision = 'c0fcd66c3df6'
-down_revision = '5e8982aea1ef'
+revision = "c0fcd66c3df6"
+down_revision = "5e8982aea1ef"
 branch_labels = None
 depends_on = None
 
 
 def upgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('user', schema=None) as batch_op:
-        batch_op.add_column(sa.Column('last_completion_date', sa.Date(), nullable=False))
-        batch_op.add_column(sa.Column('daily_streak', sa.Integer(), nullable=False))
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column("last_completion_date", sa.Date(), nullable=False)
+        )
+        batch_op.add_column(
+            sa.Column("daily_streak", sa.Integer(), nullable=False))
 
     # ### 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.drop_column('daily_streak')
-        batch_op.drop_column('last_completion_date')
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.drop_column("daily_streak")
+        batch_op.drop_column("last_completion_date")
 
     # ### end Alembic commands ###

migrations/versions/df154647d001_add_streak_column.py

@@ -5,28 +5,29 @@
 Create Date: 2025-01-04 18:48:09.868863
 
 """
+
 from alembic import op
 import sqlalchemy as sa
 
 
 # revision identifiers, used by Alembic.
-revision = 'df154647d001'
-down_revision = '868a910249be'
+revision = "df154647d001"
+down_revision = "868a910249be"
 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.add_column(sa.Column('streak', sa.Integer(), nullable=False))
+    with op.batch_alter_table("task", schema=None) as batch_op:
+        batch_op.add_column(sa.Column("streak", sa.Integer(), nullable=False))
 
     # ### end Alembic commands ###
 
 
 def downgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('task', schema=None) as batch_op:
-        batch_op.drop_column('streak')
+    with op.batch_alter_table("task", schema=None) as batch_op:
+        batch_op.drop_column("streak")
 
     # ### end Alembic commands ###