Archived
generated from imcompiling/discord-bot-template
Fix birthday clear.
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user