diff options
| author | Liam H <[email protected]> | 2019-01-23 17:45:35 -0600 |
|---|---|---|
| committer | Rapptz <[email protected]> | 2019-02-12 19:01:10 -0500 |
| commit | 79f172cf809ee7b51ee9bce379a4c7531e8d1b4e (patch) | |
| tree | 5fdb84e98785b1b6228c2f1f5ab1d3a7b515a4e6 | |
| parent | Change Activity.application_id from `str` to `int`. (diff) | |
| download | discord.py-79f172cf809ee7b51ee9bce379a4c7531e8d1b4e.tar.xz discord.py-79f172cf809ee7b51ee9bce379a4c7531e8d1b4e.zip | |
Add PremiumType enumeration and ClientUser.premium_type
| -rw-r--r-- | discord/enums.py | 7 | ||||
| -rw-r--r-- | discord/user.py | 7 |
2 files changed, 11 insertions, 3 deletions
diff --git a/discord/enums.py b/discord/enums.py index f11f1ed1..9e6ec3e6 100644 --- a/discord/enums.py +++ b/discord/enums.py @@ -29,7 +29,8 @@ from enum import Enum, IntEnum __all__ = ['ChannelType', 'MessageType', 'VoiceRegion', 'SpeakingState', 'VerificationLevel', 'ContentFilter', 'Status', 'DefaultAvatar', 'RelationshipType', 'AuditLogAction', 'AuditLogActionCategory', - 'UserFlags', 'ActivityType', 'HypeSquadHouse', 'NotificationLevel'] + 'UserFlags', 'ActivityType', 'HypeSquadHouse', 'NotificationLevel', + 'PremiumType'] class ChannelType(Enum): text = 0 @@ -245,6 +246,10 @@ class HypeSquadHouse(Enum): brilliance = 2 balance = 3 +class PremiumType(Enum): + nitro_classic = 1 + nitro = 2 + def try_enum(cls, val): """A function that tries to turn the value into enum ``cls``. diff --git a/discord/user.py b/discord/user.py index 78ed0491..900e632d 100644 --- a/discord/user.py +++ b/discord/user.py @@ -28,7 +28,7 @@ from collections import namedtuple import discord.abc from .utils import snowflake_time, _bytes_to_base64_data, parse_time, valid_icon_size -from .enums import DefaultAvatar, RelationshipType, UserFlags, HypeSquadHouse +from .enums import DefaultAvatar, RelationshipType, UserFlags, HypeSquadHouse, PremiumType, try_enum from .errors import ClientException, InvalidArgument from .colour import Colour @@ -299,8 +299,10 @@ class ClientUser(BaseUser): Specifies if the user has MFA turned on and working. premium: :class:`bool` Specifies if the user is a premium user (e.g. has Discord Nitro). + premium_type: :class:`PremiumType` + Specifies the type of premium a user has (e.g. Nitro or Nitro Classic). Could be None if the user is not premium. """ - __slots__ = ('email', 'verified', 'mfa_enabled', 'premium', '_relationships') + __slots__ = ('email', 'verified', 'mfa_enabled', 'premium', 'premium_type', '_relationships') def __init__(self, *, state, data): super().__init__(state=state, data=data) @@ -308,6 +310,7 @@ class ClientUser(BaseUser): self.email = data.get('email') self.mfa_enabled = data.get('mfa_enabled', False) self.premium = data.get('premium', False) + self.premium_type = try_enum(PremiumType, data.get('premium_type', None)) self._relationships = {} def __repr__(self): |