beaver-db 0.19.2__py3-none-any.whl → 0.19.3__py3-none-any.whl

beaver/__init__.py

@@ -2,4 +2,4 @@ from .core import BeaverDB
 from .types import Model
 from .collections import Document, WalkDirection
 
-__version__ = "0.19.2"
+__version__ = "0.19.3"

beaver/blobs.py

@@ -1,8 +1,8 @@
 import json
 import sqlite3
 from typing import Any, Dict, Iterator, NamedTuple, Optional, Type, TypeVar
-
 from .types import JsonSerializable, IDatabase
+from .locks import LockManager
 
 
 class Blob[M](NamedTuple):
@@ -16,10 +16,13 @@ class Blob[M](NamedTuple):
 class BlobManager[M]:
     """A wrapper providing a Pythonic interface to a blob store in the database."""
 
+# In beaver/blobs.py, inside class BlobManager[M]:
     def __init__(self, name: str, db: IDatabase, model: Type[M] | None = None):
         self._name = name
         self._db = db
         self._model = model
+        lock_name = f"__lock__blob__{name}"
+        self._lock = LockManager(db, lock_name)
 
     def _serialize(self, value: M) -> str | None:
         """Serializes the given value to a JSON string."""
@@ -135,3 +138,35 @@ class BlobManager[M]:
 
     def __repr__(self) -> str:
         return f"BlobManager(name='{self._name}')"
+
+    def acquire(
+        self,
+        timeout: Optional[float] = None,
+        lock_ttl: Optional[float] = None,
+        poll_interval: Optional[float] = None,
+    ) -> "BlobManager[M]":
+        """
+        Acquires an inter-process lock on this blob store, blocking until acquired.
+
+        Parameters override the default settings of the underlying LockManager.
+        """
+        self._lock.acquire(
+            timeout=timeout,
+            lock_ttl=lock_ttl,
+            poll_interval=poll_interval
+        )
+        return self
+
+    def release(self):
+        """
+        Releases the inter-process lock on this blob store.
+        """
+        self._lock.release()
+
+    def __enter__(self) -> "BlobManager[M]":
+        """Acquires the lock upon entering a 'with' statement."""
+        return self.acquire()
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """Releases the lock when exiting a 'with' statement."""
+        self.release()

beaver/collections.py

@@ -3,15 +3,16 @@ import sqlite3
 import threading
 import uuid
 from enum import Enum
-from typing import Any, Iterator, List, Literal, Tuple, Type, TypeVar
+from typing import Any, Iterator, List, Literal, Tuple, Type, TypeVar, Optional
 from .types import Model, stub, IDatabase
+from .locks import LockManager
 
 try:
     import numpy as np
     from .vectors import VectorIndex
 except ImportError:
-    np = stub("This feature requires to install beaver-db[faiss]")()
-    VectorIndex = stub("This feature requires to install beaver-db[faiss]")
+    np = stub("This feature requires to install beaver-db[vector]")()
+    VectorIndex = stub("This feature requires to install beaver-db[vector]")
 
 # --- Fuzzy Search Helper Functions ---
 
@@ -135,6 +136,9 @@ class CollectionManager[D: Document]:
         self._compaction_lock = threading.Lock()
         self._compaction_thread: threading.Thread | None = None
 
+        lock_name = f"__lock__collection__{name}"
+        self._lock = LockManager(db, lock_name)
+
     def _flatten_metadata(self, metadata: dict, prefix: str = "") -> dict[str, Any]:
         """Flattens a nested dictionary for indexing."""
         flat_dict = {}
@@ -643,6 +647,38 @@ class CollectionManager[D: Document]:
         cursor.close()
         return count
 
