alterware/aw-bot
1
0
Fork 0
mirror of https://github.com/alterware/aw-bot.git synced 2025-11-18 09:02:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: lock AI

Browse Source
This commit is contained in:
6arelyFuture
2025-06-10 22:00:01 +02:00
parent 63fb12dbba
commit 23e9529073
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
bot/events_handlers/message_events.py
View File

@@ -23,6 +23,11 @@ SPAM_ROLE_ID = 1350511935677927514
ADMIN_ROLE_ID = 1112364483915042908 ADMIN_ROLE_ID = 1112364483915042908
GROK_ROLE_ID = 1362837967919386916 GROK_ROLE_ID = 1362837967919386916
ALLOWED_CHANNELS = [
1112048063448617142, # off-topic
1119371841711112314, # vip-channel
]
# Cooldown: user_id -> [timestamps] # Cooldown: user_id -> [timestamps]
MENTION_COOLDOWNS = {} MENTION_COOLDOWNS = {}
@@ -45,6 +50,14 @@ async def handle_bot_mention(message, bot, no_context=False):
staff_role = message.guild.get_role(ADMIN_ROLE_ID) staff_role = message.guild.get_role(ADMIN_ROLE_ID)
member = message.guild.get_member(message.author.id) member = message.guild.get_member(message.author.id)
# Check if the message is in an allowed channel
if message.channel.id not in ALLOWED_CHANNELS:
await message.reply(
"The AI cannot used in this channel.",
mention_author=True,
)
return True
# Cooldown logic: max 1 use per minute per user # Cooldown logic: max 1 use per minute per user
now = time.time() now = time.time()
user_id = message.author.id user_id = message.author.id