Async Types

sanctumlabs_dbkit/sql/callbacks.py

@@ -4,8 +4,8 @@
 
 from typing import List, cast
 
-from sanctumlabs_dbkit.sql.session import Session
-from sanctumlabs_dbkit.sql.types import CommitCallback
+from sanctumlabs_dbkit.sql.session import Session, AsyncSession
+from sanctumlabs_dbkit.sql.types import CommitCallback, CommitCallbackAsync
 
 
 def on_commit(current_session: Session, callback: CommitCallback) -> None:
@@ -14,3 +14,14 @@ def on_commit(current_session: Session, callback: CommitCallback) -> None:
         List[CommitCallback], current_session.info.setdefault("on_commit_hooks", [])
     )
     commit_hooks.append(callback)
+
+
+def on_commit_async(
+    current_session: AsyncSession, callback: CommitCallbackAsync
+) -> None:
+    """Sets an async commit callback to the current session"""
+    commit_hooks = cast(
+        List[CommitCallbackAsync],
+        current_session.info.setdefault("on_commit_hooks", []),
+    )
+    commit_hooks.append(callback)

sanctumlabs_dbkit/sql/repository/__init__.py

@@ -0,0 +1,25 @@
+"""Repository package exposing sync and async repository implementations."""
+
+from sanctumlabs_dbkit.sql.repository.repository import (
+    Repository,
+    BaseRepository,
+    WriteRepository,
+    ReadRepository,
+)
+from sanctumlabs_dbkit.sql.repository.async_repository import (
+    AsyncRepository,
+    AsyncBaseRepository,
+    AsyncWriteRepository,
+    AsyncReadRepository,
+)
+
+__all__ = [
+    "Repository",
+    "BaseRepository",
+    "WriteRepository",
+    "ReadRepository",
+    "AsyncRepository",
+    "AsyncBaseRepository",
+    "AsyncWriteRepository",
+    "AsyncReadRepository",
+]