ic-python-db 0.9.0__tar.gz → 0.9.1__tar.gz

{ic_python_db-0.9.0/ic_python_db.egg-info → ic_python_db-0.9.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ic-python-db
-Version: 0.9.0
+Version: 0.9.1
 Summary: A lightweight key-value database written in Python, intended for use on the Internet Computer (IC)
 Home-page: https://github.com/smart-social-contracts/ic-python-db
 Author: Smart Social Contracts

{ic_python_db-0.9.0 → ic_python_db-0.9.1}/ic_python_db/__init__.py

@@ -20,7 +20,7 @@ from .schema import SchemaIncompatibleError, build_schema, diff_schemas, schema_
 from .storage import MemoryStorage, Storage
 from .system_time import SystemTime
 
-__version__ = "0.9.0"
+__version__ = "0.9.1"
 __all__ = [
     "Database",
     "Entity",

{ic_python_db-0.9.0 → ic_python_db-0.9.1}/ic_python_db/properties.py

@@ -526,6 +526,78 @@ class OneToOne(Relation[E]):
             obj._save()
 
 
+class ManyToManyList(list):
+    """List subclass returned by ManyToMany.__get__ that supports .add() and .remove()."""
+
+    def __init__(self, items, owner, prop):
+        super().__init__(items)
+        self._owner = owner
+        self._prop = prop
+
+    def add(self, entity):
+        """Add an entity to this many-to-many relationship."""
+        from .db_engine import Database
+        from .entity import Entity
+
+        if isinstance(entity, (str, int)):
+            resolved = self._prop.resolve_entity(self._owner, entity)
+        elif isinstance(entity, Entity):
+            resolved = entity
+        else:
+            raise TypeError(f"Cannot add {type(entity)} to ManyToMany relation")
+
+        self._prop.validate_entity(resolved)
+        db = Database.get_instance()
+
+        existing = db.reverse_index_get(
+            self._owner._type, self._owner._id, self._prop.name
+        )
+        if resolved._id not in existing:
+            db.reverse_index_add(
+                self._owner._type, self._owner._id, self._prop.name, resolved._id
+            )
+            db.reverse_index_add(
+                resolved._type, resolved._id, self._prop.reverse_name, self._owner._id
+            )
+            if not self._owner._do_not_save:
+                self._owner._save()
+        self.append(resolved)
+
+    def discard(self, entity):
+        """Remove an entity from this many-to-many relationship (no error if absent)."""
+        from .db_engine import Database
+        from .entity import Entity
+
+        if isinstance(entity, Entity):
+            entity_id = entity._id
+            entity_type = entity._type
+        else:
+            entity_id = str(entity)
+            entity_type = None
+
+        db = Database.get_instance()
+        db.reverse_index_remove(
+            self._owner._type, self._owner._id, self._prop.name, entity_id
+        )
+        if entity_type:
+            db.reverse_index_remove(
+                entity_type, entity_id, self._prop.reverse_name, self._owner._id
+            )
+        else:
+            for type_name in self._prop._get_allowed_types():
+                db.reverse_index_remove(
+                    type_name, entity_id, self._prop.reverse_name, self._owner._id
+                )
+
+        self[:] = [e for e in self if getattr(e, "_id", None) != entity_id]
+        if not self._owner._do_not_save:
+            self._owner._save()
+
+    def remove(self, entity):
+        """Remove an entity from this many-to-many relationship."""
+        self.discard(entity)
+
+
 class ManyToMany(Relation[E]):
     """Many-to-many relationship with bidirectional reverse indexes.
 
@@ -562,7 +634,7 @@ class ManyToMany(Relation[E]):
                     if entity:
                         entities.append(entity)
                         break
-        return entities  # type: ignore[return-value]
+        return ManyToManyList(entities, obj, self)  # type: ignore[return-value]
 
     def __set__(self, obj, values):
         from .db_engine import Database

{ic_python_db-0.9.0 → ic_python_db-0.9.1/ic_python_db.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ic-python-db
-Version: 0.9.0
+Version: 0.9.1
 Summary: A lightweight key-value database written in Python, intended for use on the Internet Computer (IC)
 Home-page: https://github.com/smart-social-contracts/ic-python-db
 Author: Smart Social Contracts

{ic_python_db-0.9.0 → ic_python_db-0.9.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ic-python-db"
-version = "0.9.0"
+version = "0.9.1"
 description = "A lightweight key-value database written in Python, intended for use on the Internet Computer (IC)"
 readme = "README.md"
 authors = [

{ic_python_db-0.9.0 → ic_python_db-0.9.1}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="ic-python-db",
-    version="0.9.0",
+    version="0.9.1",
     author="Smart Social Contracts",
     author_email="smartsocialcontracts@gmail.com",
     description="A lightweight key-value database with entity relationships and audit logging",