New CurrentUserContextManager

kwik/database/__init__.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
 from . import base, mixins, session
+from .current_user_context_manager import CurrentUserContextManager
 from .db_context_manager import DBContextManager
 from .db_context_switcher import DBContextSwitcher

kwik/database/current_user_context_manager.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+from contextlib import contextmanager
+from typing import TYPE_CHECKING, Generator
+
+from kwik.database.context_vars import current_user_ctx_var
+
+if TYPE_CHECKING:
+    import kwik.models
+
+
+@contextmanager
+def CurrentUserContextManager(user: kwik.models.User) -> Generator[None, None, None]:
+    """
+    Context manager for the current user.
+
+    Automatically manages the current user for the CRUD operations.
+    """
+
+    token = current_user_ctx_var.set(user)
+    try:
+        yield
+    finally:
+        current_user_ctx_var.reset(token)

kwik/database/db_context_manager.py

@@ -23,8 +23,8 @@ class DBContextManager:
     def __init__(self) -> None:
         """
         Initialize the DBContextManager.
-
         """
+
         self.db: Session | None = None
         self.token: Token[Session | None] | None = None
 

kwik/utils/enum.py

@@ -0,0 +1,7 @@
+from __future__ import annotations
+
+import enum
+
+
+def merge_str_enum(name: str, *args):
+    return enum.StrEnum(name, [(e, e) for arg in args for e in arg])