+    def acquire(
+        self,
+        timeout: Optional[float] = None,
+        lock_ttl: Optional[float] = None,
+        poll_interval: Optional[float] = None,
+    ) -> "CollectionManager[D]":
+        """
+        Acquires an inter-process lock on this collection, blocking until acquired.
+        This guarantees exclusive access for multi-step atomic operations
+        (e.g., index + connect).
+
+        Parameters override the default settings of the underlying LockManager.
+        """
+        self._lock.acquire(
+            timeout=timeout, lock_ttl=lock_ttl, poll_interval=poll_interval
+        )
+        return self
+
+    def release(self):
+        """
+        Releases the inter-process lock on this collection.
+        """
+        self._lock.release()
+
+    def __enter__(self) -> "CollectionManager[D]":
+        """Acquires the lock upon entering a 'with' statement."""
+        return self.acquire()
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """Releases the lock when exiting a 'with' statement."""
+        self.release()
+
 
 def rerank[D: Document](
     *results: list[D], weights: list[float] | None = None, k: int = 60

beaver/dicts.py

@@ -1,9 +1,9 @@
 import json
 import sqlite3
 import time
-from typing import Any, Iterator, Tuple, Type
-
+from typing import Any, Iterator, Tuple, Type, Optional
 from .types import JsonSerializable, IDatabase
+from .locks import LockManager
 
 class DictManager[T]:
     """A wrapper providing a Pythonic interface to a dictionary in the database."""
@@ -12,6 +12,8 @@ class DictManager[T]:
         self._name = name
         self._db = db
         self._model = model
+        lock_name = f"__lock__dict__{name}"
+        self._lock = LockManager(db, lock_name)
 
     def _serialize(self, value: T) -> str:
         """Serializes the given value to a JSON string."""
@@ -147,4 +149,35 @@ class DictManager[T]:
         cursor.close()
 
     def __repr__(self) -> str:
-        return f"DictManager(name='{self._name}')"
+        return f"DictManager(name='{self._name}')"
+
+    def acquire(
+        self,
+        timeout: Optional[float] = None,
+        lock_ttl: Optional[float] = None,
+        poll_interval: Optional[float] = None,
+    ) -> "DictManager[T]":
+        """
+        Acquires an inter-process lock on this dictionary, blocking until acquired.
+        Parameters override the default settings of the underlying LockManager.
+        """
+        self._lock.acquire(
+            timeout=timeout,
+            lock_ttl=lock_ttl,
+            poll_interval=poll_interval
+        )
+        return self
+
+    def release(self):
+        """
+        Releases the inter-process lock on this dictionary.
+        """
+        self._lock.release()
+
+    def __enter__(self) -> "DictManager[T]":
+        """Acquires the lock upon entering a 'with' statement."""
+        return self.acquire()
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """Releases the lock when exiting a 'with' statement."""
+        self.release()

beaver/lists.py

@@ -1,8 +1,8 @@
 import json
 import sqlite3
-from typing import Any, Iterator, Type, Union
-
+from typing import Any, Iterator, Type, Union, Optional
 from .types import JsonSerializable, IDatabase
+from .locks import LockManager
 
 
 class ListManager[T]:
@@ -12,6 +12,8 @@ class ListManager[T]:
         self._name = name
         self._db = db
         self._model = model
+        lock_name = f"__lock__list__{name}"
+        self._lock = LockManager(db, lock_name)
 
     def _serialize(self, value: T) -> str:
         """Serializes the given value to a JSON string."""
@@ -261,4 +263,36 @@ class ListManager[T]:
             cursor.execute(
                 "DELETE FROM beaver_lists WHERE rowid = ?", (rowid_to_delete,)
             )
-            return self._deserialize(value_to_return)
+            return self._deserialize(value_to_return)
+
+    def acquire(
+        self,
+        timeout: Optional[float] = None,
+        lock_ttl: Optional[float] = None,
+        poll_interval: Optional[float] = None,
+    ) -> "ListManager[T]":
+        """
+        Acquires an inter-process lock on this list, blocking until acquired.
+
+        Parameters override the default settings of the underlying LockManager.
+        """
+        self._lock.acquire(
+            timeout=timeout,
+            lock_ttl=lock_ttl,
+            poll_interval=poll_interval
+        )
+        return self
+
+    def release(self):
+        """
+        Releases the inter-process lock on this list, allowing the next waiting process access.
+        """
+        self._lock.release()
+
+    def __enter__(self) -> "ListManager[T]":
+        """Acquires the lock upon entering a 'with' statement."""
+        return self.acquire()
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """Releases the lock when exiting a 'with' statement."""
+        self.release()

beaver/locks.py

@@ -55,7 +55,11 @@ class LockManager:
         self._waiter_id = f"pid:{os.getpid()}:id:{uuid.uuid4()}"
         self._acquired = False  # State to track if this instance holds the lock
 
-    def acquire(self) -> "LockManager":
+    def acquire(self,
+                timeout: float|None = None,
+        lock_ttl: float |None = None,
+        poll_interval: float |None = None,
+        ) -> "LockManager":
         """
         Blocks until the lock is acquired or the timeout expires.
 
@@ -66,9 +70,18 @@ class LockManager:
             # This instance already holds the lock
             return self
 
+        if timeout is None:
+            timeout = self._timeout
+
+        if lock_ttl is None:
+            lock_ttl = self._lock_ttl
+
+        if poll_interval is None:
+            poll_interval = self._poll_interval
+
         start_time = time.time()
         requested_at = time.time()
-        expires_at = requested_at + self._lock_ttl
+        expires_at = requested_at + lock_ttl
 
         conn = self._db.connection
 
@@ -113,19 +126,19 @@ class LockManager:
                         return self
 
                 # 5. Check for timeout
-                if self._timeout is not None:
-                    if (time.time() - start_time) > self._timeout:
+                if timeout is not None:
+                    if (time.time() - start_time) > timeout:
                         # We timed out. Remove ourselves from the queue and raise.
                         self._release_from_queue()
                         raise TimeoutError(
-                            f"Failed to acquire lock '{self._lock_name}' within {self._timeout}s."
+                            f"Failed to acquire lock '{self._lock_name}' within {timeout}s."
                         )
 
                 # 6. Wait politely before polling again
                 # Add +/- 10% jitter to the poll interval to avoid thundering herd
-                jitter = self._poll_interval * 0.1
+                jitter = poll_interval * 0.1
                 sleep_time = random.uniform(
-                    self._poll_interval - jitter, self._poll_interval + jitter
+                    poll_interval - jitter, poll_interval + jitter
                 )
                 time.sleep(sleep_time)
 

beaver/queues.py

@@ -2,10 +2,9 @@ import asyncio
 import json
 import sqlite3
 import time
-from typing import Any, Literal, NamedTuple, Type, overload
-
+from typing import Any, Literal, NamedTuple, Type, overload, Optional
 from .types import JsonSerializable, IDatabase
-
+from .locks import LockManager
 
 class QueueItem[T](NamedTuple):
     """A data class representing a single item retrieved from the queue."""
@@ -54,6 +53,8 @@ class QueueManager[T]:
         self._name = name
         self._db = db
         self._model = model
+        lock_name = f"__lock__queue__{name}"
+        self._lock = LockManager(db, lock_name)
 
     def _serialize(self, value: T) -> str:
         """Serializes the given value to a JSON string."""
@@ -184,3 +185,37 @@ class QueueManager[T]:
 
     def __repr__(self) -> str:
         return f"QueueManager(name='{self._name}')"
+
+    def acquire(
+        self,
+        timeout: Optional[float] = None,
+        lock_ttl: Optional[float] = None,
+        poll_interval: Optional[float] = None,
+    ) -> "QueueManager[T]":
+        """
+        Acquires an inter-process lock on this queue, blocking until acquired.
+        This ensures that a sequence of operations (e.g., batch-getting tasks)
+        is performed atomically without interruption from other processes.
+
+        Parameters override the default settings of the underlying LockManager.
+        """
+        self._lock.acquire(
+            timeout=timeout,
+            lock_ttl=lock_ttl,
+            poll_interval=poll_interval
+        )
+        return self
+
+    def release(self):
+        """
+        Releases the inter-process lock on this queue.
+        """
+        self._lock.release()
+
+    def __enter__(self) -> "QueueManager[T]":
+        """Acquires the lock upon entering a 'with' statement."""
+        return self.acquire()
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """Releases the lock when exiting a 'with' statement."""
+        self.release()

{beaver_db-0.19.2.dist-info → beaver_db-0.19.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beaver-db
-Version: 0.19.2
+Version: 0.19.3
 Summary: Fast, embedded, and multi-modal DB based on SQLite for AI-powered applications.
 License-File: LICENSE
 Classifier: License :: OSI Approved :: MIT License

beaver_db-0.19.3.dist-info/RECORD

@@ -0,0 +1,19 @@
+beaver/__init__.py,sha256=DBnqxMZ-ZIvX14_FoC_ZGmYqcqhQtpLG1VXjNVCcPdc,125
+beaver/blobs.py,sha256=gauxfWOCMoP6mq61JRj-TVqqkXMAcVI2taBGYnJXsxg,5527
+beaver/channels.py,sha256=kIuwKMDBdDQObaKT23znsMXzfpKfE7pXSxvf-u4LlpY,9554
+beaver/cli.py,sha256=Sxm-mYU3LGd4tIqw-5LHb0ektWebjV9vn51hm-CMJD0,2232
+beaver/collections.py,sha256=Vc5ZF5gVIACx2XeL25PxgOcW7iF6lBInO-cnfyBBUeo,27299
+beaver/core.py,sha256=JRkRvc0Sb3FT9KlR3YbmiPcqCQ686dFKmHSNZ_UJ_aE,17100
+beaver/dicts.py,sha256=0csA1fvQb3rB2yQvwuq_kzgJ0YPp-wifBLkNcuiUltI,6582
+beaver/lists.py,sha256=vKZPVRiw0BRSx4JKaboGnWXfwNGkHX8cBFH59qQX1yY,11258
+beaver/locks.py,sha256=b6cHgAI1YQxKmVdLLzNw9fcGvk4V3SQS_wUaQTHSsUk,6630
+beaver/logs.py,sha256=a5xenwl5NZeegIU0dWVEs67lvaHzzw-JRAZtEzNNO3E,9529
+beaver/queues.py,sha256=wTlxsu12grcIPbosA1S0_23Tot4Wv3BWXeKyq4dRPKk,7709
+beaver/server.py,sha256=At3BoEV7JfpYjNtyHMdPUF8shj4V4D5nStXWb6Bv53A,15947
+beaver/types.py,sha256=m0ohT7A8r0Y1a7bJEx4VanLaOUWU2VYxaLHPsVPjrIw,1651
+beaver/vectors.py,sha256=EGZf1s364-rMubxkYoTcjBl72lRRxM1cUwypjsoC6ec,18499
+beaver_db-0.19.3.dist-info/METADATA,sha256=gRJetTcWICFhi0kQUFeNt773vTZkZYF-iRFqCtUXqjE,23431
+beaver_db-0.19.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beaver_db-0.19.3.dist-info/entry_points.txt,sha256=bd5E2s45PoBdtdR9-ToKSdLNhmHp8naV1lWP5mOzlrc,42
+beaver_db-0.19.3.dist-info/licenses/LICENSE,sha256=1xrIY5JnMk_QDQzsqmVzPIIyCgZAkWCC8kF2Ddo1UT0,1071
+beaver_db-0.19.3.dist-info/RECORD,,

beaver_db-0.19.2.dist-info/RECORD

@@ -1,19 +0,0 @@
-beaver/__init__.py,sha256=HjTNjk3x58Pw9Iv_eAPVL088wDVEua-JEgecZ28NgMc,125
-beaver/blobs.py,sha256=U5n6NLRQGAzsePGR2SJPRXHy22K8T9cJVMIb0JGzsY0,4399
-beaver/channels.py,sha256=kIuwKMDBdDQObaKT23znsMXzfpKfE7pXSxvf-u4LlpY,9554
-beaver/cli.py,sha256=Sxm-mYU3LGd4tIqw-5LHb0ektWebjV9vn51hm-CMJD0,2232
-beaver/collections.py,sha256=UAQAuRxJRCqY5PHfxJNm3CdKqMNuyY8DOLdodvY6jpk,26107
-beaver/core.py,sha256=JRkRvc0Sb3FT9KlR3YbmiPcqCQ686dFKmHSNZ_UJ_aE,17100
-beaver/dicts.py,sha256=Xp8lPfQt08O8zCbptQLWQLO79OxG6uAVER6ryj3SScQ,5495
-beaver/lists.py,sha256=rfJ8uTNLkMREYc0uGx0z1VKt2m3eR9hvbdvDD58EbmQ,10140
-beaver/locks.py,sha256=GWDSRkPw2lrAQfXIRqvkc5PK9zZ2eLYWKTuzHTs9j_A,6321
-beaver/logs.py,sha256=a5xenwl5NZeegIU0dWVEs67lvaHzzw-JRAZtEzNNO3E,9529
-beaver/queues.py,sha256=Fr3oie63EtceSoiC8EOEDSLu1tDI8q2MYLXd8MEeC3g,6476
-beaver/server.py,sha256=At3BoEV7JfpYjNtyHMdPUF8shj4V4D5nStXWb6Bv53A,15947
-beaver/types.py,sha256=m0ohT7A8r0Y1a7bJEx4VanLaOUWU2VYxaLHPsVPjrIw,1651
-beaver/vectors.py,sha256=EGZf1s364-rMubxkYoTcjBl72lRRxM1cUwypjsoC6ec,18499
-beaver_db-0.19.2.dist-info/METADATA,sha256=Iec3mTpq384nkp_R8fD2AGjXHRzarA93uxL623ZtyVE,23431
-beaver_db-0.19.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beaver_db-0.19.2.dist-info/entry_points.txt,sha256=bd5E2s45PoBdtdR9-ToKSdLNhmHp8naV1lWP5mOzlrc,42
-beaver_db-0.19.2.dist-info/licenses/LICENSE,sha256=1xrIY5JnMk_QDQzsqmVzPIIyCgZAkWCC8kF2Ddo1UT0,1071
-beaver_db-0.19.2.dist-info/RECORD,,