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

other i functions

Browse Source
This commit is contained in:
John Lancaster
2026-02-21 00:22:39 -06:00
parent bfdd295ac5
commit ba32d355e0
1 changed files with 34 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
src/daglib/set.py
View File

@@ -50,6 +50,7 @@ class DAGSet(MutableSet[T]):
self.on_add(v)
def discard(self, v: T) -> None:
if v in self._data:
self._data.discard(v)
if self.on_remove:
self.on_remove(v)
@@ -69,22 +70,38 @@ class DAGSet(MutableSet[T]):
self |= {other}
return self
# def __isub__(self, other: Iterable[T]) -> DAGSet[T]:
# # a -= b => remove those in other
# match other:
# case set():
# self -= other
# case _:
# self -= set(other)
# return self
def __isub__(self, other: Iterable[T]) -> DAGSet[T]:
# a -= b => remove those in other
match other:
case str():
other = {other}
case Iterable():
other = set(other)
case _:
other = {other}
return super().__isub__(other)
# def __iand__(self, other: Iterable[T]) -> DAGSet[T]:
# # a &= b => keep only those also in other
# return self
def __iand__(self, other: Iterable[T]) -> DAGSet[T]:
# a &= b => keep only those also in other
match other:
case str():
other = {other}
case Iterable():
other = set(other)
case _:
other = {other}
return super().__iand__(other)
# def __ixor__(self, other: Iterable[T]) -> DAGSet[T]:
# # a ^= b => symmetric difference update
# return self
def __ixor__(self, other: Iterable[T]) -> DAGSet[T]:
# a ^= b => symmetric difference update
match other:
case str():
other = {other}
case Iterable():
other = set(other)
case _:
other = {other}
return super().__ixor__(other)
# --- nice repr for debugging ---