tigrbl_engine_numpy 0.1.1.dev7__tar.gz → 0.1.1.dev9__tar.gz

{tigrbl_engine_numpy-0.1.1.dev7 → tigrbl_engine_numpy-0.1.1.dev9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tigrbl_engine_numpy
-Version: 0.1.1.dev7
+Version: 0.1.1.dev9
 Summary: NumPy engine plugin for tigrbl with array-to-table helpers.
 Project-URL: Homepage, https://github.com/swarmauri/swarmauri-sdk
 Author-email: Jacob Stewart <jacob@swarmauri.com>

{tigrbl_engine_numpy-0.1.1.dev7 → tigrbl_engine_numpy-0.1.1.dev9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "tigrbl_engine_numpy"
-version = "0.1.1.dev7"
+version = "0.1.1.dev9"
 description = "NumPy engine plugin for tigrbl with array-to-table helpers."
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -34,4 +34,5 @@ dev = [
     "pytest>=8.0.0",
     "pytest-asyncio>=0.24.0",
     "sqlalchemy>=2.0",
+    "ruff>=0.11.0",
 ]

{tigrbl_engine_numpy-0.1.1.dev7 → tigrbl_engine_numpy-0.1.1.dev9}/src/tigrbl_engine_numpy/__init__.py

@@ -2,11 +2,19 @@ from .engine import NumpyEngine, numpy_capabilities, numpy_engine
 from .session import NumpySession
 
 
+class _Registration:
+    def build(self, *, mapping, spec, dsn):
+        return numpy_engine(mapping=mapping, spec=spec, dsn=dsn)
+
+    def capabilities(self, *, spec, mapping=None):
+        return numpy_capabilities()
+
+
 def register() -> None:
     """Entry-point hook invoked by tigrbl to register this engine."""
     from tigrbl.engine.registry import register_engine
 
-    register_engine("numpy", build=numpy_engine, capabilities=numpy_capabilities)
+    register_engine("numpy", _Registration())
 
 
 __all__ = [

{tigrbl_engine_numpy-0.1.1.dev7 → tigrbl_engine_numpy-0.1.1.dev9}/src/tigrbl_engine_numpy/session.py

@@ -205,6 +205,30 @@ class NumpySession(TigrblSessionBase):
         out = fn(self)
         return await out if hasattr(out, "__await__") else out
 
+    async def begin(self) -> None:
+        await self._tx_begin_impl()
+
+    async def commit(self) -> None:
+        await self._tx_commit_impl()
+
+    async def rollback(self) -> None:
+        await self._tx_rollback_impl()
+
+    async def flush(self) -> None:
+        await self._flush_impl()
+
+    async def refresh(self, obj: Any) -> None:
+        await self._refresh_impl(obj)
+
+    async def get(self, model: type, ident: Any) -> Any | None:
+        return await self._get_impl(model, ident)
+
+    async def execute(self, stmt: Any) -> Any:
+        return await self._execute_impl(stmt)
+
+    async def close(self) -> None:
+        await self._close_impl()
+
     async def _tx_begin_impl(self) -> None:
         self._snap = [dict(row) for row in self._engine.catalog.rows]
         self._snap_ver = self._engine.catalog.table_ver