✨ add error handling + start drawing procedure
This commit is contained in:
parent
3f3520aaf5
commit
55e5de95c6
|
@ -2,12 +2,14 @@
|
||||||
|
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
|
from typing import List
|
||||||
|
|
||||||
import discord
|
import discord
|
||||||
from discord.ext import commands
|
from discord.ext import commands
|
||||||
import random
|
import random
|
||||||
|
|
||||||
from discord.ext.commands import Context
|
from discord.ext.commands import Context
|
||||||
|
from discord.utils import get
|
||||||
|
|
||||||
TOKEN = os.environ.get("TFJM_DISCORD_TOKEN")
|
TOKEN = os.environ.get("TFJM_DISCORD_TOKEN")
|
||||||
|
|
||||||
|
@ -20,11 +22,77 @@ if TOKEN is None:
|
||||||
quit(1)
|
quit(1)
|
||||||
|
|
||||||
GUILD = "690934836696973404"
|
GUILD = "690934836696973404"
|
||||||
|
ORGA_ROLE = "Orga"
|
||||||
|
CAPTAIN_ROLE = "Capitaine"
|
||||||
|
|
||||||
|
|
||||||
|
class TfjmError(Exception):
|
||||||
|
def __init__(self, msg):
|
||||||
|
self.msg = msg
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return self.msg
|
||||||
|
|
||||||
|
|
||||||
|
class Team:
|
||||||
|
def __init__(self, name):
|
||||||
|
self.name = name
|
||||||
|
|
||||||
|
|
||||||
|
class Tirage:
|
||||||
|
def __init__(self, *teams):
|
||||||
|
assert len(teams) in (3, 4)
|
||||||
|
|
||||||
|
self.teams = {team: Team(team) for team in teams}
|
||||||
|
|
||||||
|
|
||||||
bot = commands.Bot(
|
bot = commands.Bot(
|
||||||
"!", help_command=commands.DefaultHelpCommand(no_category="Commandes")
|
"!", help_command=commands.DefaultHelpCommand(no_category="Commandes")
|
||||||
)
|
)
|
||||||
|
|
||||||
|
draws = {}
|
||||||
|
|
||||||
|
|
||||||
|
@bot.command(
|
||||||
|
name="start-draw",
|
||||||
|
help="Commence un tirage avec 3 ou 4 équipes.",
|
||||||
|
usage="équipe1 équipe2 équipe3 (équipe4)",
|
||||||
|
)
|
||||||
|
@commands.has_role(ORGA_ROLE)
|
||||||
|
async def start_draw(ctx: Context, *teams):
|
||||||
|
guild: discord.Guild = ctx.guild
|
||||||
|
|
||||||
|
channel = ctx.channel.id
|
||||||
|
if channel in draws:
|
||||||
|
raise TfjmError("Il y a déjà un tirage en cours sur cette Channel.")
|
||||||
|
|
||||||
|
if len(teams) not in (3, 4):
|
||||||
|
raise TfjmError("Il faut 3 ou 4 équipes pour un tirage.")
|
||||||
|
|
||||||
|
roles = [role.name for role in ctx.guild.roles]
|
||||||
|
for team in teams:
|
||||||
|
if team not in roles:
|
||||||
|
raise TfjmError("Le nom de l'équipe doit être exactement celui du rôle.")
|
||||||
|
|
||||||
|
captain: discord.Role = get(guild.roles, name=CAPTAIN_ROLE)
|
||||||
|
|
||||||
|
# Here everything should be alright
|
||||||
|
await ctx.send(
|
||||||
|
"Nous allons commencer le tirage du premier tour. "
|
||||||
|
"Seuls les capitaines de chaque équipe peuvent désormais écrire ici. "
|
||||||
|
"Pour plus de détails sur le déroulement du tirgae au sort, le règlement "
|
||||||
|
"est accessible sur https://tfjm.org/reglement."
|
||||||
|
)
|
||||||
|
await ctx.send(
|
||||||
|
"Nous allons d'abord tirer au sort l'ordre de tirage des problèmes, "
|
||||||
|
"puis l'ordre de passage lors du tour."
|
||||||
|
)
|
||||||
|
await ctx.send(
|
||||||
|
f"Les {captain.mention}s, vous pouvez désormais lancer un dé 100 "
|
||||||
|
"comme ceci `!dice 100`. "
|
||||||
|
"L'ordre des tirages suivants sera l'ordre croissant des lancers. "
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@bot.event
|
@bot.event
|
||||||
async def on_ready():
|
async def on_ready():
|
||||||
|
@ -39,7 +107,7 @@ async def on_ready():
|
||||||
)
|
)
|
||||||
async def dice(ctx: Context, n: int):
|
async def dice(ctx: Context, n: int):
|
||||||
if n < 1:
|
if n < 1:
|
||||||
raise ValueError(f"Called dice with n={n}")
|
raise TfjmError(f"Je ne peux pas lancer un dé à {n} faces, désolé.")
|
||||||
|
|
||||||
dice = random.randint(1, n)
|
dice = random.randint(1, n)
|
||||||
await ctx.send(f"Le dé à {n} faces s'est arrêté sur... **{dice}**")
|
await ctx.send(f"Le dé à {n} faces s'est arrêté sur... **{dice}**")
|
||||||
|
@ -69,10 +137,14 @@ async def random_problem(ctx: Context):
|
||||||
|
|
||||||
|
|
||||||
@bot.event
|
@bot.event
|
||||||
async def on_error(event, *args, **kwargs):
|
async def on_command_error(ctx: Context, error, *args, **kwargs):
|
||||||
print(event)
|
if isinstance(error, commands.CommandInvokeError):
|
||||||
print(*args)
|
msg = str(error.original)
|
||||||
print(kwargs)
|
else:
|
||||||
|
msg = str(error)
|
||||||
|
|
||||||
|
print(repr(error), file=sys.stderr)
|
||||||
|
await ctx.send(msg)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|
Loading…
Reference in New Issue