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

Bitmask: add __eq__ method

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

11
bitmask.py
View File

@ -120,6 +120,15 @@ class Bitmask:
return f"{_fullname(self)}({args})" return f"{_fullname(self)}({args})"
def __eq__(self, other):
"""Check equality."""
if not issubclass(type(other), type(self)):
return False
elif not issubclass(other.AllFlags, self.AllFlags):
return False
else:
return other.value == self.value
def _flag_op(self, flag, op): def _flag_op(self, flag, op):
"""Apply a single flag to the bitmask. """Apply a single flag to the bitmask.
@ -147,7 +156,7 @@ class Bitmask:
for flag in other: for flag in other:
new_bitmask._flag_op(flag, op) new_bitmask._flag_op(flag, op)
elif issubclass(type(other), self.AllFlags): elif issubclass(type(other), self.AllFlags):
new_bitmask._flag_op(flag, op) new_bitmask._flag_op(other, op)
else: else:
raise TypeError(f"can only apply {type(self)} or {self.AllFlags} (not '{type(other)}') to {type(self)}") raise TypeError(f"can only apply {type(self)} or {self.AllFlags} (not '{type(other)}') to {type(self)}")