diff --git a/litecord/storage.py b/litecord/storage.py index 71839dd..7d9542d 100644 --- a/litecord/storage.py +++ b/litecord/storage.py @@ -74,7 +74,7 @@ class Storage: # TODO: query less instead of popping when secure=True user_row = await self.db.fetchrow(""" SELECT id::text, username, discriminator, avatar, email, - flags, bot, mfa_enabled, verified, premium + flags, bot, mfa_enabled, verified, premium_since FROM users WHERE users.id = $1 """, user_id) @@ -84,6 +84,9 @@ class Storage: duser = dict(user_row) + duser['premium'] = duser['premium_since'] is not None + duser.pop('premium_since') + if not secure: duser.pop('email') duser.pop('verified')