v1.3.2-beta.4: revised forget node handler to check cache allowing deletion of proposed edges

Author: lukvmil

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "koi-net"
-version = "1.3.2-beta.3"
+version = "1.3.2-beta.4"
 description = "Implementation of KOI-net protocol in Python"
 authors = [
     {name = "Luke Miller", email = "luke@block.science"}

src/koi_net/components/knowledge_handlers/forget_node_handler.py

@@ -1,20 +1,18 @@
 from dataclasses import dataclass
 
-from rid_lib.types import KoiNetNode
+from rid_lib.types import KoiNetEdge, KoiNetNode
 
 from koi_net.protocol.edge import EdgeProfile
 from koi_net.protocol.knowledge_object import KnowledgeObject
 from koi_net.protocol.event import EventType
 from ..interfaces import KnowledgeHandler, HandlerType
 from ..kobj_queue import KobjQueue
-from ..graph import NetworkGraph
 from ..cache import Cache
 
 
 @dataclass
 class ForgetNodeHandler(KnowledgeHandler):
     cache: Cache
-    graph: NetworkGraph
     kobj_queue: KobjQueue
 
     handler_type = HandlerType.Final
@@ -26,14 +24,11 @@ def handle(self, kobj: KnowledgeObject):
         if kobj.normalized_event_type != EventType.FORGET:
             return
 
-        for edge in self.graph.dg.edges:
-            edge_rid = self.graph.dg.get_edge_data(*edge).get('rid')
-            if not edge_rid:
-                continue
-            
+        for edge_rid in self.cache.list_rids(rid_types=[KoiNetEdge]):
             edge_bundle = self.cache.read(edge_rid)
             if not edge_bundle:
                 continue
+            
             edge_profile = edge_bundle.validate_contents(EdgeProfile)
 
             if kobj.rid in (edge_profile.source, edge_profile.target):