john/daglib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

setitem

Browse Source
This commit is contained in:
John Lancaster
2026-02-21 00:22:52 -06:00
parent ba32d355e0
commit 9074d4b4a2
1 changed files with 19 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/daglib/dag.py
View File

@@ -38,13 +38,13 @@ class DAG(Generic[T], MutableMapping[T, MutableSet[T]]):
dagset = DAGSet(self._succ.get(u))
def on_add(v: T) -> None:
self._succ[u].add(v)
self._pred[v].add(u)
self._succ[u] |= {v}
self._pred[v] |= {u}
print(f"Adding edge {u} -> {v}")
def on_remove(v: T) -> None:
self._succ[u].remove(v)
self._pred[v].remove(u)
self._succ[u] -= {v}
self._pred[v] -= {u}
print(f"Removing edge {u} -> {v}")
dagset.on_add = on_add
@@ -52,9 +52,20 @@ class DAG(Generic[T], MutableMapping[T, MutableSet[T]]):
return dagset
def __setitem__(self, u: T, vs: Iterable[T]) -> None:
view = DAGSet(self, u)
view.clear()
view |= vs # uses our in-place operator
match vs:
case set():
self._succ[u] |= vs
for v in vs:
self._pred[v] |= {u}
case str():
self._succ[u] |= {vs}
self._pred[vs] |= {u}
case Iterable():
self._succ[u] |= set(vs)
for v in vs:
self._pred[v] |= {u}
case _:
raise TypeError(f"Expected set or iterable, got {type(vs).__name__}")
def __delitem__(self, u: T) -> None:
self.discard_node(u)