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
|
import nextcord
|
||||||
from nextcord.ext import commands
|
from nextcord.ext import commands
|
||||||
from pytz import UTC
|
from pytz import UTC
|
||||||
from sqlmodel import Session, select
|
from sqlmodel import Session, col, select
|
||||||
|
|
||||||
from database.db import engine
|
from database.db import engine
|
||||||
from database.models import UserBirthday
|
from database.models import UserBirthday
|
||||||
@@ -83,8 +83,12 @@ class BirthdayCog(commands.Cog, name="Birthday Commands"):
|
|||||||
birthday_list.remove(interaction.user.id)
|
birthday_list.remove(interaction.user.id)
|
||||||
|
|
||||||
with Session(engine) as session:
|
with Session(engine) as session:
|
||||||
stmt = select(UserBirthday).where(UserBirthday.user_id == interaction.user.id)
|
stmt = select(UserBirthday).where(col(UserBirthday.user_id) == interaction.user.id)
|
||||||
session.delete(stmt)
|
if (user := session.scalar(stmt)) is None:
|
||||||
|
await interaction.response.send_message("No birthday to clear.", ephemeral=True)
|
||||||
|
return
|
||||||
|
|
||||||
|
session.delete(user)
|
||||||
session.commit()
|
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