Merge pull request #155 from onerandomusername/migrate-to-disnake

feat: migrate to disnake and dpy 2.0

Author: Akarys42

bot/bot.py

@@ -5,8 +5,8 @@
 import asyncpg
 from aiohttp import ClientSession
 from bot.postgres import create_tables
-from discord import AllowedMentions, Embed, Intents, Object
-from discord.ext import commands
+from disnake import AllowedMentions, Embed, Intents, Object
+from disnake.ext import commands
 from loguru import logger
 
 from . import constants
@@ -105,7 +105,9 @@ async def startup_greeting(self) -> None:
         """Announce presence to the devlog channel."""
         embed = Embed(description="Connected!")
         embed.set_author(
-            name="Gurkbot", url=constants.BOT_REPO_URL, icon_url=self.user.avatar_url
+            name="Gurkbot",
+            url=constants.BOT_REPO_URL,
+            icon_url=self.user.display_avatar.url,
         )
         await self.get_channel(constants.Channels.devlog).send(embed=embed)
 

bot/converters.py

@@ -1,4 +1,4 @@
-from discord.ext.commands import BadArgument, Context, Converter
+from disnake.ext.commands import BadArgument, Context, Converter
 
 
 class OffTopicName(Converter):

bot/exts/backend/error_handler.py

@@ -2,10 +2,10 @@
 import random
 from typing import Optional
 
-import discord
+import disnake
 from bot.constants import Channels, Colours, ERROR_REPLIES
-from discord import Embed, Message
-from discord.ext import commands
+from disnake import Embed, Message
+from disnake.ext import commands
 from loguru import logger
 
 
@@ -140,7 +140,7 @@ async def handle_unexpected_error(
             )
             try:
                 dev_alerts = await self.bot.fetch_channel(Channels.devalerts)
-            except discord.HTTPException as discord_exc:
+            except disnake.HTTPException as discord_exc:
                 logger.exception("Fetch failed", exc_info=discord_exc)
                 return
 

bot/exts/fun/bonker.py

@@ -4,9 +4,9 @@
 from io import BytesIO
 from typing import Dict
 
-import discord
+import disnake
 from PIL import Image, ImageDraw, ImageFile, ImageSequence
-from discord.ext import commands
+from disnake.ext import commands
 from loguru import logger
 
 ImageFile.LOAD_TRUNCATED_IMAGES = True
@@ -97,9 +97,9 @@ def _generate_gif(self, pfp: bytes) -> BytesIO:
 
     @commands.command()
     @commands.max_concurrency(3)
-    async def bonk(self, ctx: commands.Context, member: discord.User) -> None:
+    async def bonk(self, ctx: commands.Context, member: disnake.User) -> None:
         """Sends gif of mentioned member being "bonked" by Yoda."""
-        pfp = await member.avatar_url.read()
+        pfp = await member.display_avatar.read()
         created_at = ctx.message.created_at.strftime("%Y-%m-%d_%H-%M")
         out_filename = f"bonk_{member.id}_{created_at}.gif"
         func = functools.partial(self._generate_gif, pfp)
@@ -109,7 +109,7 @@ async def bonk(self, ctx: commands.Context, member: discord.User) -> None:
                 out_gif = await asyncio.get_running_loop().run_in_executor(pool, func)
 
             out_gif.seek(0)
-            await ctx.send(file=discord.File(out_gif, out_filename))
+            await ctx.send(file=disnake.File(out_gif, out_filename))
 
 
 def setup(bot: commands.Bot) -> None:

bot/exts/fun/ciphers.py

@@ -3,8 +3,8 @@
 
 from bot.bot import Bot
 from bot.constants import Colours
-from discord import Embed
-from discord.ext.commands import BadArgument, Cog, Context, group
+from disnake import Embed
+from disnake.ext.commands import BadArgument, Cog, Context, group
 
 logger = logging.getLogger(__name__)
 

bot/exts/fun/off_topic.py

@@ -8,9 +8,9 @@
 from bot.converters import OffTopicName as OT_Converter
 from bot.postgres.utils import db_execute, db_fetch
 from bot.utils.pagination import LinePaginator
-from discord import Embed, Reaction, TextChannel, User
-from discord.ext.commands import Cog, Context, group, has_any_role
-from discord.utils import sleep_until
+from disnake import Embed, Reaction, TextChannel, User
+from disnake.ext.commands import Cog, Context, group, has_any_role
+from disnake.utils import sleep_until
 from fuzzywuzzy import fuzz
 from loguru import logger
 

bot/exts/fun/xkcd.py

@@ -6,9 +6,9 @@
 
 from bot.bot import Bot
 from bot.constants import Colours
-from discord import Embed
-from discord.ext import tasks
-from discord.ext.commands import Cog, Context, command
+from disnake import Embed
+from disnake.ext import tasks
+from disnake.ext.commands import Cog, Context, command
 
 log = logging.getLogger(__name__)
 

