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

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

@@ -1,12 +1,12 @@
 Metadata-Version: 2.5
 Name: archipy-sqlite
-Version: 5.24.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.24.0
+Requires-Dist: archipy==5.26.0
 Requires-Dist: sqlalchemy[asyncio]>=2.1.6
 Description-Content-Type: text/markdown
 

{archipy_sqlite-5.24.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 (
@@ -73,6 +73,37 @@ def _sqlite_pool_kwargs(configs: SQLiteSQLAlchemyConfig, pool_kwargs: dict[str,
     return {key: value for key, value in pool_kwargs.items() if key not in dropped}
 
 
+def _build_connect_args(configs: SQLiteSQLAlchemyConfig, *, sync_driver: bool) -> dict[str, Any]:
+    """Build the sqlite3 connection arguments from the configuration.
+
+    Args:
+        configs: SQLite configuration.
+        sync_driver: Whether the sync driver is used (``check_same_thread`` has no meaning for aiosqlite,
+            which runs its own connection thread).
+
+    Returns:
+        dict[str, Any]: Only the options that are configured (the sqlite3 defaults apply to the rest).
+    """
+    connect_args: dict[str, Any] = {}
+    if configs.TIMEOUT is not None:
+        connect_args["timeout"] = configs.TIMEOUT
+    if sync_driver and configs.CHECK_SAME_THREAD is not None:
+        connect_args["check_same_thread"] = configs.CHECK_SAME_THREAD
+    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.
 
@@ -96,7 +127,10 @@ class SQLiteSQLAlchemySessionManager(BaseSQLAlchemySessionManager[SQLiteSQLAlche
             orm_config: SQLite-specific configuration. If None, uses global config.
         """
         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]:
@@ -116,6 +150,15 @@ class SQLiteSQLAlchemySessionManager(BaseSQLAlchemySessionManager[SQLiteSQLAlche
         """
         return "sqlite"
 
+    @override
+    def _get_connect_args(self) -> dict[str, Any]:
+        """Return the sqlite3 connection arguments.
+
+        Returns:
+            dict[str, Any]: The configured busy timeout and thread check, otherwise empty.
+        """
+        return _build_connect_args(self._configs, sync_driver=True)
+
     @override
     def _get_pool_kwargs(self, configs: SQLiteSQLAlchemyConfig) -> dict[str, Any]:
         """Return pool arguments valid for the SQLite pool class.
@@ -179,7 +222,10 @@ class AsyncSQLiteSQLAlchemySessionManager(
             orm_config: SQLite-specific configuration. If None, uses global config.
         """
         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]:
@@ -199,6 +245,15 @@ class AsyncSQLiteSQLAlchemySessionManager(
         """
         return "sqlite"
 
+    @override
+    def _get_connect_args(self) -> dict[str, Any]:
+        """Return the sqlite3 connection arguments.
+
+        Returns:
+            dict[str, Any]: The configured busy timeout, otherwise empty.
+        """
+        return _build_connect_args(self._configs, sync_driver=False)
+
     @override
     def _get_pool_kwargs(self, configs: SQLiteSQLAlchemyConfig) -> dict[str, Any]:
         """Return pool arguments valid for the SQLite pool class.

{archipy_sqlite-5.24.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.24.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.24.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",
 ]