velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

enums: make Intents an IntFlag subclass

This commit is contained in:
Luna 2021-07-11 01:21:03 -03:00
parent 49e1938ed6
commit 566920b120
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
litecord/enums.py
View File

@ -19,7 +19,7 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
import inspect import inspect
from typing import List, Any from typing import List, Any
from enum import Enum, IntEnum from enum import Enum, IntEnum, IntFlag
class EasyEnum(Enum): class EasyEnum(Enum):
@ -66,7 +66,7 @@ class Flags:
for attr, val in cls._attrs: for attr, val in cls._attrs:
has_attr = (value & val) == val has_attr = (value & val) == val
# set attributes dynamically # set attributes dynamically
setattr(res, f"is_{attr}", has_attr) setattr(res, f"is_{attr.lower()}", has_attr)
return res return res
@ -249,7 +249,7 @@ class Feature(EasyEnum):
news = "NEWS" news = "NEWS"
class Intents(Flags): class Intents(IntFlag):
GUILDS = 1 << 0 GUILDS = 1 << 0
GUILD_MEMBERS = 1 << 1 GUILD_MEMBERS = 1 << 1
GUILD_BANS = 1 << 2 GUILD_BANS = 1 << 2
@ -265,3 +265,7 @@ class Intents(Flags):
DIRECT_MESSAGES = 1 << 12 DIRECT_MESSAGES = 1 << 12
DIRECT_MESSAGE_REACTIONS = 1 << 13 DIRECT_MESSAGE_REACTIONS = 1 << 13
DIRECT_MESSAGE_TYPING = 1 << 14 DIRECT_MESSAGE_TYPING = 1 << 14
@classmethod
def default(cls):
return cls(-1)