mirror of
https://github.com/Paillat-dev/Botator.git
synced 2026-01-02 09:16:19 +00:00
[GENERAL] Code reformatting using Black standard
This commit is contained in:
Alexis LEBEL
@@ -1,7 +1,8 @@
|
||||
import discord
|
||||
from config import debug, conn, c, connp, cp
|
||||
|
||||
class Setup (discord.Cog) :
|
||||
|
||||
class Setup(discord.Cog):
|
||||
def __init__(self, bot: discord.Bot):
|
||||
super().__init__()
|
||||
self.bot = bot
|
||||
@@ -9,111 +10,178 @@ class Setup (discord.Cog) :
|
||||
@discord.slash_command(name="setup", description="Setup the bot")
|
||||
@discord.option(name="channel_id", description="The channel id", required=True)
|
||||
@discord.option(name="api_key", description="The api key", required=True)
|
||||
async def setup(self, ctx: discord.ApplicationContext, channel: discord.TextChannel, api_key: str):
|
||||
#check if the api key is valid
|
||||
debug(f"The user {ctx.author} ran the setup command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}")
|
||||
#check if the channel is valid
|
||||
async def setup(
|
||||
self,
|
||||
ctx: discord.ApplicationContext,
|
||||
channel: discord.TextChannel,
|
||||
api_key: str,
|
||||
):
|
||||
# check if the api key is valid
|
||||
debug(
|
||||
f"The user {ctx.author} ran the setup command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}"
|
||||
)
|
||||
# check if the channel is valid
|
||||
if channel is None:
|
||||
await ctx.respond("Invalid channel id", ephemeral=True)
|
||||
return
|
||||
#check if the guild is already in the database bi checking if there is a key for the guild
|
||||
try:
|
||||
# check if the guild is already in the database bi checking if there is a key for the guild
|
||||
try:
|
||||
c.execute("SELECT * FROM data WHERE guild_id = ?", (ctx.guild.id,))
|
||||
data = c.fetchone()
|
||||
if data[3] == None: data = None
|
||||
if data[3] == None:
|
||||
data = None
|
||||
except:
|
||||
data = None
|
||||
|
||||
if data != None:
|
||||
c.execute("UPDATE data SET channel_id = ?, api_key = ? WHERE guild_id = ?", (channel.id, api_key, ctx.guild.id))
|
||||
# c.execute("UPDATE data SET is_active = ?, max_tokens = ?, temperature = ?, frequency_penalty = ?, presence_penalty = ?, prompt_size = ? WHERE guild_id = ?", (False, 64, 0.9, 0.0, 0.0, 5, ctx.guild.id))
|
||||
conn.commit()
|
||||
await ctx.respond("The channel id and the api key have been updated", ephemeral=True)
|
||||
else:
|
||||
#in this case, the guild is not in the database, so we add it
|
||||
c.execute("INSERT INTO data VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", (ctx.guild.id, channel.id, api_key, False, 64, 0.9, 0.0, 0.0, 0, 5, "", False, "", False))
|
||||
conn.commit()
|
||||
await ctx.respond("The channel id and the api key have been added", ephemeral=True)
|
||||
|
||||
@discord.slash_command(name="delete", description="Delete the information about this server")
|
||||
if data != None:
|
||||
c.execute(
|
||||
"UPDATE data SET channel_id = ?, api_key = ? WHERE guild_id = ?",
|
||||
(channel.id, api_key, ctx.guild.id),
|
||||
)
|
||||
# c.execute("UPDATE data SET is_active = ?, max_tokens = ?, temperature = ?, frequency_penalty = ?, presence_penalty = ?, prompt_size = ? WHERE guild_id = ?", (False, 64, 0.9, 0.0, 0.0, 5, ctx.guild.id))
|
||||
conn.commit()
|
||||
await ctx.respond(
|
||||
"The channel id and the api key have been updated", ephemeral=True
|
||||
)
|
||||
else:
|
||||
# in this case, the guild is not in the database, so we add it
|
||||
c.execute(
|
||||
"INSERT INTO data VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
|
||||
(
|
||||
ctx.guild.id,
|
||||
channel.id,
|
||||
api_key,
|
||||
False,
|
||||
64,
|
||||
0.9,
|
||||
0.0,
|
||||
0.0,
|
||||
0,
|
||||
5,
|
||||
"",
|
||||
False,
|
||||
"",
|
||||
False,
|
||||
),
|
||||
)
|
||||
conn.commit()
|
||||
await ctx.respond(
|
||||
"The channel id and the api key have been added", ephemeral=True
|
||||
)
|
||||
|
||||
@discord.slash_command(
|
||||
name="delete", description="Delete the information about this server"
|
||||
)
|
||||
##@discord.commands.permissions(administrator=True)
|
||||
async def delete(self, ctx: discord.ApplicationContext):
|
||||
debug(f"The user {ctx.author} ran the delete command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}")
|
||||
#check if the guild is in the database
|
||||
debug(
|
||||
f"The user {ctx.author} ran the delete command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}"
|
||||
)
|
||||
# check if the guild is in the database
|
||||
c.execute("SELECT * FROM data WHERE guild_id = ?", (ctx.guild.id,))
|
||||
if c.fetchone() is None:
|
||||
await ctx.respond("This server is not setup", ephemeral=True)
|
||||
return
|
||||
#delete the guild from the database, except the guild id and the uses_count_today
|
||||
c.execute("UPDATE data SET api_key = ?, channel_id = ?, is_active = ?, max_tokens = ?, temperature = ?, frequency_penalty = ?, presence_penalty = ?, prompt_size = ? WHERE guild_id = ?", (None, None, False, 50, 0.9, 0.0, 0.0, 0, ctx.guild.id))
|
||||
# delete the guild from the database, except the guild id and the uses_count_today
|
||||
c.execute(
|
||||
"UPDATE data SET api_key = ?, channel_id = ?, is_active = ?, max_tokens = ?, temperature = ?, frequency_penalty = ?, presence_penalty = ?, prompt_size = ? WHERE guild_id = ?",
|
||||
(None, None, False, 50, 0.9, 0.0, 0.0, 0, ctx.guild.id),
|
||||
)
|
||||
conn.commit()
|
||||
await ctx.respond("Deleted", ephemeral=True)
|
||||
|
||||
#create a command called "enable" that only admins can use
|
||||
# create a command called "enable" that only admins can use
|
||||
@discord.slash_command(name="enable", description="Enable the bot")
|
||||
##@discord.commands.permissions(administrator=True)
|
||||
async def enable(self, ctx: discord.ApplicationContext):
|
||||
#if the guild is eqal to 1014156298226515970, the guild is banned
|
||||
# if the guild is eqal to 1014156298226515970, the guild is banned
|
||||
if ctx.guild.id == 1014156298226515970:
|
||||
await ctx.respond("This server is banned for bad and nsfw use.", ephemeral=True)
|
||||
await ctx.respond(
|
||||
"This server is banned for bad and nsfw use.", ephemeral=True
|
||||
)
|
||||
return
|
||||
#check if the guild is in the database
|
||||
debug(f"The user {ctx.author} ran the enable command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}")
|
||||
# check if the guild is in the database
|
||||
debug(
|
||||
f"The user {ctx.author} ran the enable command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}"
|
||||
)
|
||||
c.execute("SELECT * FROM data WHERE guild_id = ?", (ctx.guild.id,))
|
||||
if c.fetchone() is None:
|
||||
await ctx.respond("This server is not setup", ephemeral=True)
|
||||
return
|
||||
#enable the guild
|
||||
c.execute("UPDATE data SET is_active = ? WHERE guild_id = ?", (True, ctx.guild.id))
|
||||
# enable the guild
|
||||
c.execute(
|
||||
"UPDATE data SET is_active = ? WHERE guild_id = ?", (True, ctx.guild.id)
|
||||
)
|
||||
conn.commit()
|
||||
await ctx.respond("Enabled", ephemeral=True)
|
||||
|
||||
#create a command called "disable" that only admins can use
|
||||
# create a command called "disable" that only admins can use
|
||||
@discord.slash_command(name="disable", description="Disable the bot")
|
||||
##@discord.commands.permissions(administrator=True)
|
||||
async def disable(self, ctx: discord.ApplicationContext):
|
||||
debug(f"The user {ctx.author} ran the disable command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}")
|
||||
#check if the guild is in the database
|
||||
debug(
|
||||
f"The user {ctx.author} ran the disable command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}"
|
||||
)
|
||||
# check if the guild is in the database
|
||||
c.execute("SELECT * FROM data WHERE guild_id = ?", (ctx.guild.id,))
|
||||
if c.fetchone() is None:
|
||||
await ctx.respond("This server is not setup", ephemeral=True)
|
||||
return
|
||||
#disable the guild
|
||||
c.execute("UPDATE data SET is_active = ? WHERE guild_id = ?", (False, ctx.guild.id))
|
||||
# disable the guild
|
||||
c.execute(
|
||||
"UPDATE data SET is_active = ? WHERE guild_id = ?", (False, ctx.guild.id)
|
||||
)
|
||||
conn.commit()
|
||||
await ctx.respond("Disabled", ephemeral=True)
|
||||
|
||||
#create a command calles "add channel" that can only be used in premium servers
|
||||
@discord.slash_command(name="add_channel", description="Add a channel to the list of channels. Premium only.")
|
||||
@discord.option(name="channel", description="The channel to add", type=discord.TextChannel, required=False)
|
||||
async def add_channel(self, ctx: discord.ApplicationContext, channel: discord.TextChannel = None):
|
||||
debug(f"The user {ctx.author} ran the add_channel command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}")
|
||||
#check if the guild is in the database
|
||||
# create a command calles "add channel" that can only be used in premium servers
|
||||
@discord.slash_command(
|
||||
name="add_channel",
|
||||
description="Add a channel to the list of channels. Premium only.",
|
||||
)
|
||||
@discord.option(
|
||||
name="channel",
|
||||
description="The channel to add",
|
||||
type=discord.TextChannel,
|
||||
required=False,
|
||||
)
|
||||
async def add_channel(
|
||||
self, ctx: discord.ApplicationContext, channel: discord.TextChannel = None
|
||||
):
|
||||
debug(
|
||||
f"The user {ctx.author} ran the add_channel command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}"
|
||||
)
|
||||
# check if the guild is in the database
|
||||
c.execute("SELECT * FROM data WHERE guild_id = ?", (ctx.guild.id,))
|
||||
if c.fetchone() is None:
|
||||
await ctx.respond("This server is not setup", ephemeral=True)
|
||||
return
|
||||
#check if the guild is premium
|
||||
try :
|
||||
# check if the guild is premium
|
||||
try:
|
||||
cp.execute("SELECT premium FROM data WHERE guild_id = ?", (ctx.guild.id,))
|
||||
premium = cp.fetchone()[0]
|
||||
except :
|
||||
except:
|
||||
premium = 0
|
||||
if not premium:
|
||||
await ctx.respond("This server is not premium", ephemeral=True)
|
||||
return
|
||||
if channel is None:
|
||||
channel = ctx.channel
|
||||
#check if the channel is already in the list
|
||||
# check if the channel is already in the list
|
||||
c.execute("SELECT channel_id FROM data WHERE guild_id = ?", (ctx.guild.id,))
|
||||
if str(channel.id) == c.fetchone()[0]:
|
||||
await ctx.respond("This channel is already set as the main channel", ephemeral=True)
|
||||
await ctx.respond(
|
||||
"This channel is already set as the main channel", ephemeral=True
|
||||
)
|
||||
return
|
||||
cp.execute("SELECT * FROM channels WHERE guild_id = ?", (ctx.guild.id,))
|
||||
guild_channels = cp.fetchone()
|
||||
if guild_channels is None:
|
||||
# if the channel is not in the list, add it
|
||||
cp.execute("INSERT INTO channels VALUES (?, ?, ?, ?, ?, ?)", (ctx.guild.id, channel.id, None, None, None, None))
|
||||
cp.execute(
|
||||
"INSERT INTO channels VALUES (?, ?, ?, ?, ?, ?)",
|
||||
(ctx.guild.id, channel.id, None, None, None, None),
|
||||
)
|
||||
connp.commit()
|
||||
await ctx.respond(f"Added channel **{channel.name}**", ephemeral=True)
|
||||
return
|
||||
@@ -123,51 +191,76 @@ class Setup (discord.Cog) :
|
||||
return
|
||||
for i in range(5):
|
||||
if channels[i] == None:
|
||||
cp.execute(f"UPDATE channels SET channel{i} = ? WHERE guild_id = ?", (channel.id, ctx.guild.id))
|
||||
cp.execute(
|
||||
f"UPDATE channels SET channel{i} = ? WHERE guild_id = ?",
|
||||
(channel.id, ctx.guild.id),
|
||||
)
|
||||
connp.commit()
|
||||
await ctx.respond(f"Added channel **{channel.name}**", ephemeral=True)
|
||||
return
|
||||
await ctx.respond("You can only add 5 channels", ephemeral=True)
|
||||
|
||||
#create a command called "remove channel" that can only be used in premium servers
|
||||
@discord.slash_command(name="remove_channel", description="Remove a channel from the list of channels. Premium only.")
|
||||
@discord.option(name="channel", description="The channel to remove", type=discord.TextChannel, required=False)
|
||||
async def remove_channel(self, ctx: discord.ApplicationContext, channel: discord.TextChannel = None):
|
||||
debug(f"The user {ctx.author} ran the remove_channel command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}")
|
||||
#check if the guild is in the database
|
||||
# create a command called "remove channel" that can only be used in premium servers
|
||||
@discord.slash_command(
|
||||
name="remove_channel",
|
||||
description="Remove a channel from the list of channels. Premium only.",
|
||||
)
|
||||
@discord.option(
|
||||
name="channel",
|
||||
description="The channel to remove",
|
||||
type=discord.TextChannel,
|
||||
required=False,
|
||||
)
|
||||
async def remove_channel(
|
||||
self, ctx: discord.ApplicationContext, channel: discord.TextChannel = None
|
||||
):
|
||||
debug(
|
||||
f"The user {ctx.author} ran the remove_channel command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}"
|
||||
)
|
||||
# check if the guild is in the database
|
||||
c.execute("SELECT * FROM data WHERE guild_id = ?", (ctx.guild.id,))
|
||||
if c.fetchone() is None:
|
||||
await ctx.respond("This server is not setup", ephemeral=True)
|
||||
return
|
||||
#check if the guild is premium
|
||||
try :
|
||||
# check if the guild is premium
|
||||
try:
|
||||
cp.execute("SELECT premium FROM data WHERE guild_id = ?", (ctx.guild.id,))
|
||||
premium = cp.fetchone()[0]
|
||||
except :
|
||||
except:
|
||||
premium = 0
|
||||
if not premium:
|
||||
await ctx.respond("This server is not premium", ephemeral=True)
|
||||
return
|
||||
if channel is None:
|
||||
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,))
|
||||
guild_channels = cp.fetchone()
|
||||
c.execute("SELECT channel_id FROM data WHERE guild_id = ?", (ctx.guild.id,))
|
||||
if str(channel.id) == c.fetchone()[0]:
|
||||
await ctx.respond("This channel is set as the main channel and therefore cannot be removed. Type /setup to change the main channel.", ephemeral=True)
|
||||
await ctx.respond(
|
||||
"This channel is set as the main channel and therefore cannot be removed. Type /setup to change the main channel.",
|
||||
ephemeral=True,
|
||||
)
|
||||
return
|
||||
if guild_channels is None:
|
||||
await ctx.respond("This channel was not added. Nothing changed", ephemeral=True)
|
||||
await ctx.respond(
|
||||
"This channel was not added. Nothing changed", ephemeral=True
|
||||
)
|
||||
return
|
||||
channels = guild_channels[1:]
|
||||
if str(channel.id) not in channels:
|
||||
await ctx.respond("This channel was not added. Nothing changed", ephemeral=True)
|
||||
await ctx.respond(
|
||||
"This channel was not added. Nothing changed", ephemeral=True
|
||||
)
|
||||
return
|
||||
#remove the channel from the list
|
||||
# remove the channel from the list
|
||||
for i in range(5):
|
||||
if channels[i] == str(channel.id):
|
||||
cp.execute(f"UPDATE channels SET channel{i} = ? WHERE guild_id = ?", (None, ctx.guild.id))
|
||||
cp.execute(
|
||||
f"UPDATE channels SET channel{i} = ? WHERE guild_id = ?",
|
||||
(None, ctx.guild.id),
|
||||
)
|
||||
connp.commit()
|
||||
await ctx.respond(f"Removed channel **{channel.name}**", ephemeral=True)
|
||||
return
|
||||
return
|
||||
|
||||
Reference in New Issue
Block a user