tigrbl_engine_duckdb 0.1.1.dev2__tar.gz

tigrbl_engine_duckdb-0.1.1.dev2/.gitignore

@@ -0,0 +1,38 @@
+
+.spyproject
+.spyder
+*.pyc
+build
+dist
+*.egg-info
+*.txt
+env
+.env
+.env.*
+/.vs
+.pyirc
+.venv/
+.ipynb_checkpoints/
+.env
+.DS_STORE
+/combined
+.venv_core*
+*.obj
+pytest_results.json
+pkgs/community/swarmauri_vectorstore_annoy/test_annoy.ann
+/pkgs/standards/ptree_dag/pkgs
+*secrets/*
+peagen_artifacts/
+pkgs/standards/peagen/peagen.zip
+/pkgs/uv.lock
+pkgs/standards/peagen/.pymon
+*.pem
+gateway.db
+kms.db
+*.asc
+*.so
+*.db
+target/
+!.gitkeep          # keep the empty dir in repo
+pkgs/experimental/swarmakit/libs/svelte/.vscode/extensions.json
+node_modules/

tigrbl_engine_duckdb-0.1.1.dev2/LICENSE

@@ -0,0 +1,9 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+   ...
+END OF TERMS AND CONDITIONS

tigrbl_engine_duckdb-0.1.1.dev2/PKG-INFO

@@ -0,0 +1,56 @@
+Metadata-Version: 2.4
+Name: tigrbl_engine_duckdb
+Version: 0.1.1.dev2
+Summary: DuckDB engine extension for Tigrbl (optional plugin).
+Project-URL: Homepage, https://github.com/swarmauri/swarmauri-sdk
+Project-URL: Repository, https://github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/experimental/tigrbl_engine_duckdb
+Author-email: Jacob Stewart <jacob@swarmauri.com>
+License:                                  Apache License
+                                   Version 2.0, January 2004
+                                http://www.apache.org/licenses/
+        
+           TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+        
+           1. Definitions.
+           ...
+        END OF TERMS AND CONDITIONS
+License-File: LICENSE
+Keywords: experimental,tigrbl,tigrbl_engine_duckdb
+Classifier: Development Status :: 1 - Planning
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: <3.13,>=3.10
+Requires-Dist: duckdb>=1.0.0
+Requires-Dist: tigrbl>=0.3.0.dev4
+Description-Content-Type: text/markdown
+
+# tigrbl_engine_duckdb
+
+DuckDB engine extension for **Tigrbl**. This package registers the `duckdb`
+engine kind with Tigrbl’s engine registry via entry points.
+
+## Install
+
+```bash
+pip install tigrbl_engine_duckdb
+```
+
+## Use
+
+After installing, you can bind DuckDB using `engine_ctx`:
+
+```python
+from tigrbl.engine.decorators import engine_ctx
+from tigrbl.session.decorators import session_ctx
+
+@engine_ctx({"kind": "duckdb", "path": "./data/app.duckdb",
+             "pragmas": {"memory_limit": "2GB"}})
+@session_ctx({"isolation": "repeatable_read"})
+class AnalyticsAPI:
+    pass
+```
+
+No import of this package is required in your app; Tigrbl auto-loads the
+plugin via entry points on import.

tigrbl_engine_duckdb-0.1.1.dev2/README.md

@@ -0,0 +1,28 @@
+# tigrbl_engine_duckdb
+
+DuckDB engine extension for **Tigrbl**. This package registers the `duckdb`
+engine kind with Tigrbl’s engine registry via entry points.
+
+## Install
+
+```bash
+pip install tigrbl_engine_duckdb
+```
+
+## Use
+
+After installing, you can bind DuckDB using `engine_ctx`:
+
+```python
+from tigrbl.engine.decorators import engine_ctx
+from tigrbl.session.decorators import session_ctx
+
+@engine_ctx({"kind": "duckdb", "path": "./data/app.duckdb",
+             "pragmas": {"memory_limit": "2GB"}})
+@session_ctx({"isolation": "repeatable_read"})
+class AnalyticsAPI:
+    pass
+```
+
+No import of this package is required in your app; Tigrbl auto-loads the
+plugin via entry points on import.

tigrbl_engine_duckdb-0.1.1.dev2/distout/.gitignore

@@ -0,0 +1 @@
+*

tigrbl_engine_duckdb-0.1.1.dev2/pyproject.toml

@@ -0,0 +1,48 @@
+[build-system]
+requires = ["hatchling>=1.18.0"]
+build-backend = "hatchling.build"
+
+[project]
+name = "tigrbl_engine_duckdb"
+version = "0.1.1.dev2"
+description = "DuckDB engine extension for Tigrbl (optional plugin)."
+readme = "README.md"
+license = { file = "LICENSE" }
+authors = [{ name = "Jacob Stewart", email = "jacob@swarmauri.com" }]
+requires-python = ">=3.10,<3.13"
+classifiers = [
+    "Development Status :: 1 - Planning",
+    "License :: OSI Approved :: Apache Software License",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+]
+keywords = ["tigrbl_engine_duckdb", "tigrbl", "experimental"]
+dependencies = [
+  "duckdb>=1.0.0",
+  "tigrbl>=0.3.0.dev4",
+]
+
+[project.urls]
+Homepage = "https://github.com/swarmauri/swarmauri-sdk"
+Repository = "https://github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/experimental/tigrbl_engine_duckdb"
+
+[project.entry-points."tigrbl.engine"]
+duckdb = "tigrbl_engine_duckdb.plugin:register"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/tigrbl_engine_duckdb"]
+
+[dependency-groups]
+dev = [
+  "pytest>=8.0",
+  "pytest-asyncio>=0.24.0",
+  "pytest-xdist>=3.6.1",
+  "pytest-json-report>=1.5.0",
+  "python-dotenv",
+  "requests>=2.32.3",
+  "flake8>=7.0",
+  "pytest-timeout>=2.3.1",
+  "ruff>=0.9.9",
+  "pytest-benchmark>=4.0.0",
+]

tigrbl_engine_duckdb-0.1.1.dev2/src/tigrbl_engine_duckdb/__init__.py

@@ -0,0 +1,12 @@
+from __future__ import annotations
+
+from .duck_builder import duckdb_engine, duckdb_capabilities
+from .duck_session import DuckDBSession
+from .plugin import register
+
+__all__ = [
+    "duckdb_engine",
+    "duckdb_capabilities",
+    "DuckDBSession",
+    "register",
+]

tigrbl_engine_duckdb-0.1.1.dev2/src/tigrbl_engine_duckdb/duck_builder.py

@@ -0,0 +1,56 @@
+from __future__ import annotations
+from typing import Any, Callable, Optional, Tuple, Mapping
+
+import duckdb
+
+from tigrbl.session.spec import SessionSpec
+from .duck_session import DuckDBSession
+
+SessionFactory = Callable[[], Any]
+
+
+def duckdb_engine(
+    *,
+    path: Optional[str] = None,  # None or ":memory:" → in-memory
+    read_only: bool = False,
+    threads: Optional[int] = None,
+    pragmas: Optional[Mapping[str, Any]] = None,
+    mapping: Optional[Mapping[str, Any]] = None,  # accepted but not required
+    spec: Optional[Any] = None,  # accepted for signature parity
+    dsn: Optional[str] = None,  # accepted for signature parity
+) -> Tuple[Any, SessionFactory]:
+    """
+    Build a DuckDB 'engine' and a session factory that yields DuckDBSession.
+    No SQLAlchemy; pure duckdb bindings.
+    Returns:
+        (engine_handle, sessionmaker)
+    """
+    db_path = path or ":memory:"
+
+    def mk_session(spec_in: Optional[SessionSpec] = None) -> DuckDBSession:
+        conn = duckdb.connect(db_path, read_only=read_only)
+        # Pragmas per-session for determinism.
+        if threads is not None:
+            conn.execute(f"PRAGMA threads={int(threads)}")
+        if pragmas:
+            for k, v in pragmas.items():
+                if isinstance(v, bool):
+                    v = "true" if v else "false"
+                conn.execute(f"PRAGMA {k}={v}")
+        return DuckDBSession(conn, spec_in)
+
+    engine_handle = {"kind": "duckdb", "path": db_path, "read_only": read_only}
+    return engine_handle, mk_session
+
+
+def duckdb_capabilities() -> dict[str, Any]:
+    """
+    Capability advertisement for session_ctx validation.
+    DuckDB provides MVCC snapshot-style isolation.
+    """
+    return {
+        "transactional": True,
+        "isolation_levels": {"snapshot", "repeatable_read"},
+        "read_only_enforced": False,
+        "async_native": False,
+    }

tigrbl_engine_duckdb-0.1.1.dev2/src/tigrbl_engine_duckdb/duck_session.py

@@ -0,0 +1,121 @@
+from __future__ import annotations
+
+import asyncio
+from typing import Any, Callable, Optional, Sequence
+
+import duckdb
+
+from tigrbl.session.base import TigrblSessionBase
+from tigrbl.session.spec import SessionSpec
+from tigrbl.core.crud.helpers.model import _single_pk_name, _model_columns
+from tigrbl.core.crud.helpers import NoResultFound
+
+
+class _ScalarResult:
+    """Minimal result facade for Tigrbl core CRUD."""
+
+    def __init__(self, items: Sequence[Any]) -> None:
+        self._items = list(items)
+
+    def scalars(self) -> "_ScalarResult":
+        return self
+
+    def all(self) -> list[Any]:
+        return list(self._items)
+
+    def scalar_one(self) -> Any:
+        if len(self._items) != 1:
+            raise NoResultFound("expected exactly one row")
+        return self._items[0]
+
+
+class DuckDBSession(TigrblSessionBase):
+    """Transactional session over a synchronous duckdb.Connection."""
+
+    def __init__(
+        self, conn: duckdb.DuckDBPyConnection, spec: Optional[SessionSpec] = None
+    ) -> None:
+        super().__init__(spec)
+        self._c = conn
+
+    # ---------- async marker ----------
+    async def run_sync(self, fn: Callable[[Any], Any]) -> Any:
+        return await asyncio.to_thread(fn, self._c)
+
+    # ---------- TX primitives ----------
+    async def _tx_begin_impl(self) -> None:
+        await asyncio.to_thread(self._c.execute, "BEGIN")
+
+    async def _tx_commit_impl(self) -> None:
+        await asyncio.to_thread(self._c.execute, "COMMIT")
+
+    async def _tx_rollback_impl(self) -> None:
+        await asyncio.to_thread(self._c.execute, "ROLLBACK")
+
+    # ---------- CRUD primitives ----------
+    def _add_impl(self, obj: Any) -> Any:
+        table = getattr(obj.__class__, "__tablename__", obj.__class__.__name__)
+        cols = list(_model_columns(obj.__class__).keys())
+        placeholders = ", ".join(["?"] * len(cols))
+        col_list = ", ".join(cols)
+        sql = f'INSERT INTO "{table}" ({col_list}) VALUES ({placeholders})'
+        values = [getattr(obj, c, None) for c in cols]
+
+        def _exec():
+            self._c.execute(sql, values)
+
+        return _exec()
+
+    async def _delete_impl(self, obj: Any) -> None:
+        table = getattr(obj.__class__, "__tablename__", obj.__class__.__name__)
+        pk = _single_pk_name(obj.__class__)
+        ident = getattr(obj, pk)
+        await asyncio.to_thread(
+            self._c.execute, f'DELETE FROM "{table}" WHERE "{pk}" = ?', [ident]
+        )
+
+    async def _flush_impl(self) -> None:
+        return
+
+    async def _refresh_impl(self, obj: Any) -> None:
+        pk = _single_pk_name(obj.__class__)
+        ident = getattr(obj, pk)
+        fresh = await self._get_impl(obj.__class__, ident)
+        if fresh:
+            for c in _model_columns(obj.__class__).keys():
+                setattr(obj, c, getattr(fresh, c, None))
+
+    async def _get_impl(self, model: type, ident: Any) -> Any | None:
+        table = getattr(model, "__tablename__", model.__name__)
+        pk = _single_pk_name(model)
+        cols = list(_model_columns(model).keys())
+        col_list = ", ".join([f'"{c}"' for c in cols])
+        sql = f'SELECT {col_list} FROM "{table}" WHERE "{pk}" = ?'
+        cur = await asyncio.to_thread(self._c.execute, sql, [ident])
+        row = cur.fetchone()
+        if not row:
+            return None
+        obj = model()
+        for i, c in enumerate(cols):
+            setattr(obj, c, row[i])
+        return obj
+
+    async def _execute_impl(self, stmt: Any) -> Any:
+        if isinstance(stmt, tuple) and len(stmt) == 2 and isinstance(stmt[0], str):
+            sql, params = stmt
+            cur = await asyncio.to_thread(self._c.execute, sql, params or [])
+            rows = cur.fetchall() or []
+            return _ScalarResult(rows)
+
+        if isinstance(stmt, str):
+            cur = await asyncio.to_thread(self._c.execute, stmt)
+            rows = cur.fetchall() or []
+            return _ScalarResult(rows)
+
+        raise NotImplementedError(f"Unsupported statement type: {type(stmt)}")
+
+    async def _close_impl(self) -> None:
+        try:
+            await asyncio.to_thread(self._c.close)
+        except Exception:
+            pass

tigrbl_engine_duckdb-0.1.1.dev2/src/tigrbl_engine_duckdb/plugin.py

@@ -0,0 +1,10 @@
+from __future__ import annotations
+
+from tigrbl.engine.registry import register_engine
+from .duck_builder import duckdb_engine, duckdb_capabilities
+
+
+def register() -> None:
+    # Entry point hook, called by Tigrbl's plugin loader.
+    # Registers the 'duckdb' engine kind.
+    register_engine("duckdb", duckdb_engine, duckdb_capabilities)

tigrbl_engine_duckdb-0.1.1.dev2/tests/test_smoke.py

@@ -0,0 +1,8 @@
+from pathlib import Path
+
+
+def test_package_assets_present() -> None:
+    package_dir = Path(__file__).resolve().parents[1]
+    assert (package_dir / "README.md").is_file()
+    assert (package_dir / "LICENSE").is_file()
+    assert (package_dir / "pyproject.toml").is_file()