apexdevkit 1.9.3__tar.gz → 1.9.5__tar.gz

{apexdevkit-1.9.3 → apexdevkit-1.9.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apexdevkit
-Version: 1.9.3
+Version: 1.9.5
 Summary: Apex Development Tools for python.
 Author: Apex Dev
 Author-email: dev@apex.ge

{apexdevkit-1.9.3 → apexdevkit-1.9.5}/apexdevkit/repository/connector.py

@@ -35,10 +35,8 @@ class SqliteInMemoryConnector:
 
 
 @dataclass
-class MongoDBConnector:
+class PyMongoConnector:
     dsn: str
 
     def connect(self) -> ContextManager[MongoClient[Any]]:
-        connection: MongoClient[Any] = MongoClient(self.dsn)
-
-        return connection
+        return MongoClient(self.dsn)

{apexdevkit-1.9.3 → apexdevkit-1.9.5}/apexdevkit/repository/database.py

@@ -1,11 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import Any, ContextManager, Dict, Iterable, Iterator, Protocol, Self
-
-from pymongo import MongoClient
-from pymongo.collection import Collection, ReturnDocument
-from pymongo.results import DeleteResult, InsertOneResult
+from typing import Any, ContextManager, Iterable, Protocol, Self
 
 _RawData = dict[str, Any]
 
@@ -46,59 +42,11 @@ class Database:
         return [dict(raw or {}) for raw in raw]
 
 
-@dataclass
-class MongoDatabase:
-    connector: MongoConnector
-    database_name: str
-    collection_name: str
-
-    def collection(self, client: MongoClient[Any]) -> Collection[Any]:
-        return client[self.database_name][self.collection_name]
-
-    def __iter__(self) -> Iterator[Any]:
-        with self.connector.connect() as client:
-            for raw in self.collection(client).find():
-                yield raw
-
-    def __len__(self) -> int:
-        with self.connector.connect() as client:
-            return self.collection(client).count_documents({})
-
-    def create(self, item: Dict[str, Any]) -> InsertOneResult:
-        with self.connector.connect() as client:
-            return self.collection(client).insert_one(item)
-
-    def read(self, item_id: str) -> Dict[str, Any] | None:
-        with self.connector.connect() as client:
-            return self.collection(client).find_one({"id": item_id})
-
-    def update(self, item_id: str, item: Dict[str, Any]) -> Any:
-        with self.connector.connect() as client:
-            return self.collection(client).find_one_and_update(
-                {"id": item_id},
-                {"$set": item},
-                return_document=ReturnDocument.AFTER,
-            )
-
-    def delete(self, item_id: str) -> DeleteResult:
-        with self.connector.connect() as client:
-            return self.collection(client).delete_one({"id": item_id})
-
-    def delete_all(self) -> DeleteResult:
-        with self.connector.connect() as client:
-            return self.collection(client).delete_many({})
-
-
 class Connector(Protocol):  # pragma: no cover
     def connect(self) -> ContextManager[Connection]:
         pass
 
 
-class MongoConnector(Protocol):  # pragma: no cover
-    def connect(self) -> ContextManager[MongoClient[Any]]:
-        pass
-
-
 class Connection(Protocol):  # pragma: no cover
     def cursor(self) -> Cursor:
         pass

apexdevkit-1.9.5/apexdevkit/repository/mongo.py

@@ -0,0 +1,110 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any, ContextManager, Dict, Generic, Iterator, Protocol, TypeVar
+
+from pymongo import MongoClient, ReturnDocument
+from pymongo.collection import Collection
+from pymongo.results import DeleteResult, InsertOneResult
+
+from apexdevkit.error import DoesNotExistError, ExistsError
+from apexdevkit.formatter import Formatter
+
+
+class _Item(Protocol):  # pragma: no cover
+    @property
+    def id(self) -> Any:
+        pass
+
+
+ItemT = TypeVar("ItemT", bound=_Item)
+
+
+@dataclass
+class MongoRepository(Generic[ItemT]):
+    database: MongoDatabase
+    formatter: Formatter[dict[str, Any], ItemT]
+
+    def __iter__(self) -> Iterator[ItemT]:
+        for raw in self.database:
+            yield self.formatter.load(raw)
+
+    def __len__(self) -> int:
+        return len(self.database)
+
+    def create(self, item: ItemT) -> ItemT:
+        try:
+            self.read(item.id)
+            raise ExistsError(item).with_duplicate(
+                lambda i: f"_Item with id<{i.id}> already exists."
+            )
+        except DoesNotExistError:
+            self.database.create(self.formatter.dump(item))
+            return item
+
+    def create_many(self, items: list[ItemT]) -> list[ItemT]:
+        return [self.create(item) for item in items]
+
+    def read(self, item_id: str) -> ItemT:
+        raw = self.database.read(item_id)
+
+        if not raw:
+            raise DoesNotExistError(item_id)
+
+        return self.formatter.load(raw)
+
+    def update(self, item: ItemT) -> None:
+        self.database.update(item.id, self.formatter.dump(item))
+
+    def update_many(self, items: list[ItemT]) -> None:
+        for item in items:
+            self.update(item)
+
+    def delete(self, item_id: str) -> None:
+        result = self.database.delete(item_id)
+        if result.deleted_count == 0:
+            raise DoesNotExistError(item_id)
+
+
+@dataclass
+class MongoDatabase:
+    connector: MongoConnector
+    database_name: str
+    collection_name: str
+
+    def collection(self, client: MongoClient[Any]) -> Collection[Any]:
+        return client[self.database_name][self.collection_name]
+
+    def __iter__(self) -> Iterator[Any]:
+        with self.connector.connect() as client:
+            for raw in self.collection(client).find():
+                yield raw
+
+    def __len__(self) -> int:
+        with self.connector.connect() as client:
+            return self.collection(client).count_documents({})
+
+    def create(self, item: Dict[str, Any]) -> InsertOneResult:
+        with self.connector.connect() as client:
+            return self.collection(client).insert_one(item)
+
+    def read(self, item_id: str) -> Dict[str, Any] | None:
+        with self.connector.connect() as client:
+            return self.collection(client).find_one({"id": item_id})
+
+    def update(self, item_id: str, item: Dict[str, Any]) -> Any:
+        with self.connector.connect() as client:
+            return self.collection(client).find_one_and_update(
+                {"id": item_id},
+                {"$set": item},
+                return_document=ReturnDocument.AFTER,
+            )
+
+    def delete(self, item_id: str) -> DeleteResult:
+        with self.connector.connect() as client:
+            return self.collection(client).delete_one({"id": item_id})
+
+
+class MongoConnector(Protocol):  # pragma: no cover
+    def connect(self) -> ContextManager[MongoClient[Any]]:
+        pass

{apexdevkit-1.9.3 → apexdevkit-1.9.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "apexdevkit"
-version = "1.9.3"
+version = "1.9.5"
 description = "Apex Development Tools for python."
 authors = ["Apex Dev <dev@apex.ge>"]
 readme = "README.md"
@@ -17,6 +17,7 @@ pytest-cov = "*"
 pytest-recording = "*"
 coverage = "*"
 faker = "*"
+mongomock = "*"
 
 [tool.poetry.group.lint.dependencies]
 mypy = "*"

apexdevkit-1.9.3/apexdevkit/repository/mongo.py

@@ -1,79 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass
-from typing import Any, Dict, Generic, Iterator, Protocol, TypeVar
-
-from apexdevkit.error import DoesNotExistError, ExistsError
-from apexdevkit.repository.database import MongoDatabase
-
-
-class _Item(Protocol):  # pragma: no cover
-    @property
-    def id(self) -> Any:
-        pass
-
-
-ItemT = TypeVar("ItemT", bound=_Item)
-
-
-@dataclass
-class MongoDBRepository(Generic[ItemT]):
-    database: MongoDatabase
-    table: MongoTable[ItemT]
-
-    def __iter__(self) -> Iterator[ItemT]:
-        for raw in self.database:
-            yield self.table.load(raw)
-
-    def __len__(self) -> int:
-        return len(self.database)
-
-    def create(self, item: ItemT) -> ItemT:
-        try:
-            self.read(item.id)
-            raise ExistsError(item).with_duplicate(
-                lambda i: f"_Item with id<{i.id}> already exists."
-            )
-        except DoesNotExistError:
-            self.database.create(self.table.to_dict(item))
-            return item
-
-    def create_many(self, items: list[ItemT]) -> list[ItemT]:
-        return [self.create(item) for item in items]
-
-    def read(self, item_id: str) -> ItemT:
-        raw = self.database.read(item_id)
-
-        if not raw:
-            raise DoesNotExistError(item_id)
-
-        return self.table.load(raw)
-
-    def update(self, item: ItemT) -> None:
-        self.database.update(
-            self.table.get_id(item),
-            self.table.to_dict(item),
-        )
-
-    def update_many(self, items: list[ItemT]) -> None:
-        for item in items:
-            self.update(item)
-
-    def delete(self, item_id: str) -> None:
-        result = self.database.delete(item_id)
-        if result.deleted_count == 0:
-            raise DoesNotExistError(item_id)
-
-    def delete_all(self) -> None:
-        self.database.delete_all()
-
-
-class MongoTable(Generic[ItemT]):
-    def to_dict(self, item: ItemT) -> Dict[str, Any]:
-        raise NotImplementedError("Not implemented")
-
-    def load(self, data: Dict[str, Any]) -> ItemT:
-        raise NotImplementedError("Not implemented")
-
-    def get_id(self, item: ItemT) -> str:
-        raise NotImplementedError("Not implemented")