Initialize database stuff & setup didier

didier/__init__.py

@@ -0,0 +1 @@
+from .didier import Didier

didier/data/constants.py

@@ -0,0 +1 @@
+PREFIXES = ["didier", "big d"]

didier/didier.py

@@ -0,0 +1,36 @@
+import discord
+from discord import Message
+from discord.ext import commands
+from sqlalchemy.ext.asyncio import AsyncSession
+
+import settings
+from database.engine import DBSession
+from didier.utils.prefix import get_prefix
+
+
+class Didier(commands.Bot):
+    """DIDIER <3"""
+
+    def __init__(self):
+        activity = discord.Activity(type=discord.ActivityType.playing, name=settings.DISCORD_STATUS_MESSAGE)
+        status = discord.Status.online
+
+        intents = discord.Intents.default()
+        intents.members = True
+        intents.message_content = True
+
+        super().__init__(
+            command_prefix=get_prefix, case_insensitive=True, intents=intents, activity=activity, status=status
+        )
+
+    @property
+    def db_session(self) -> AsyncSession:
+        """Obtain a database session"""
+        return DBSession()
+
+    async def on_ready(self):
+        """Event triggered when the bot is ready"""
+        print(settings.DISCORD_READY_MESSAGE)
+
+    async def on_message(self, message: Message, /) -> None:
+        print(message.content)

didier/utils/prefix.py

@@ -0,0 +1,27 @@
+import re
+
+from discord import Message
+from discord.ext import commands
+
+from didier.data import constants
+
+
+def get_prefix(client: commands.Bot, message: Message) -> str:
+    """Match a prefix against a message
+    This is done dynamically to allow variable amounts of whitespace,
+    and through regexes to allow case-insensitivity among other things.
+    """
+    mention = f"<@!{client.user.id}>"
+    regex = r"^({})\s*"
+
+    # Check which prefix was used
+    for prefix in [*constants.PREFIXES, mention]:
+        match = re.match(regex.format(prefix), message.content, flags=re.I)
+
+        if match is not None:
+            # Get the part of the message that was matched
+            # .group() is inconsistent with whitespace, so that can't be used
+            return message.content[: match.end()]
+
+    # Matched nothing
+    return "didier"