From 566920b120249e4b453c15746eba648a1a0327ba Mon Sep 17 00:00:00 2001 From: Luna Date: Sun, 11 Jul 2021 01:21:03 -0300 Subject: [PATCH] enums: make Intents an IntFlag subclass --- litecord/enums.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/litecord/enums.py b/litecord/enums.py index 2534ca2..dc20a9f 100644 --- a/litecord/enums.py +++ b/litecord/enums.py @@ -19,7 +19,7 @@ along with this program. If not, see . import inspect from typing import List, Any -from enum import Enum, IntEnum +from enum import Enum, IntEnum, IntFlag class EasyEnum(Enum): @@ -66,7 +66,7 @@ class Flags: for attr, val in cls._attrs: has_attr = (value & val) == val # set attributes dynamically - setattr(res, f"is_{attr}", has_attr) + setattr(res, f"is_{attr.lower()}", has_attr) return res @@ -249,7 +249,7 @@ class Feature(EasyEnum): news = "NEWS" -class Intents(Flags): +class Intents(IntFlag): GUILDS = 1 << 0 GUILD_MEMBERS = 1 << 1 GUILD_BANS = 1 << 2 @@ -265,3 +265,7 @@ class Intents(Flags): DIRECT_MESSAGES = 1 << 12 DIRECT_MESSAGE_REACTIONS = 1 << 13 DIRECT_MESSAGE_TYPING = 1 << 14 + + @classmethod + def default(cls): + return cls(-1)