From 0c7a520aa0368b9940ba3b0a2b1003456e25b04d Mon Sep 17 00:00:00 2001 From: Noah Date: Sat, 9 Jul 2022 14:14:06 +0100 Subject: refactor(client): Use REST for `Client.user`, add enums for flags and premium type --- discord/flags.py | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 discord/flags.py (limited to 'discord/flags.py') diff --git a/discord/flags.py b/discord/flags.py new file mode 100644 index 0000000..65a9253 --- /dev/null +++ b/discord/flags.py @@ -0,0 +1,35 @@ +from enum import Enum, unique + + +@unique +class Flags(Enum): + STAFF = 1 + PARTNER = 2 + HYPESQUAD = 4 + BUG_HUNTER_LEVEL_1 = 8 + HYPESQUAD_ONLINE_HOUSE_1 = 64 + HYPESQUAD_ONLINE_HOUSE_2 = 128 + HYPESQUAD_ONLINE_HOUSE_3 = 256 + TEAM_PSUEDO_USER = 1024 + BUG_HUNTER_LEVEL_2 = 16384 + VERIFIED_BOT = 65536 + VERIFIED_DEVELOPER = 131072 + CERTIFIED_MODERATOR = 262144 + BOT_HTTP_INTERACTIONS = 524288 + + +def get_number(flags: list[Flags]): + number = 1 + for i in flags: + number += i.value + return number + +def get_flags(number: int): + flags = [] + while number != 0: + for i in Flags: + if number >= i.value: + flags.append(i) + number -= i.value + return flags + -- cgit 1.4.1-2-gfad0