SQLModel Migration

.env.example

@@ -10,6 +10,7 @@
 # Database URLs (Required: one depending on mode)
 # The application uses DEV_DATABASE_URL when run with '--dev' flag,
 # and PROD_DATABASE_URL otherwise (production mode).
+# PREFIX SHOULD BE postgresql+asyncpg://"
 DEV_DATABASE_URL=""
 PROD_DATABASE_URL=""
 

pyproject.toml

@@ -34,8 +34,7 @@ githubkit = { version = ">=0.12.0", extras = ["auth-app"] }
 httpx = ">=0.28.0"
 jishaku = ">=2.5.2"
 loguru = ">=0.7.2"
-pillow = ">=11.2.1,<11.3.0"
-prisma = ">=0.15.0"
+pillow = ">=11.3.0,<11.4.0"
 psutil = ">=6.0.0"
 pynacl = ">=1.5.0"
 python-dotenv = ">=1.0.1"
@@ -52,6 +51,9 @@ arrow = "^1.3.0"
 click = "^8.1.8"
 levenshtein = "^0.27.1"
 jinja2 = "^3.1.6"
+sqlmodel = "^0.0.24"
+asyncpg = "^0.30.0"
+alembic = "^1.16.2"
 
 [tool.poetry.group.dev.dependencies]
 pre-commit = "==4.2.0"

tests/unit/tux/utils/test_exceptions.py

@@ -5,7 +5,7 @@
 
 import pytest
 
-from prisma.models import Case
+from tux.database.schemas import Case
 from tux.utils.exceptions import (
     APIConnectionError,
     APIRequestError,

tux/cogs/moderation/__init__.py

@@ -8,9 +8,9 @@
 from discord.ext import commands
 from loguru import logger
 
-from prisma.enums import CaseType
 from tux.bot import Tux
 from tux.database.controllers import DatabaseController
+from tux.database.schemas import CaseType
 from tux.ui.embeds import EmbedCreator, EmbedType
 from tux.utils.constants import CONST
 from tux.utils.exceptions import handle_case_result, handle_gather_result

tux/cogs/moderation/ban.py

@@ -1,8 +1,8 @@
 import discord
 from discord.ext import commands
 
-from prisma.enums import CaseType
 from tux.bot import Tux
+from tux.database.schemas import CaseType
 from tux.utils import checks
 from tux.utils.flags import BanFlags
 from tux.utils.functions import generate_usage

tux/cogs/moderation/cases.py

@@ -5,10 +5,8 @@
 from loguru import logger
 from reactionmenu import ViewButton, ViewMenu
 
-from prisma.enums import CaseType
-from prisma.models import Case
-from prisma.types import CaseWhereInput
 from tux.bot import Tux
+from tux.database.schemas import Case, CaseType
 from tux.ui.embeds import EmbedCreator, EmbedType
 from tux.utils import checks
 from tux.utils.constants import CONST
@@ -246,7 +244,7 @@ async def _view_cases_with_flags(
         """
         assert ctx.guild
 
-        options: CaseWhereInput = {}
+        options: dict[str, Any] = {}
 
         if flags.type:
             options["case_type"] = flags.type

tux/cogs/moderation/jail.py

@@ -2,8 +2,8 @@
 from discord.ext import commands
 from loguru import logger
 
-from prisma.enums import CaseType
 from tux.bot import Tux
+from tux.database.schemas import CaseType
 from tux.utils import checks
 from tux.utils.flags import JailFlags
 from tux.utils.functions import generate_usage

tux/cogs/moderation/kick.py

@@ -1,8 +1,8 @@
 import discord
 from discord.ext import commands
 
-from prisma.enums import CaseType
 from tux.bot import Tux
+from tux.database.schemas import CaseType
 from tux.utils import checks
 from tux.utils.flags import KickFlags
 from tux.utils.functions import generate_usage

tux/cogs/moderation/pollban.py

@@ -1,8 +1,8 @@
 import discord
 from discord.ext import commands
 
-from prisma.enums import CaseType
 from tux.bot import Tux
+from tux.database.schemas import CaseType
 from tux.utils import checks
 from tux.utils.flags import PollBanFlags
 from tux.utils.functions import generate_usage

tux/cogs/moderation/pollunban.py

@@ -1,8 +1,8 @@
 import discord
 from discord.ext import commands
 
-from prisma.enums import CaseType
 from tux.bot import Tux
+from tux.database.schemas import CaseType
 from tux.utils import checks
 from tux.utils.flags import PollUnbanFlags
 from tux.utils.functions import generate_usage

tux/cogs/moderation/snippetban.py

@@ -1,8 +1,8 @@
 import discord
 from discord.ext import commands
 
-from prisma.enums import CaseType
 from tux.bot import Tux
+from tux.database.schemas import CaseType
 from tux.utils import checks
 from tux.utils.flags import SnippetBanFlags
 from tux.utils.functions import generate_usage

tux/cogs/moderation/snippetunban.py

@@ -1,8 +1,8 @@
 import discord
 from discord.ext import commands
 
-from prisma.enums import CaseType
 from tux.bot import Tux
+from tux.database.schemas import CaseType
 from tux.utils import checks
 from tux.utils.flags import SnippetUnbanFlags
 from tux.utils.functions import generate_usage

tux/cogs/moderation/tempban.py

@@ -4,9 +4,8 @@
 from discord.ext import commands, tasks
 from loguru import logger
 
-from prisma.enums import CaseType
-from prisma.models import Case
 from tux.bot import Tux
+from tux.database.schemas import Case, CaseType
 from tux.utils import checks
 from tux.utils.flags import TempBanFlags
 from tux.utils.functions import generate_usage

tux/cogs/moderation/timeout.py

@@ -3,8 +3,8 @@
 import discord
 from discord.ext import commands
 
-from prisma.enums import CaseType
 from tux.bot import Tux
+from tux.database.schemas import CaseType
 from tux.utils import checks
 from tux.utils.flags import TimeoutFlags
 from tux.utils.functions import generate_usage, parse_time_string

tux/cogs/moderation/unban.py

@@ -3,8 +3,8 @@
 import discord
 from discord.ext import commands
 
-from prisma.enums import CaseType
 from tux.bot import Tux
+from tux.database.schemas import CaseType
 from tux.utils import checks
 from tux.utils.constants import CONST
 from tux.utils.flags import UnbanFlags

tux/cogs/moderation/unjail.py

@@ -4,9 +4,8 @@
 from discord.ext import commands
 from loguru import logger
 
-from prisma.enums import CaseType
-from prisma.models import Case
 from tux.bot import Tux
+from tux.database.schemas import Case, CaseType
 from tux.utils import checks
 from tux.utils.flags import UnjailFlags
 from tux.utils.functions import generate_usage

tux/cogs/moderation/untimeout.py

@@ -1,8 +1,8 @@
 import discord
 from discord.ext import commands
 
-from prisma.enums import CaseType
 from tux.bot import Tux
+from tux.database.schemas import CaseType
 from tux.utils import checks
 from tux.utils.flags import UntimeoutFlags
 from tux.utils.functions import generate_usage