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

[BUG] Replaced database conn names

Browse Source
This commit is contained in:
Paillat
2023-04-03 15:12:50 +02:00
parent 88a7ad2085
commit da21aa283f
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
code/cogs/setup.py
View File

@@ -158,8 +158,8 @@ class Setup(discord.Cog):
return return
# check if the guild is premium # check if the guild is premium
try: try:
cp.execute("SELECT premium FROM data WHERE guild_id = ?", (ctx.guild.id,)) con_premium.execute("SELECT premium FROM data WHERE guild_id = ?", (ctx.guild.id,))
premium = cp.fetchone()[0] premium = con_premium.fetchone()[0]
except: except:
premium = 0 premium = 0
if not premium: if not premium:
@@ -174,11 +174,11 @@ class Setup(discord.Cog):
"This channel is already set as the main channel", ephemeral=True "This channel is already set as the main channel", ephemeral=True
) )
return return
cp.execute("SELECT * FROM channels WHERE guild_id = ?", (ctx.guild.id,)) con_premium.execute("SELECT * FROM channels WHERE guild_id = ?", (ctx.guild.id,))
guild_channels = cp.fetchone() guild_channels = con_premium.fetchone()
if guild_channels is None: if guild_channels is None:
# if the channel is not in the list, add it # if the channel is not in the list, add it
cp.execute( con_premium.execute(
"INSERT INTO channels VALUES (?, ?, ?, ?, ?, ?)", "INSERT INTO channels VALUES (?, ?, ?, ?, ?, ?)",
(ctx.guild.id, channel.id, None, None, None, None), (ctx.guild.id, channel.id, None, None, None, None),
) )
@@ -191,7 +191,7 @@ class Setup(discord.Cog):
return return
for i in range(5): for i in range(5):
if channels[i] == None: if channels[i] == None:
cp.execute( con_premium.execute(
f"UPDATE channels SET channel{i} = ? WHERE guild_id = ?", f"UPDATE channels SET channel{i} = ? WHERE guild_id = ?",
(channel.id, ctx.guild.id), (channel.id, ctx.guild.id),
) )
@@ -224,8 +224,8 @@ class Setup(discord.Cog):
return return
# check if the guild is premium # check if the guild is premium
try: try:
cp.execute("SELECT premium FROM data WHERE guild_id = ?", (ctx.guild.id,)) con_premium.execute("SELECT premium FROM data WHERE guild_id = ?", (ctx.guild.id,))
premium = cp.fetchone()[0] premium = con_premium.fetchone()[0]
except: except:
premium = 0 premium = 0
if not premium: if not premium:
@@ -234,8 +234,8 @@ class Setup(discord.Cog):
if channel is None: if channel is None:
channel = ctx.channel channel = ctx.channel
# check if the channel is in the list # check if the channel is in the list
cp.execute("SELECT * FROM channels WHERE guild_id = ?", (ctx.guild.id,)) con_premium.execute("SELECT * FROM channels WHERE guild_id = ?", (ctx.guild.id,))
guild_channels = cp.fetchone() guild_channels = con_premium.fetchone()
curs_data.execute("SELECT channel_id FROM data WHERE guild_id = ?", (ctx.guild.id,)) curs_data.execute("SELECT channel_id FROM data WHERE guild_id = ?", (ctx.guild.id,))
if str(channel.id) == curs_data.fetchone()[0]: if str(channel.id) == curs_data.fetchone()[0]:
await ctx.respond( await ctx.respond(
@@ -257,7 +257,7 @@ class Setup(discord.Cog):
# remove the channel from the list # remove the channel from the list
for i in range(5): for i in range(5):
if channels[i] == str(channel.id): if channels[i] == str(channel.id):
cp.execute( con_premium.execute(
f"UPDATE channels SET channel{i} = ? WHERE guild_id = ?", f"UPDATE channels SET channel{i} = ? WHERE guild_id = ?",
(None, ctx.guild.id), (None, ctx.guild.id),
) )