Paillat-dev/Botator
1
0
Fork 0
mirror of https://github.com/Paillat-dev/Botator.git synced 2026-01-02 01:06:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update code.py

Browse Source
This commit is contained in:
Paillat
2022-12-08 12:37:21 +01:00
parent ec99b8c920
commit 7b2029a40c
1 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
code/code.py
View File

@@ -291,15 +291,16 @@ async def on_message(message: discord.Message):
return return
#check if the message has been sent in the channel set in the database #check if the message has been sent in the channel set in the database
c.execute("SELECT channel_id FROM data WHERE guild_id = ?", (message.guild.id,)) c.execute("SELECT channel_id FROM data WHERE guild_id = ?", (message.guild.id,))
try : replied_message: discord.Message = await message.to_reference() try : original_message = await message.channel.fetch_message(message.reference.message_id)
except : replied_message = None except : original_message = None
else : if original_message != None and original_message.author.id != bot.user.id:
if replied_message.author.id != bot.user.id: original_message = None
replied_message = None
if str(message.channel.id) != str(c.fetchone()[0]): if str(message.channel.id) != str(c.fetchone()[0]):
#check if the message is a mention or if the message replies to the bot #check if the message is a mention or if the message replies to the bot
if message.content.find("<@1046051875755134996>") != -1 or replied_message != None: if original_message != None:
debug("wrong channel, but mention or reply") debug("wrong channel, but reply")
elif message.content.find("<@"+str(bot.user.id)+">") != -1:
debug("wrong channel, but mention")
else : else :
debug("The message has been sent in the wrong channel") debug("The message has been sent in the wrong channel")
return return
@@ -320,13 +321,13 @@ async def on_message(message: discord.Message):
#get the advanced settings from the database #get the advanced settings from the database
c.execute("SELECT max_tokens, temperature, frequency_penalty, presence_penalty, prompt_size FROM data WHERE guild_id = ?", (message.guild.id,)) c.execute("SELECT max_tokens, temperature, frequency_penalty, presence_penalty, prompt_size FROM data WHERE guild_id = ?", (message.guild.id,))
max_tokens, temperature, frequency_penalty, presence_penalty, prompt_size = c.fetchone() max_tokens, temperature, frequency_penalty, presence_penalty, prompt_size = c.fetchone()
if replied_message == None: if original_message == None:
messages = await message.channel.history(limit=prompt_size).flatten() messages = await message.channel.history(limit=prompt_size).flatten()
messages.reverse() messages.reverse()
else : else :
messages = await message.channel.history(limit=prompt_size, before=replied_message.created_at).flatten() messages = await message.channel.history(limit=prompt_size, before=original_message).flatten()
messages.reverse() messages.reverse()
messages.append(replied_message) messages.append(original_message)
messages.append(message) messages.append(message)
prompt = "" prompt = ""
#get the channel id from the database #get the channel id from the database
@@ -528,4 +529,5 @@ bot.loop.create_task(check_day_task())
# Replace the following with your bot's token # Replace the following with your bot's token
with open("key.txt") as f: with open("key.txt") as f:
key = f.read() key = f.read()
bot.run(key) bot.run(key)