scruby 0.9.2__py3-none-any.whl → 0.10.0__py3-none-any.whl

scruby/db.py

@@ -6,6 +6,7 @@ __all__ = ("Scruby",)
 
 import concurrent.futures
 import contextlib
+import datetime
 import logging
 import zlib
 from collections.abc import Callable
@@ -15,6 +16,7 @@ from typing import Any, Never, TypeVar, assert_never
 
 import orjson
 from anyio import Path, to_thread
+from pydantic import BaseModel
 
 from scruby import constants
 
@@ -23,6 +25,14 @@ logger = logging.getLogger(__name__)
 T = TypeVar("T")
 
 
+class _Meta(BaseModel):
+    """Metadata of collection."""
+
+    count_documents: int
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+
+
 class Scruby[T]:
     """Creation and management of database.
 
@@ -34,6 +44,7 @@ class Scruby[T]:
         self,
         class_model: T,
     ) -> None:
+        self.__meta = _Meta
         self.__class_model = class_model
         self.__db_root = constants.DB_ROOT
         self.__length_reduction_hash = constants.LENGTH_REDUCTION_HASH
@@ -51,10 +62,76 @@ class Scruby[T]:
                 msg: str = f"{unreachable} - Unacceptable value for LENGTH_REDUCTION_HASH."
                 logger.critical(msg)
                 assert_never(Never(unreachable))
+        # Create metadata if absent.
+        self._create_metadata()
+
+    def _create_metadata(self) -> None:
+        """Create metadata for collection if absent.
+
+        This method is for internal use.
+        """
+        key: int = 0
+        key_as_hash: str = f"{key:08x}"[self.__length_reduction_hash :]
+        separated_hash: str = "/".join(list(key_as_hash))
+        branch_path = SyncPath(
+            *(
+                self.__db_root,
+                self.__class_model.__name__,
+                separated_hash,
+            ),
+        )
+        if not branch_path.exists():
+            branch_path.mkdir(parents=True)
+            meta = _Meta(
+                count_documents=0,
+                created_at=datetime.datetime.now(),  # noqa: DTZ005
+                updated_at=datetime.datetime.now(),  # noqa: DTZ005
+            )
+            meta_json = meta.model_dump_json()
+            meta_path = SyncPath(*(branch_path, "meta.json"))
+            meta_path.write_text(meta_json, "utf-8")
+
+    async def _get_meta_path(self) -> Path:
+        """Asynchronous method for getting path to metadata of collection.
+
+        This method is for internal use.
+        """
+        key: int = 0
+        key_as_hash: str = f"{key:08x}"[self.__length_reduction_hash :]
+        separated_hash: str = "/".join(list(key_as_hash))
+        return Path(
+            *(
+                self.__db_root,
+                self.__class_model.__name__,
+                separated_hash,
+                "meta.json",
+            ),
+        )
 
-    async def get_leaf_path(self, key: str) -> Path:
+    async def _get_meta(self) -> _Meta:
+        """Asynchronous method for getting metadata of collection.
+
+        This method is for internal use.
+        """
+        meta_path = await self._get_meta_path()
+        meta_json = await meta_path.read_text()
+        meta: _Meta = self.__meta.model_validate_json(meta_json)
+        return meta
+
+    async def _set_meta(self, meta: _Meta) -> None:
+        """Asynchronous method for updating metadata of collection.
+
+        This method is for internal use.
+        """
+        meta_path = await self._get_meta_path()
+        meta_json = meta.model_dump_json()
+        await meta_path.write_text(meta_json, "utf-8")
+
+    async def _get_leaf_path(self, key: str) -> Path:
         """Asynchronous method for getting path to collection cell by key.
 
+        This method is for internal use.
+
         Args:
             key: Key name.
         """
@@ -95,7 +172,7 @@ class Scruby[T]:
             value: Value of key.
         """
         # The path to the database cell.
-        leaf_path: Path = await self.get_leaf_path(key)
+        leaf_path: Path = await self._get_leaf_path(key)
         value_json: str = value.model_dump_json()
         # Write key-value to the database.
         if await leaf_path.exists():
@@ -115,7 +192,7 @@ class Scruby[T]:
             key: Key name.
         """
         # The path to the database cell.
-        leaf_path: Path = await self.get_leaf_path(key)
+        leaf_path: Path = await self._get_leaf_path(key)
         # Get value of key.
         if await leaf_path.exists():
             data_json: bytes = await leaf_path.read_bytes()
@@ -133,7 +210,7 @@ class Scruby[T]:
             key: Key name.
         """
         # The path to the database cell.
-        leaf_path: Path = await self.get_leaf_path(key)
+        leaf_path: Path = await self._get_leaf_path(key)
         # Checking whether there is a key.
         if await leaf_path.exists():
             data_json: bytes = await leaf_path.read_bytes()