bot/exts/github/_issues.py

@@ -1,11 +1,11 @@
 from random import choice
 from typing import Optional
 
-import discord
+import disnake
 from aiohttp import ClientSession
 from bot.constants import Channels, ERROR_REPLIES, Emojis
-from discord import Embed
-from discord.ext import commands
+from disnake import Embed
+from disnake.ext import commands
 from loguru import logger
 
 BAD_RESPONSE = {
@@ -28,23 +28,23 @@ def __init__(self, http_session: ClientSession) -> None:
         self.http_session = http_session
 
     @staticmethod
-    def get_repo(channel: discord.TextChannel) -> Optional[str]:
+    def get_repo(channel: disnake.TextChannel) -> Optional[str]:
         """Get repository for the particular channel."""
         return REPO_CHANNEL_MAP.get(channel.id, "gurkbot")
 
     @staticmethod
     def error_embed(error_msg: str) -> Embed:
         """Generate Error Embed for Issues command."""
-        embed = discord.Embed(
+        embed = disnake.Embed(
             title=choice(ERROR_REPLIES),
-            color=discord.Color.red(),
+            color=disnake.Color.red(),
             description=error_msg,
         )
         return embed
 
     async def issue(
         self,
-        channel: discord.TextChannel,
+        channel: disnake.TextChannel,
         numbers: commands.Greedy[int],
         repository: Optional[str],
         user: str,
@@ -104,8 +104,8 @@ async def issue(
                 )
             )
 
-        resp = discord.Embed(
-            colour=discord.Color.green(),
+        resp = disnake.Embed(
+            colour=disnake.Color.green(),
             description="\n".join("{0} [{1}]({2})".format(*link) for link in links),
         )
         resp.set_author(name="GitHub", url=f"https://github.com/{user}/{repository}")

bot/exts/github/_profile.py

@@ -2,10 +2,10 @@
 from random import choice
 from typing import Optional
 
-import discord
+import disnake
 from aiohttp import ClientSession
 from bot.constants import ERROR_REPLIES
-from discord import Embed
+from disnake import Embed
 
 
 class GithubInfo:
@@ -33,12 +33,12 @@ def get_data(username: Optional[str], user_data: dict, org_data: dict) -> Embed:
             else:
                 blog = "-"
 
-            embed = discord.Embed(
+            embed = disnake.Embed(
                 title=f"{user_data['login']}'s GitHub profile info",
                 description=f"```{user_data['bio']}```\n"
                 if user_data["bio"] is not None
                 else "",
-                colour=discord.Colour.green(),
+                colour=disnake.Colour.green(),
                 url=user_data["html_url"],
                 timestamp=datetime.strptime(
                     user_data["created_at"], "%Y-%m-%dT%H:%M:%SZ"
@@ -76,11 +76,11 @@ async def get_github_info(self, username: str) -> Embed:
 
         # User_data will not have a message key if the user exists
         if user_data.get("message") is not None:
-            embed = discord.Embed(
+            embed = disnake.Embed(
                 title=choice(ERROR_REPLIES),
                 description=f"The profile for `{username}` was not found.",
                 url=Embed.Empty,
-                colour=discord.Colour.red(),
+                colour=disnake.Colour.red(),
             )
             return embed
 

bot/exts/github/_source.py

@@ -3,10 +3,10 @@
 from textwrap import dedent
 from typing import Optional
 
-import discord
+import disnake
 from aiohttp import ClientSession
 from bot import constants
-from discord.ext.commands import Command
+from disnake.ext.commands import Command
 
 doc_reg_class = r'("""|\'\'\')([\s\S]*?)(\1\s*)'
 
@@ -15,13 +15,13 @@ class Source:
     """Displays information about the bot's source code."""
 
     def __init__(
-        self, http_session: ClientSession, bot_avatar: discord.asset.Asset
+        self, http_session: ClientSession, bot_avatar: disnake.asset.Asset
     ) -> None:
         self.http_session = http_session
         self.MAX_FIELD_LENGTH = 500
         self.bot_avatar = bot_avatar
 
-    async def inspect(self, cmd: Optional[Command]) -> Optional[discord.Embed]:
+    async def inspect(self, cmd: Optional[Command]) -> Optional[disnake.Embed]:
         """Display information and a GitHub link to the source code of a command."""
         if cmd is None:
             return
@@ -48,7 +48,7 @@ async def inspect(self, cmd: Optional[Command]) -> Optional[discord.Embed]:
                 + "\n... (truncated - too many lines)"
             )
 
-        embed = discord.Embed(title="Gurkbot's Source Link", description=f"{url}")
+        embed = disnake.Embed(title="Gurkbot's Source Link", description=f"{url}")
         embed.add_field(
             name="Source Code Snippet", value=f"```python\n{sanitized}\n```"
         )