kover 2.7.0__tar.gz → 2.7.2__tar.gz

{kover-2.7.0 → kover-2.7.2}/.gitignore

@@ -1,3 +1,4 @@
+uv.lock
 __pycache__
 .vscode
 test.py
@@ -8,6 +9,4 @@ badges.txt
 dist
 .ruff_cache
 .python-version
-.token
-uv.lock
-kover/bulk
+.token

{kover-2.7.0 → kover-2.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kover
-Version: 2.7.0
+Version: 2.7.2
 Summary: Fully async mongodb driver for mongod and replica sets.
 Project-URL: Repository, https://github.com/megawattka/kover
 Project-URL: Documentation, https://github.com/megawattka/kover
@@ -66,7 +66,7 @@ The main reason why i created this project is that Motor - official async wrappe
 
 # Status
 it still missing features. <br>
-e.g: **bulk write API** and **Encryption**
+e.g: **Encryption** and **AWS auth**
 but its already very cool! <br>
 
 # Dependencies

{kover-2.7.0 → kover-2.7.2}/README.md

@@ -36,7 +36,7 @@ The main reason why i created this project is that Motor - official async wrappe
 
 # Status
 it still missing features. <br>
-e.g: **bulk write API** and **Encryption**
+e.g: **Encryption** and **AWS auth**
 but its already very cool! <br>
 
 # Dependencies

{kover-2.7.0 → kover-2.7.2}/kover/__init__.py

@@ -20,11 +20,12 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-__version__ = "2.7.0"
+__version__ = "2.7.2"
 __author__ = "megawattka"
 __license__ = "MIT"
 __copyright__ = "Copyright (C) 2024-present megawattka"
 
+from .bulk_write_builder import BulkWriteBuilder
 from .client import Kover
 from .collection import Collection
 from .cursor import Cursor
@@ -65,6 +66,7 @@ from .typings import xJsonT
 __all__ = (
     "AuthCredentials",
     "BuildInfo",
+    "BulkWriteBuilder",
     "Collation",
     "CollationStrength",
     "Collection",

kover-2.7.2/kover/bulk_write_builder.py

@@ -0,0 +1,104 @@
+"""Builder for kover BulkWrite operations."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from .helpers import filter_non_null
+
+if TYPE_CHECKING:
+    from .models import Delete, Update
+    from .typings import xJsonT
+
+
+# https://www.mongodb.com/docs/manual/reference/command/bulkWrite
+class BulkWriteBuilder:
+    """Builder for bulk write operations."""
+
+    def __init__(
+        self,
+        *,
+        ordered: bool = True,
+        bypass_document_validation: bool = False,
+        comment: str | None = None,
+        let: xJsonT | None = None,
+        errors_only: bool = False,
+        cursor_batch_size: int | None = None,
+        write_concern: str = "majority",
+    ) -> None:
+        self.ordered = ordered
+        self.bypass_document_validation = bypass_document_validation
+        self.comment = comment
+        self.let = let
+        self.errors_only = errors_only
+        self.cursor = {"batchSize": cursor_batch_size or 101}
+        self.write_concern = write_concern
+        self._operations: list[xJsonT] = []
+        self._namespaces: list[str] = []
+
+    def _prepair_additional_params(self) -> xJsonT:
+        return filter_non_null({
+            "ordered": self.ordered,
+            "bypassDocumentValidation": self.bypass_document_validation,
+            "comment": self.comment,
+            "let": self.let,
+            "errorsOnly": self.errors_only,
+            "cursor": self.cursor,
+            "writeConcern": {"w": self.write_concern},
+        })
+
+    def _get_ns_idx(self, namespace: str) -> int:
+        if namespace not in self._namespaces:
+            self._namespaces.append(namespace)
+        return self._namespaces.index(namespace)
+
+    def add_insert(
+        self,
+        documents: list[xJsonT],
+        /,
+        *,
+        ns: str,
+    ) -> None:
+        """Adds an Insert operations into the builder."""
+        idx = self._get_ns_idx(ns)
+        for document in documents:
+            self._operations.append({"insert": idx, "document": document})
+
+    def add_update(
+        self,
+        *updates: Update,
+        ns: str,
+    ) -> None:
+        """Adds an Update operations into the builder."""
+        idx = self._get_ns_idx(ns)
+        for update in updates:
+            self._operations.append({
+                "update": idx,
+                **update.as_bulk_write_op(),
+            })
+
+    def add_delete(
+        self,
+        *deletes: Delete,
+        ns: str,
+    ) -> None:
+        """Adds an Delete operations into the builder."""
+        idx = self._get_ns_idx(ns)
+        for delete in deletes:
+            self._operations.append({
+                "delete": idx,
+                **delete.as_bulk_write_op(),
+            })
+
+    def build(self) -> xJsonT:
+        """Build the command.
+
+        Returns:
+            The command that was built for Kover.bulk_write method.
+        """
+        return {
+            "bulkWrite": 1,
+            "ops": self._operations,
+            "nsInfo": [{"ns": ns} for ns in self._namespaces],
+            **self._prepair_additional_params(),
+        }

{kover-2.7.0 → kover-2.7.2}/kover/client.py

@@ -239,6 +239,18 @@ class Kover:
         finally:
             await self._pool.put(conn)
 
+    async def bulk_write(
+        self,
+        document: xJsonT,
+        transaction: Transaction | None = None,
+    ) -> xJsonT:
+        """Execute a bulkWrite operation and return info about it.
+
+        Returns:
+            Document, containing info about this operation.
+        """
+        return await self.request(document, transaction=transaction)
+
     async def refresh_sessions(self, sessions: list[Session]) -> None:
         """Refresh the provided list of sessions.
 

{kover-2.7.0 → kover-2.7.2}/kover/database.py

@@ -16,7 +16,7 @@ if TYPE_CHECKING:
     from .typings import xJsonT
 
 
-@classrepr("name")
+@classrepr("name", "client")
 class Database:
     """Represents a database instance, providing methods to manage database.
 

{kover-2.7.0 → kover-2.7.2}/kover/models/operations.py

@@ -2,11 +2,10 @@
 
 from __future__ import annotations
 
-from typing import Literal
-
 from pydantic import BaseModel
 
 from .._internals._mixins import ModelMixin as _ModelMixin
+from ..helpers import filter_non_null
 from ..typings import xJsonT  # noqa: TC001
 from .other import Collation  # noqa: TC001
 
@@ -27,13 +26,29 @@ class Update(_ModelMixin):
 
     q: xJsonT
     u: xJsonT
-    c: xJsonT | None = None
+    c: xJsonT | None = None  # constants
     upsert: bool = False
     multi: bool = False
     collation: Collation | None = None
     array_filters: xJsonT | None = None
     hint: str | None = None
 
+    def as_bulk_write_op(self) -> xJsonT:
+        """Serialize This model for BulkWriteBuilder.
+
+        Returns:
+            Serialized operation.
+        """
+        return filter_non_null({
+            "filter": self.q,
+            "updateMods": self.u,
+            "arrayFilters": self.array_filters,
+            "multi": self.multi,
+            "hint": self.hint,
+            "constants": self.c,
+            "collation": self.collation,
+        })
+
 
 # https://www.mongodb.com/docs/manual/reference/command/delete/#syntax
 class Delete(_ModelMixin):
@@ -42,7 +57,20 @@ class Delete(_ModelMixin):
     def __init__(self, q: xJsonT, /, **kwargs: object) -> None:
         BaseModel.__init__(self, q=q, **kwargs)
 
-    q: xJsonT
-    limit: Literal[0, 1]
+    q: xJsonT  # query
+    limit: int
     collation: Collation | None = None
     hint: xJsonT | str | None = None
+
+    def as_bulk_write_op(self) -> xJsonT:
+        """Serialize This model for BulkWriteBuilder.
+
+        Returns:
+            Serialized operation.
+        """
+        return filter_non_null({
+            "filter": self.q,
+            "multi": self.limit != 1,
+            "hint": self.hint,
+            "collation": self.collation,
+        })

{kover-2.7.0 → kover-2.7.2}/kover/network/transport.py

@@ -76,7 +76,7 @@ class MongoTransport:
                 if not self._writer.is_closing():
                     self._writer.close()
 
-    async def send(self, msg: bytes) -> None:
+    async def _send(self, msg: bytes) -> None:
         """Send a message to the MongoDB server.
 
         Raises:
@@ -87,7 +87,7 @@ class MongoTransport:
         self._writer.write(msg)
         await self._writer.drain()
 
-    async def recv(self, size: int) -> bytes:
+    async def _recv(self, size: int) -> bytes:
         """Receive a message from the MongoDB server.
 
         Returns:
@@ -119,11 +119,11 @@ class MongoTransport:
             transaction.apply_to(doc)
         rid, msg = self._helper.get_message(doc, compressor=self._compressor)
 
-        await self.send(msg)
+        await self._send(msg)
         if wait_response:
-            header = await self.recv(16)
+            header = await self._recv(16)
             length, op_code = self._helper.verify_rid(header, rid)
-            data = await self.recv(length - 16)  # exclude header
+            data = await self._recv(length - 16)  # exclude header
             reply = self._helper.get_reply(data, op_code)
         else:  # cases like kover.shutdown()
             return {}

{kover-2.7.0 → kover-2.7.2}/kover/schema.py

@@ -202,9 +202,9 @@ class SchemaGenerator:
 
     @staticmethod
     def _generate_metadata(metadata: list[Any]) -> xJsonT:
-        for _meta in metadata:
-            if isinstance(_meta, SchemaMetadata):
-                return _meta.serialize()
+        for meta in metadata:
+            if isinstance(meta, SchemaMetadata):
+                return meta.serialize()
         return {}
 
 

{kover-2.7.0 → kover-2.7.2}/kover/transaction.py

@@ -9,6 +9,7 @@ from typing_extensions import Self
 
 from .bson import Int64
 from .enums import TxnState
+from .helpers import classrepr
 
 if TYPE_CHECKING:
     from types import TracebackType
@@ -17,6 +18,7 @@ if TYPE_CHECKING:
     from .typings import xJsonT
 
 
+@classrepr("_id", "state", "session_document")
 class Transaction:
     """Represents a MongoDB transaction.
 
@@ -36,7 +38,7 @@ class Transaction:
     ) -> None:
         self.client = client
         self.session_document: xJsonT = session_document
-        self.id: Int64 = Int64(-1)
+        self._id: Int64 = Int64(-1)
         self.state: TxnState = TxnState.NONE
         self.action_count: int = 0
         self.exception: BaseException | None = None

{kover-2.7.0 → kover-2.7.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kover"
-version = "2.7.0"
+version = "2.7.2"
 description = "Fully async mongodb driver for mongod and replica sets."
 readme = "README.md"
 authors = [{ name = "megawattka", email = "gunter56741@gmail.com" }]