diff --git a/cogs/commands/birthday.py b/cogs/commands/birthday.py index 0ad3204..a7fa2d4 100644 --- a/cogs/commands/birthday.py +++ b/cogs/commands/birthday.py @@ -5,7 +5,7 @@ from datetime import datetime import nextcord from nextcord.ext import commands from pytz import UTC -from sqlmodel import Session, select +from sqlmodel import Session, col, select from database.db import engine from database.models import UserBirthday @@ -83,8 +83,12 @@ class BirthdayCog(commands.Cog, name="Birthday Commands"): birthday_list.remove(interaction.user.id) with Session(engine) as session: - stmt = select(UserBirthday).where(UserBirthday.user_id == interaction.user.id) - session.delete(stmt) + stmt = select(UserBirthday).where(col(UserBirthday.user_id) == interaction.user.id) + if (user := session.scalar(stmt)) is None: + await interaction.response.send_message("No birthday to clear.", ephemeral=True) + return + + session.delete(user) session.commit() - await interaction.followup.send("Birthday cleared.", ephemeral=True) + await interaction.response.send_message("Birthday cleared.", ephemeral=True)