adding comments and removing certain methods from class

This commit is contained in:
Mylloon 2021-07-30 14:04:57 +02:00
parent c50cc204d9
commit f84cbc5838
2 changed files with 25 additions and 24 deletions

View file

@ -1,6 +1,6 @@
from twitchio.ext import commands from twitchio.ext import commands
from utils.core import load, listCommands from utils.core import load, listCommands
from utils.commands import CommandesDB from utils.commands import CommandesDB, existeCommande, existeTouteCommande
def prepare(client: commands.Bot): def prepare(client: commands.Bot):
client.add_cog(Commandes(client)) client.add_cog(Commandes(client))
@ -19,7 +19,7 @@ class Commandes(commands.Cog): # Les méthodes qui ont no_global_checks de Vrai
if commandName == None or commandMessage == None: if commandName == None or commandMessage == None:
return return
if ctx.author.is_mod: if ctx.author.is_mod:
if CommandesDB().existeTouteCommande(self.client, commandName)[0] == False: if existeTouteCommande(self.client, commandName)[0] == False:
CommandesDB().ajoutCommande(commandName, commandMessage) CommandesDB().ajoutCommande(commandName, commandMessage)
await ctx.send(f"@{ctx.author.name}, commande {commandName} ajoutée !") await ctx.send(f"@{ctx.author.name}, commande {commandName} ajoutée !")
else: else:
@ -33,7 +33,7 @@ class Commandes(commands.Cog): # Les méthodes qui ont no_global_checks de Vrai
if commandName == None: if commandName == None:
return return
if ctx.author.is_mod: if ctx.author.is_mod:
if CommandesDB().existeCommande(commandName)[0]: if existeCommande(commandName)[0]:
CommandesDB().suppressionCommande(commandName) CommandesDB().suppressionCommande(commandName)
await ctx.send(f"@{ctx.author.name}, commande {commandName} supprimée !") await ctx.send(f"@{ctx.author.name}, commande {commandName} supprimée !")
else: else:
@ -72,7 +72,7 @@ class Commandes(commands.Cog): # Les méthodes qui ont no_global_checks de Vrai
if commandName == None or commandMessage == None: if commandName == None or commandMessage == None:
return return
if ctx.author.is_mod: if ctx.author.is_mod:
if CommandesDB().existeCommande(commandName)[0]: if existeCommande(commandName)[0]:
CommandesDB().suppressionCommande(commandName) CommandesDB().suppressionCommande(commandName)
CommandesDB().ajoutCommande(commandName, commandMessage) CommandesDB().ajoutCommande(commandName, commandMessage)
await ctx.send(f"@{ctx.author.name}, commande {commandName} modifiée !") await ctx.send(f"@{ctx.author.name}, commande {commandName} modifiée !")
@ -84,6 +84,6 @@ class Commandes(commands.Cog): # Les méthodes qui ont no_global_checks de Vrai
@commands.Cog.event() @commands.Cog.event()
async def event_message(self, message): async def event_message(self, message):
if message.content.startswith(self.keys["PREFIX"]): if message.content.startswith(self.keys["PREFIX"]):
command = CommandesDB().existeCommande(message.content[1:].split(" ")[0]) # récupère le nom de la commande command = existeCommande(message.content[1:].split(" ")[0]) # récupère le nom de la commande
if command[0]: # vérification si existe if command[0]: # vérification si existe
await message.channel.send(f"@{message.author.name}, {command[1]}") # envois le contenu de la commande await message.channel.send(f"@{message.author.name}, {command[1]}") # envois le contenu de la commande

View file

@ -39,16 +39,18 @@ class CommandesDB(Database):
"""Retourne la liste des commandes.""" """Retourne la liste des commandes."""
return self.affichageResultat(self.requete("SELECT * FROM commandes;")) return self.affichageResultat(self.requete("SELECT * FROM commandes;"))
def existeCommande(self, command: str): def existeCommande(command: str):
commandes = self.listeCommande() """Vérifie qu'une commande existe dans la base de donnée."""
commandes = CommandesDB().listeCommande()
for commande in commandes: for commande in commandes:
if commande[0] == command: if commande[0] == command:
return (True, commande[1]) return (True, commande[1])
return (False,) return (False,)
def existeTouteCommande(self, client, commande: str): def existeTouteCommande(client, commande: str):
"""Vérifie qu'une commande existe dans la base de donnée et dans le bot en lui-même."""
commandes = [] commandes = []
for command in self.listeCommande(): for command in CommandesDB().listeCommande():
commandes.append(command[0]) commandes.append(command[0])
for command in listCommands(client): for command in listCommands(client):
commandes.append(command.name) commandes.append(command.name)
@ -58,4 +60,3 @@ class CommandesDB(Database):
if commande in commandes: if commande in commandes:
return (True,) return (True,)
return (False,) return (False,)