event listens for sqlalchemy

src/api/v1/users.py

@@ -29,11 +29,8 @@ async def get_user_by_id(db: sessionDep, id: int, _: CurrentOrAdmin):
 
 @router.patch("/{id}")
 async def patch_user(
-    db: sessionDep,
-    id: int,
-    _: CurrentOrAdmin,
-    user_update: UserUpdate = Body()
-    ):
+    db: sessionDep, id: int, _: CurrentOrAdmin, user_update: UserUpdate = Body()
+):
     updated_user = await UserService(db).update_user(id=id, update_data=user_update)
     return updated_user
 

src/core/database.py

@@ -1,12 +1,24 @@
 from datetime import datetime
 
-from sqlalchemy import TIMESTAMP, func
+from sqlalchemy import TIMESTAMP, func, event
 from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker
 from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
 
 from src.core.settings import settings
 
-engine = create_async_engine(settings.db.url, echo=True)
+engine = create_async_engine(
+    settings.db.url,
+    echo=True
+)
+
+
+@event.listens_for(engine.sync_engine, "connect")
+def set_sqlite_pragma(dbapi_connection, connection_record):
+    if "sqlite" in settings.db.url:
+        cursor = dbapi_connection.cursor()
+        cursor.execute("PRAGMA foreign_keys=ON")
+        cursor.close()
+
 
 async_session_maker = async_sessionmaker(bind=engine, expire_on_commit=False)
 

src/migrations/env.py

@@ -1,8 +1,7 @@
 from logging.config import fileConfig
 
-from sqlalchemy import engine_from_config
+from sqlalchemy import engine_from_config, event
 from sqlalchemy import pool
-
 from alembic import context
 
 from src.core.database import Base
@@ -67,6 +66,14 @@ def run_migrations_online() -> None:
         poolclass=pool.NullPool,
     )
 
+    # Enable foreign keys for SQLite in migrations
+    @event.listens_for(connectable, "connect")
+    def set_sqlite_pragma(dbapi_connection, connection_record):
+        print("⚙️ Enabling PRAGMA foreign_keys=ON for Alembic")
+        cursor = dbapi_connection.cursor()
+        cursor.execute("PRAGMA foreign_keys=ON")
+        cursor.close()
+
     with connectable.connect() as connection:
         context.configure(connection=connection, target_metadata=target_metadata)
 

src/repository/users.py

@@ -1,6 +1,5 @@
 from sqlalchemy import delete, select, update
 
-from schemas.users import User
 from src.models import UsersORM
 from src.repository.base import BaseRepo
 
@@ -18,7 +17,12 @@ class UsersRepo(BaseRepo):
         await self.session.execute(delete(self.model).where(self.model.id == id))
 
     async def update_one(self, id: int, data: dict) -> UsersORM:
-        stmt = update(self.model).where(self.model.id == id).values(data.model_dump(exclude_unset=True)).returning(self.model)
+        stmt = (
+            update(self.model)
+            .where(self.model.id == id)
+            .values(data.model_dump(exclude_unset=True))
+            .returning(self.model)
+        )
         result = await self.session.execute(stmt)
         model = result.scalar_one()
         return model

src/services/users.py

@@ -36,4 +36,3 @@ class UserService(BaseService):
         user = await self.session.user.update_one(id=id, data=update_data)
         await self.session.commit()
         return User.model_validate(user)
-