Make ruff happy

Author: febus982

src/alembic/env.py

@@ -3,13 +3,13 @@
 import logging
 import sys
 from asyncio import get_event_loop
-from datetime import datetime, timezone
+from datetime import datetime
 from os import listdir, path
 from os.path import isfile, join
 
-from sqlalchemy import Table, Column, String, DateTime, UniqueConstraint, text, select
+from sqlalchemy import DateTime, String
 from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, async_sessionmaker
-from sqlalchemy.orm import registry, sessionmaker, Mapped, mapped_column
+from sqlalchemy.orm import Mapped, mapped_column
 
 from alembic import context
 from common.bootstrap import application_init
@@ -40,16 +40,21 @@
 db_names = target_metadata.keys()
 config.set_main_option("databases", ",".join(db_names))
 
+
 def generate_fixture_migration_model(declarative_base: type):
     class FixtureMigration(declarative_base):
         __tablename__ = "alembic_fixtures"
 
         bind: Mapped[str] = mapped_column(String(), primary_key=True)
         filename: Mapped[str] = mapped_column(String(), primary_key=True)
         signature: Mapped[str] = mapped_column(String(), nullable=False)
-        processed_at: Mapped[datetime] = mapped_column(DateTime(), nullable=False, default=datetime.now)
+        processed_at: Mapped[datetime] = mapped_column(
+            DateTime(), nullable=False, default=datetime.now
+        )
+
     return FixtureMigration
 
+
 fixture_migration_models = {}
 for name in db_names:
     fixture_migration_models[name] = generate_fixture_migration_model(
@@ -75,55 +80,64 @@ class FixtureMigration(declarative_base):
 # my_important_option = config.get_main_option("my_important_option")
 # ... etc.
 
-def calculate_md5(file_path):
-    hasher = hashlib.md5()
-    with open(file_path, 'rb') as file:
+
+def calculate_signature(file_path):
+    hasher = hashlib.sha256()
+    with open(file_path, "rb") as file:
         hasher.update(file.read())
     return hasher.hexdigest()
 
-async def a_migrate_fixtures(bind_name: str, Session: async_sessionmaker[AsyncSession]) -> str:
+
+async def a_migrate_fixtures(
+    bind_name: str, session: async_sessionmaker[AsyncSession]
+) -> str:
     alembic_path = path.dirname(path.realpath(__file__))
     fixtures_path = alembic_path + "/fixtures"
     sys.path.append(alembic_path)
     module_names = [
-        f[:-3] for f in listdir(fixtures_path)
-        if isfile(join(fixtures_path, f))
-        and f.endswith(".py")
-        and f != "__init__.py"
+        f[:-3]
+        for f in listdir(fixtures_path)
+        if isfile(join(fixtures_path, f)) and f.endswith(".py") and f != "__init__.py"
     ]
-    async with Session() as session:
+    async with session() as session:
         for module_name in module_names:
             logging.debug(f"Creating {module_name} fixtures for {bind_name}")
             m = importlib.import_module(f"fixtures.{module_name}")
             fixture_migration = await session.get(
-                fixture_migration_models[bind_name],
-                (bind_name, f"{module_name}.py")
+                fixture_migration_models[bind_name], (bind_name, f"{module_name}.py")
             )
-            signature = calculate_md5(f"{fixtures_path}/{module_name}.py")
+            signature = calculate_signature(f"{fixtures_path}/{module_name}.py")
             if fixture_migration:
                 if signature != fixture_migration.signature:
-                    logging.warning(f"Signature mismatch for {fixture_migration.filename} fixture."
-                                    f" The file has been already processed but has been modified since then."
-                                    f" It will not be processed again.")
+                    logging.warning(
+                        f"Signature mismatch for {fixture_migration.filename} fixture."
+                        f" The file has been already processed but has been modified"
+                        f" since then. It will not be processed again."
+                    )
                 else:
-                    logging.debug(f"{module_name} fixtures already processed for {bind_name}")
+                    logging.debug(
+                        f"{module_name} fixtures already processed for {bind_name}"
+                    )
                 continue
 
             session.add_all(m.fixtures().get(bind_name, []))
-            session.add(fixture_migration_models[bind_name](
-                bind=bind_name,
-                filename=f"{module_name}.py",
-                signature=signature,
-            ))
+            session.add(
+                fixture_migration_models[bind_name](
+                    bind=bind_name,
+                    filename=f"{module_name}.py",
+                    signature=signature,
+                )
+            )
             try:
                 await session.commit()
-                logging.info(f"{module_name} fixtures correctly created for {bind_name}")
-            except:
+                logging.info(
+                    f"{module_name} fixtures correctly created for {bind_name}"
+                )
+            except Exception:
                 await session.rollback()
                 logging.error(f"{module_name} fixtures failed to apply to {bind_name}")
 
 
-
 def run_migrations_offline() -> None:
     """Run migrations in 'offline' mode.
 
@@ -206,12 +220,14 @@ async def run_migrations_online() -> None:
         for name, rec in engines.items():
             logger.info(f"Migrating database {name}")
             if isinstance(rec["engine"], AsyncEngine):
+
                 def migration_callable(*args, **kwargs):
                     return do_run_migration(*args, name=name, **kwargs)
 
                 await rec["connection"].run_sync(migration_callable)
-                Session = async_sessionmaker(bind=rec["connection"])
-                await a_migrate_fixtures(bind_name=name, Session=Session)
+                await a_migrate_fixtures(
+                    bind_name=name, session=async_sessionmaker(bind=rec["connection"])
+                )
 
             else:
                 do_run_migration(rec["connection"], name)

src/alembic/fixtures/books_example.py

@@ -3,6 +3,7 @@
 
 "BIND_NAME": "LIST_OF_FACTORIES"
 """
+
 from typing import Dict, List
 
 from factory import Factory

src/alembic/versions/2025-01-26-212326-52b1246eda46_initialize_fixture_tables.py

@@ -26,14 +26,15 @@ def downgrade(engine_name: str) -> None:
 
 
 def upgrade_default() -> None:
-    op.create_table('alembic_fixtures',
-        sa.Column('bind', sa.String(), nullable=False),
-        sa.Column('filename', sa.String(), nullable=False),
-        sa.Column('signature', sa.String(), nullable=False),
-        sa.Column('processed_at', sa.DateTime(timezone=True), nullable=False),
-        sa.PrimaryKeyConstraint('bind', 'filename'),
+    op.create_table(
+        "alembic_fixtures",
+        sa.Column("bind", sa.String(), nullable=False),
+        sa.Column("filename", sa.String(), nullable=False),
+        sa.Column("signature", sa.String(), nullable=False),
+        sa.Column("processed_at", sa.DateTime(timezone=True), nullable=False),
+        sa.PrimaryKeyConstraint("bind", "filename"),
     )
 
 
 def downgrade_default() -> None:
-    op.drop_table('alembic_fixtures')
+    op.drop_table("alembic_fixtures")

src/alembic/versions/2025-01-26-212826-bd73bd8a2ac4_create_books_table.py

@@ -5,13 +5,14 @@
 Create Date: 2025-01-26 21:28:26.321986
 
 """
-from alembic import op
+
 import sqlalchemy as sa
 
+from alembic import op
 
 # revision identifiers, used by Alembic.
-revision = 'bd73bd8a2ac4'
-down_revision = '52b1246eda46'
+revision = "bd73bd8a2ac4"
+down_revision = "52b1246eda46"
 branch_labels = None
 depends_on = None