aboutsummaryrefslogtreecommitdiff
path: root/discord/user.py
diff options
context:
space:
mode:
authorLiam H <[email protected]>2019-01-23 17:45:35 -0600
committerRapptz <[email protected]>2019-02-12 19:01:10 -0500
commit79f172cf809ee7b51ee9bce379a4c7531e8d1b4e (patch)
tree5fdb84e98785b1b6228c2f1f5ab1d3a7b515a4e6 /discord/user.py
parentChange Activity.application_id from `str` to `int`. (diff)
downloaddiscord.py-79f172cf809ee7b51ee9bce379a4c7531e8d1b4e.tar.xz
discord.py-79f172cf809ee7b51ee9bce379a4c7531e8d1b4e.zip
Add PremiumType enumeration and ClientUser.premium_type
Diffstat (limited to 'discord/user.py')
-rw-r--r--discord/user.py7
1 files changed, 5 insertions, 2 deletions
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):