apexdevkit 1.9.8__tar.gz → 1.10.1__tar.gz

{apexdevkit-1.9.8 → apexdevkit-1.10.1}/PKG-INFO

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

{apexdevkit-1.9.8 → apexdevkit-1.10.1}/apexdevkit/repository/in_memory.py

@@ -1,8 +1,8 @@
 from copy import deepcopy
 from dataclasses import dataclass, field
-from typing import Any, Generic, Iterable, Iterator, Protocol, Self, TypeVar
+from typing import Any, Callable, Generic, Iterable, Iterator, Protocol, Self, TypeVar
 
-from apexdevkit.error import Criteria, DoesNotExistError, ExistsError
+from apexdevkit.error import DoesNotExistError, ExistsError
 from apexdevkit.formatter import DataclassFormatter, Formatter
 
 
@@ -12,32 +12,33 @@ class _Item(Protocol):  # pragma: no cover
         pass
 
 
+KeyFunction = Callable[[Any], str]
+
 ItemT = TypeVar("ItemT", bound=_Item)
 _Raw = dict[str, Any]
 
 
+@dataclass
+class AttributeKey:
+    name: str
+
+    def __call__(self, item: Any) -> str:
+        return str(getattr(item, self.name))
+
+
 @dataclass
 class InMemoryRepository(Generic[ItemT]):
     formatter: Formatter[_Raw, ItemT]
     items: dict[str, _Raw] = field(default_factory=dict)
 
-    _uniques: list[Criteria] = field(init=False, default_factory=list)
-    _search_by: list[str] = field(init=False, default_factory=list)
+    _key_functions: list[KeyFunction] = field(init=False, default_factory=list)
 
     @classmethod
     def for_dataclass(cls, value: type[ItemT]) -> "InMemoryRepository[ItemT]":
         return cls(DataclassFormatter(value))
 
-    def __post_init__(self) -> None:
-        self._search_by = ["id", *self._search_by]
-
-    def with_searchable(self, attribute: str) -> Self:
-        self._search_by.append(attribute)
-
-        return self
-
-    def with_unique(self, criteria: Criteria) -> Self:
-        self._uniques.append(criteria)
+    def with_key(self, function: KeyFunction) -> Self:
+        self._key_functions.append(function)
 
         return self
 
@@ -62,18 +63,16 @@ class InMemoryRepository(Generic[ItemT]):
         for existing in self:
             error = ExistsError(existing)
 
-            for criteria in self._uniques:
-                if criteria(new) == criteria(existing):
-                    error.with_duplicate(criteria)
+            for key in self._key_functions:
+                if key(new) == key(existing):
+                    error.with_duplicate(key)
 
             error.fire()
 
-        assert str(new.id) not in self.items, f"Item with id<{new.id}> already exists"
-
     def read(self, item_id: Any) -> ItemT:
         for item in self:
-            for attribute in self._search_by:
-                if getattr(item, attribute) == item_id:
+            for key in self._key_functions:
+                if key(item) == str(item_id):
                     return item
 
         raise DoesNotExistError(item_id)

{apexdevkit-1.9.8 → apexdevkit-1.10.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "apexdevkit"
-version = "1.9.8"
+version = "1.10.1"
 description = "Apex Development Tools for python."
 authors = ["Apex Dev <dev@apex.ge>"]
 readme = "README.md"