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

added conn.commit

Browse Source
This commit is contained in:
Paillat
2022-12-10 23:50:17 +01:00
parent 222e3c5c80
commit aba130224c
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
code/cogs/chat.py
View File

@@ -2,7 +2,7 @@ import discord
import re import re
import asyncio import asyncio
import openai import openai
from config import debug, c, max_uses, cp from config import debug, c, max_uses, cp, conn, connp
import random import random
class Chat (discord.Cog) : class Chat (discord.Cog) :
def __init__(self, bot: discord.Bot): def __init__(self, bot: discord.Bot):
@@ -39,15 +39,19 @@ class Chat (discord.Cog) :
return return
#check if the bot hasn't been used more than 5000 times in the last 24 hours (uses_count_today) #check if the bot hasn't been used more than 5000 times in the last 24 hours (uses_count_today)
c.execute("SELECT uses_count_today FROM data WHERE guild_id = ?", (message.guild.id,)) c.execute("SELECT uses_count_today FROM data WHERE guild_id = ?", (message.guild.id,))
if c.fetchone()[0] >= 500: uses = c.fetchone()[0]
cp.execute("SELECT premium FROM data WHERE user_id = ? AND guild_id = ?", (message.author.id, message.guild.id))
premium = cp.fetchone()
if c.fetchone()[0] >= 500 and premium != 1:
debug(f"The bot has been used more than {max_uses} times in the last 24 hours in this guild. Please try again in 24h.") debug(f"The bot has been used more than {max_uses} times in the last 24 hours in this guild. Please try again in 24h.")
await message.channel.send("The bot has been used more than 5000 times in the last 24 hours in this guild. Please try again in 24h.") await message.channel.send("The bot has been used more than 500 times in the last 24 hours in this guild. Please try again in 24h.")
return return
#add 1 to the uses_count_today #add 1 to the uses_count_today
#show that the bot is typing #show that the bot is typing
await message.channel.trigger_typing() await message.channel.trigger_typing()
if message.guild.id != 1021872219888033903: if message.guild.id != 1021872219888033903:
c.execute("UPDATE data SET uses_count_today = uses_count_today + 1 WHERE guild_id = ?", (message.guild.id,)) c.execute("UPDATE data SET uses_count_today = uses_count_today + 1 WHERE guild_id = ?", (message.guild.id,))
conn.commit()
#get the api key from the database #get the api key from the database
c.execute("SELECT api_key FROM data WHERE guild_id = ?", (message.guild.id,)) c.execute("SELECT api_key FROM data WHERE guild_id = ?", (message.guild.id,))
api_key = c.fetchone()[0] api_key = c.fetchone()[0]