feat: Add more image URL helpers (#699)

* fix: UnboundLocalError when server has no icon

* feat: implement Guild.banner_url

* feat: add discovery & invite splash banner URLs

* feat: add user banner URL

* feat: get member avatar for a specified guild

* fix: UnboundLocalError when server has no icon

* feat: implement Guild.banner_url

* feat: add discovery & invite splash banner URLs

* feat: add user banner URL

* feat: get member avatar for a specified guild

* ci: correct from checks.

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: mAxYoLo01

interactions/api/models/guild.py

@@ -1660,17 +1660,57 @@ async def get_all_members(self) -> List[Member]:
         return [Member(**_, _client=self._client) for _ in _all_members]
 
     @property
-    def icon_url(self) -> str:
+    def icon_url(self) -> Optional[str]:
         """
         Returns the URL of the guild's icon.
-        :return: URL of the guild's icon (None will be returned if none of icon is set)
+        :return: URL of the guild's icon (None will be returned if no icon is set)
         :rtype: str
         """
-        if self.icon is not None:
-            url = f"https://cdn.discordapp.com/icons/{int(self.id)}/{self.icon}"
-            url += ".gif" if self.icon.startswith("a_") else ".png"
+        if not self.icon:
+            return None
+
+        url = f"https://cdn.discordapp.com/icons/{int(self.id)}/{self.icon}"
+        url += ".gif" if self.icon.startswith("a_") else ".png"
+        return url
+
+    @property
+    def banner_url(self) -> Optional[str]:
+        """
+        Returns the URL of the guild's banner.
+        :return: URL of the guild's banner (None will be returned if no banner is set)
+        :rtype: str
+        """
+        if not self.banner:
+            return None
+
+        url = f"https://cdn.discordapp.com/banners/{int(self.id)}/{self.banner}"
+        url += ".gif" if self.banner.startswith("a_") else ".png"
         return url
 
+    @property
+    def splash_url(self) -> Optional[str]:
+        """
+        Returns the URL of the guild's invite splash banner.
+        :return: URL of the guild's invite splash banner (None will be returned if no banner is set)
+        :rtype: str
+        """
+        if not self.banner:
+            return None
+
+        return f"https://cdn.discordapp.com/splashes/{int(self.id)}/{self.splash}.png"
+
+    @property
+    def discovery_splash_url(self) -> Optional[str]:
+        """
+        Returns the URL of the guild's discovery splash banner.
+        :return: URL of the guild's discovery splash banner (None will be returned if no banner is set)
+        :rtype: str
+        """
+        if not self.banner:
+            return None
+
+        return f"https://cdn.discordapp.com/discovery-splashes/{int(self.id)}/{self.discovery_splash}.png"
+
 
 class GuildPreview(DictSerializerMixin):
     """

interactions/api/models/guild.pyi

@@ -414,7 +414,13 @@ class Guild(DictSerializerMixin):
     ) -> List[Member]: ...
     async def get_all_members(self) -> List[Member]: ...
     @property
-    def icon_url(self) -> str: ...
+    def icon_url(self) -> Optional[str]: ...
+    @property
+    def banner_url(self) -> Optional[str]: ...
+    @property
+    def splash_url(self) -> Optional[str]: ...
+    @property
+    def discovery_splash_url(self) -> Optional[str]: ...
 
 class GuildPreview(DictSerializerMixin):
     _json: dict

interactions/api/models/member.py

@@ -370,3 +370,18 @@ async def add_to_thread(
             user_id=int(self.user.id),
             thread_id=thread_id,
         )
+
+    def get_member_avatar_url(self, guild_id: int) -> Optional[str]:
+        """
+        Returns the URL of the member's avatar for the specified guild.
+        :param guild_id: The id of the guild to get the member's avatar from
+        :type guild_id: int
+        :return: URL of the members's avatar (None will be returned if no avatar is set)
+        :rtype: str
+        """
+        if not self.avatar:
+            return None
+
+        url = f"https://cdn.discordapp.com/guilds/{guild_id}/users/{int(self.user.id)}/avatars/{self.avatar}"
+        url += ".gif" if self.avatar.startswith("a_") else ".png"
+        return url

interactions/api/models/member.pyi

@@ -90,3 +90,4 @@ class Member(DictSerializerMixin):
         self,
         thread_id: int,
     ) -> None: ...
+    def get_member_avatar_url(self, guild_id: int) -> Optional[str]: ...

interactions/api/models/user.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 from .flags import UserFlags
 from .misc import DictSerializerMixin, Snowflake
 
@@ -88,3 +90,17 @@ def avatar_url(self) -> str:
         else:
             url += f"embed/avatars/{int(self.discriminator) % 5}.png"
         return url
+
+    @property
+    def banner_url(self) -> Optional[str]:
+        """
+        Returns the URL of the user's banner.
+        :return: URL of the user's banner (None will be returned if no banner is set)
+        :rtype: str
+        """
+        if not self.banner:
+            return None
+
+        url = f"https://cdn.discordapp.com/banners/{int(self.id)}/{self.banner}"
+        url += ".gif" if self.banner.startswith("a_") else ".png"
+        return url

interactions/api/models/user.pyi

@@ -27,3 +27,5 @@ class User(DictSerializerMixin):
     def mention(self) -> str: ...
     @property
     def avatar_url(self) -> str: ...
+    @property
+    def banner_url(self) -> Optional[str]: ...