archipy-sqlite 5.25.0__tar.gz → 5.26.0__tar.gz

{archipy_sqlite-5.25.0 → archipy_sqlite-5.26.0}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.5
 Name: archipy-sqlite
-Version: 5.25.0
+Version: 5.26.0
 Summary: ArchiPy sqlite adapter.
 Author-email: Hossein Nejati <hosseinnejati14@gmail.com>
 License-Expression: Apache-2.0
 Requires-Python: <4,>=3.14
 Requires-Dist: aiosqlite>=0.22.1
-Requires-Dist: archipy==5.25.0
+Requires-Dist: archipy==5.26.0
 Requires-Dist: sqlalchemy[asyncio]>=2.1.6
 Description-Content-Type: text/markdown
 

{archipy_sqlite-5.25.0 → archipy_sqlite-5.26.0}/archipy/adapters/sqlite/sqlalchemy/session_managers.py

@@ -2,7 +2,7 @@
 
 from typing import Any, override
 
-from sqlalchemy import URL
+from sqlalchemy import URL, Engine, event
 from sqlalchemy.exc import SQLAlchemyError
 
 from archipy.adapters.base.sqlalchemy.session_managers import (
@@ -92,6 +92,18 @@ def _build_connect_args(configs: SQLiteSQLAlchemyConfig, *, sync_driver: bool) -
     return connect_args
 
 
+def _enable_foreign_keys(engine: Engine) -> None:
+    """Turn on foreign key enforcement for every connection the engine opens (SQLite has it off by default)."""
+
+    @event.listens_for(engine, "connect")
+    def set_foreign_keys_pragma(dbapi_connection: Any, _connection_record: object) -> None:  # noqa: ANN401
+        cursor = dbapi_connection.cursor()
+        try:
+            cursor.execute("PRAGMA foreign_keys=ON")
+        finally:
+            cursor.close()
+
+
 class SQLiteSQLAlchemySessionManager(BaseSQLAlchemySessionManager[SQLiteSQLAlchemyConfig], metaclass=Singleton):
     """Synchronous SQLAlchemy session manager for SQLite.
 
@@ -117,6 +129,8 @@ class SQLiteSQLAlchemySessionManager(BaseSQLAlchemySessionManager[SQLiteSQLAlche
         configs = orm_config or self._default_config()
         self._configs = configs  # read by _get_connect_args, which the base initializer calls
         super().__init__(configs)
+        if configs.FOREIGN_KEYS:
+            _enable_foreign_keys(self.engine)
 
     @override
     def _expected_config_type(self) -> type[SQLiteSQLAlchemyConfig]:
@@ -210,6 +224,8 @@ class AsyncSQLiteSQLAlchemySessionManager(
         configs = orm_config or self._default_config()
         self._configs = configs  # read by _get_connect_args, which the base initializer calls
         super().__init__(configs)
+        if configs.FOREIGN_KEYS:
+            _enable_foreign_keys(self.engine.sync_engine)
 
     @override
     def _expected_config_type(self) -> type[SQLiteSQLAlchemyConfig]:

{archipy_sqlite-5.25.0 → archipy_sqlite-5.26.0}/pyproject.toml

@@ -5,10 +5,10 @@ authors = [
     { name = "Hossein Nejati", email = "hosseinnejati14@gmail.com" },
 ]
 readme = "README.md"
-version = "5.25.0" # This version is set on publish flow!
+version = "5.26.0" # This version is set on publish flow!
 requires-python = ">=3.14,<4"
 dependencies = [
-    "archipy==5.25.0", # Pinned to core version on publish flow!
+    "archipy==5.26.0", # Pinned to core version on publish flow!
     "sqlalchemy[asyncio]>=2.1.6",
     "aiosqlite>=0.22.1",
 ]