@@ -152,7 +229,7 @@ class Scruby[T]:
             key: Key name.
         """
         # The path to the database cell.
-        leaf_path: Path = await self.get_leaf_path(key)
+        leaf_path: Path = await self._get_leaf_path(key)
         # Deleting key.
         if await leaf_path.exists():
             data_json: bytes = await leaf_path.read_bytes()
@@ -178,14 +255,17 @@ class Scruby[T]:
         return
 
     @staticmethod
-    def search_task(
+    def _search_task(
         key: int,
         filter_fn: Callable,
         length_reduction_hash: str,
         db_root: str,
         class_model: T,
     ) -> dict[str, Any] | None:
-        """Search task."""
+        """Task for searching for documents.
+
+        This method is for internal use.
+        """
         key_as_hash: str = f"{key:08x}"[length_reduction_hash:]
         separated_hash: str = "/".join(list(key_as_hash))
         leaf_path: SyncPath = SyncPath(
@@ -226,7 +306,7 @@ class Scruby[T]:
                      If None, then there is no limit on the wait time.
         """
         keys: range = range(1, self.__max_num_keys)
-        search_task_fn: Callable = self.search_task
+        search_task_fn: Callable = self._search_task
         length_reduction_hash: int = self.__length_reduction_hash
         db_root: str = self.__db_root
         class_model: T = self.__class_model
@@ -245,7 +325,7 @@ class Scruby[T]:
                     return doc
         return None
 
-    def find_many(
+    def find(
         self,
         filter_fn: Callable,
         db_query_docs_limit: int = 1000,
@@ -268,7 +348,7 @@ class Scruby[T]:
                      If None, then there is no limit on the wait time.
         """
         keys: range = range(1, self.__max_num_keys)
-        search_task_fn: Callable = self.search_task
+        search_task_fn: Callable = self._search_task
         length_reduction_hash: int = self.__length_reduction_hash
         db_root: str = self.__db_root
         class_model: T = self.__class_model

{scruby-0.9.2.dist-info → scruby-0.10.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: scruby
-Version: 0.9.2
+Version: 0.10.0
 Summary: A fast key-value storage library.
 Project-URL: Homepage, https://github.com/kebasyaty/scruby
 Project-URL: Repository, https://github.com/kebasyaty/scruby
@@ -274,7 +274,7 @@ async def main() -> None:
         await db.set_key(f"+44798612345{num}", user)
 
     # Find users by email.
-    users: list[User] | None = user_coll.find_many(
+    users: list[User] | None = user_coll.find(
         filter_fn=lambda doc: doc.email == "John_Smith_5@gmail.com" or doc.email == "John_Smith_8@gmail.com",
     )
     if users is not None:

scruby-0.10.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+scruby/__init__.py,sha256=wFwUS1KcLxfIopXOVS8gPue9fNzIIU2cVj_RgK5drz4,849
+scruby/constants.py,sha256=GbB-O0qaVdi5EHUp-zRAppFXLR-oHxpXUFVAOCpS0C8,1022
+scruby/db.py,sha256=kwF1X_HB2CaNGNY0DVrC6s8OQ5Iix1azQOmfA_10JBo,13563
+scruby/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+scruby-0.10.0.dist-info/METADATA,sha256=seuAAwZHI-iso9O4durUh4dSye1KQrSZ8MjTkyOvCq8,10829
+scruby-0.10.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+scruby-0.10.0.dist-info/licenses/LICENSE,sha256=2zZINd6m_jNYlowdQImlEizyhSui5cBAJZRhWQURcEc,1095
+scruby-0.10.0.dist-info/RECORD,,

scruby-0.9.2.dist-info/RECORD

@@ -1,8 +0,0 @@
-scruby/__init__.py,sha256=wFwUS1KcLxfIopXOVS8gPue9fNzIIU2cVj_RgK5drz4,849
-scruby/constants.py,sha256=GbB-O0qaVdi5EHUp-zRAppFXLR-oHxpXUFVAOCpS0C8,1022
-scruby/db.py,sha256=W3VvC3ss1NXonQnFI6Yw2MobBARMgus5Acw-k5uL3TE,10946
-scruby/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-scruby-0.9.2.dist-info/METADATA,sha256=vZab1ietT1of_GXLHEISYhIkXZJuBIp3vk_rAFGeNPA,10833
-scruby-0.9.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-scruby-0.9.2.dist-info/licenses/LICENSE,sha256=2zZINd6m_jNYlowdQImlEizyhSui5cBAJZRhWQURcEc,1095
-scruby-0.9.2.dist-info/RECORD,,