diff --git a/flask_discord/configs.py b/flask_discord/configs.py index df5ca91..3fcb0d3 100644 --- a/flask_discord/configs.py +++ b/flask_discord/configs.py @@ -17,3 +17,4 @@ DEFAULT_SCOPES = [ IMAGE_BASE_URL = "https://cdn.discordapp.com/" USER_AVATAR_BASE_URL = IMAGE_BASE_URL + "avatars/{user_id}/{avatar_hash}.png" +GUILD_ICON_BASE_URL = IMAGE_BASE_URL + "icons/{guild_id}/{icon_hash}.png" diff --git a/flask_discord/models/guild.py b/flask_discord/models/guild.py new file mode 100644 index 0000000..3a8e76b --- /dev/null +++ b/flask_discord/models/guild.py @@ -0,0 +1,19 @@ +from .. import configs + + +class Guild(object): + + def __init__(self, payload): + self._payload = payload + self.id = self._payload["id"] + self.name = self._payload["name"] + self.icon_hash = self._payload.get("icon") + self.is_owner = self._payload.get["owner"] + self.permissions_value = self._payload.get("permissions") + + def __str__(self): + return self.name + + @property + def icon_url(self): + return configs.GUILD_ICON_BASE_URL.format(guild_id=self.id, icon_hash=self.icon_hash)