dogeystamp/bitmask
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

bitmask.py: clarify logic on discard, remove

Browse Source
This commit is contained in:
dogeystamp 2022-07-27 22:07:15 -04:00
parent 0e67984d99
commit fe2bfd9356
Signed by: dogeystamp
GPG Key ID: 7225FE3592EFFA38
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
bitmask.py
View File

@ -173,6 +173,13 @@ class Bitmask:
def __radd__(self, other): def __radd__(self, other):
return self.__add__(other) return self.__add__(other)
def discard(self, flag):
"""Remove flag bitmask if present.
This behaves the same as built-in `set.discard()`.
"""
self._flag_op(flag, lambda a, b : a & ~b)
def remove(self, flag): def remove(self, flag):
"""Remove `flag` from the bitmask. """Remove `flag` from the bitmask.
@ -183,12 +190,4 @@ class Bitmask:
""" """
if not flag in self: if not flag in self:
raise KeyError(flag) raise KeyError(flag)
self._flag_op(flag, lambda a, b : a ^ b) self.discard(flag)
def discard(self, flag):
"""Remove flag bitmask if present.
This behaves the same as built-in `set.discard()`.
"""
if flag in self:
self.remove(flag)