mirror of
https://github.com/yawaflua/Flask-Discord.git
synced 2025-12-10 04:19:31 +02:00
111 lines
3.8 KiB
Python
111 lines
3.8 KiB
Python
from flask import current_app, session
|
|
|
|
import requests
|
|
|
|
from .base import DiscordModelsBase
|
|
from .. import configs, exceptions
|
|
|
|
|
|
class User(DiscordModelsBase):
|
|
"""Class representing Discord User.
|
|
|
|
Attributes
|
|
----------
|
|
id : int
|
|
The discord ID of the user.
|
|
username : str
|
|
The discord username of the user.
|
|
discriminator : str
|
|
4 length string representing discord tag of the user.
|
|
avatar_hash : str
|
|
Hash of users avatar.
|
|
bot : bool
|
|
A boolean representing whether the user belongs to an OAuth2 application.
|
|
mfa_enabled : bool
|
|
A boolean representing whether the user has two factor enabled on their account.
|
|
locale : str
|
|
The user's chosen language option.
|
|
verified : bool
|
|
A boolean representing whether the email on this account has been verified.
|
|
email : str
|
|
User's email ID.
|
|
flags : int
|
|
An integer representing the
|
|
`user flags <https://discordapp.com/developers/docs/resources/user#user-object-user-flags>`_.
|
|
premium_type : int
|
|
An integer representing the
|
|
`type of nitro subscription <https://discordapp.com/developers/docs/resources/user#user-object-premium-types>`_.
|
|
|
|
"""
|
|
|
|
def __init__(self, payload):
|
|
self._payload = payload
|
|
self.id = int(self._payload["id"])
|
|
self.username = self._payload["username"]
|
|
self.discriminator = self._payload["discriminator"]
|
|
self.avatar_hash = self._payload.get("avatar", self.discriminator)
|
|
self.bot = self._payload.get("bot", False)
|
|
self.mfa_enabled = self._payload.get("mfa_enabled")
|
|
self.locale = self._payload.get("locale")
|
|
self.verified = self._payload.get("verified")
|
|
self.email = self._payload.get("email")
|
|
self.flags = self._payload.get("flags")
|
|
self.premium_type = self._payload.get("premium_type")
|
|
|
|
def __str__(self):
|
|
return f"{self.name}#{self.discriminator}"
|
|
|
|
@property
|
|
def name(self):
|
|
"""An alias to the username attribute."""
|
|
return self.username
|
|
|
|
@property
|
|
def avatar_url(self):
|
|
"""A property returning direct URL to user's avatar."""
|
|
image_format = configs.DISCORD_ANIMATED_IMAGE_FORMAT \
|
|
if self.is_avatar_animated else configs.DISCORD_IMAGE_FORMAT
|
|
return configs.DISCORD_USER_AVATAR_BASE_URL.format(
|
|
user_id=self.id, avatar_hash=self.avatar_hash, format=image_format)
|
|
|
|
@property
|
|
def is_avatar_animated(self):
|
|
"""A boolean representing if avatar of user is animated. Meaning user has GIF avatar."""
|
|
return self.avatar_hash.startswith("a_")
|
|
|
|
def add_to_guild(self, guild_id) -> dict:
|
|
"""Method to add user to the guild, provided OAuth2 session has already been created with ``guilds.join`` scope.
|
|
|
|
Parameters
|
|
----------
|
|
guild_id : int
|
|
The ID of the guild you want this user to be added.
|
|
|
|
Returns
|
|
-------
|
|
dict
|
|
A dict of guild member object. Returns an empty dict if user is already present in the guild.
|
|
|
|
Raises
|
|
------
|
|
flask_discord.Unauthorized
|
|
Raises :py:class:`flask_discord.Unauthorized` if current user is not authorized.
|
|
|
|
"""
|
|
route = configs.DISCORD_API_BASE_URL + f"/guilds/{guild_id}/members/{self.id}"
|
|
data = {"access_token": session["DISCORD_OAUTH2_TOKEN"]["access_token"]}
|
|
headers = {"Authorization": f"Bot {current_app.config['DISCORD_BOT_TOKEN']}"}
|
|
response = requests.put(route, json=data, headers=headers)
|
|
|
|
if response.status_code == 401:
|
|
raise exceptions.Unauthorized
|
|
|
|
if response.status_code == 204:
|
|
return dict()
|
|
|
|
return response.json()
|
|
|
|
|
|
class Bot(User):
|
|
"""Class representing the client user itself."""
|