From eff492ceb8b1131c9584edc4da4b820f033b03a1 Mon Sep 17 00:00:00 2001 From: ddorn Date: Wed, 13 May 2020 15:49:00 +0200 Subject: [PATCH] =?UTF-8?q?:sparkles:=20poules=20=C3=A0=205?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit + 120s timeout for fracals + relative path for memes --- src/base_tirage.py | 33 +++++++++++++++++++++-- src/cogs/misc.py | 8 +++--- src/cogs/teams.py | 4 ++- src/cogs/tirages.py | 64 +++++++++++++++++++++++++++++++-------------- src/constants.py | 4 +-- 5 files changed, 85 insertions(+), 28 deletions(-) diff --git a/src/base_tirage.py b/src/base_tirage.py index bdea566..204d025 100644 --- a/src/base_tirage.py +++ b/src/base_tirage.py @@ -226,7 +226,9 @@ class BaseTirage(yaml.YAMLObject): accept = await self.next(bool, team.name) if accept.value: team.accepted_problems[poule.rnd] = pevent.value - await self.info_accepted(team, pevent.value) + await self.info_accepted( + team, pevent.value, self.availaible(pevent.value, poule) + ) else: await self.info_rejected(team, pevent.value, rnd=poule.rnd) team.rejected[poule.rnd].add(pevent.value) @@ -234,6 +236,33 @@ class BaseTirage(yaml.YAMLObject): current += 1 current %= len(teams) + if len(teams) == 5: + # We can determine the passage order only once problems are drawn. + order = [self.teams[tri] for tri in self.poules[poule]] + pbs = [team.accepted_problems[poule.rnd] for team in order] + + doubles = [] + i = 0 + while i < len(order): + team = order[i] + if pbs.count(team.accepted_problems[poule.rnd]) == 2: + # We pop the two with the same pb and add them to the doubles + doubles.append(order.pop(i)) + other = next( + filter( + lambda t: team.accepted_problems[poule.rnd] + == t.accepted_problems[poule.rnd], + order, + ) + ) + doubles.append(other) + order.remove(other) + else: + i += 1 + # The conflicts + order = doubles + order + self.poules[poule] = order + await self.annonce_poule(poule) async def draw_order(self, poule): @@ -292,7 +321,7 @@ class BaseTirage(yaml.YAMLObject): async def info_draw_pb(self, team, pb, rnd): """Called when a team draws a problem.""" - async def info_accepted(self, team, pb): + async def info_accepted(self, team, pb, still_available): """Called when a team accepts a problem.""" async def info_rejected(self, team, pb, rnd): diff --git a/src/cogs/misc.py b/src/cogs/misc.py index 832d575..a43581b 100644 --- a/src/cogs/misc.py +++ b/src/cogs/misc.py @@ -115,7 +115,7 @@ class MiscCog(Cog, name="Divers"): seed = seed or str(random.randint(0, 1_000_000_000)) async with aiohttp.ClientSession() as session: async with session.get( - FRACTAL_URL.format(seed=urllib.parse.quote(seed)) + FRACTAL_URL.format(seed=urllib.parse.quote(seed)), timeout=120 ) as resp: if resp.status != 200: return await ctx.send( @@ -247,7 +247,7 @@ class MiscCog(Cog, name="Divers"): raise TfjmError("Il n'y a pas de blague avec cet ID.") if joke.file: - file = discord.File(joke.file) + file = discord.File(File.MEMES / joke.file) else: file = None @@ -273,8 +273,8 @@ class MiscCog(Cog, name="Divers"): if message.attachments: file: discord.Attachment = message.attachments[0] - joke.file = str(File.MEMES / f"{joke_id}-{file.filename}") - await file.save(joke.file) + joke.file = str(f"{joke_id}-{file.filename}") + await file.save(File.MEMES / joke.file) jokes.append(joke) self.save_jokes(jokes) diff --git a/src/cogs/teams.py b/src/cogs/teams.py index d1273e0..da30be0 100644 --- a/src/cogs/teams.py +++ b/src/cogs/teams.py @@ -279,7 +279,9 @@ class TeamsCog(Cog, name="Teams"): channel_name, overwrites={ guild.default_role: discord.PermissionOverwrite(read_messages=False), - team_role: discord.PermissionOverwrite(read_messages=True), + team_role: discord.PermissionOverwrite( + read_messages=True, manage_channels=True + ), }, category=team_channel_category, reason=f"{ctx.author.name} à demandé une channel pour son équipe.", diff --git a/src/cogs/tirages.py b/src/cogs/tirages.py index c78c029..0ad7ad4 100644 --- a/src/cogs/tirages.py +++ b/src/cogs/tirages.py @@ -273,17 +273,17 @@ class DiscordTirage(BaseTirage): if len(teams) == 3: table = """``` -+-----+---------+---------+---------+ -| | Phase 1 | Phase 2 | Phase 3 | -| | Pb {0.pb} | Pb {1.pb} | Pb {2.pb} | -+-----+---------+---------+---------+ -| {0.name} | Déf | Rap | Opp | -+-----+---------+---------+---------+ -| {1.name} | Opp | Déf | Rap | -+-----+---------+---------+---------+ -| {2.name} | Rap | Opp | Déf | -+-----+---------+---------+---------+```""" - else: +╔═════╦═════════╦═════════╦═════════╗ +║ ║ Phase 1 ║ Phase 2 ║ Phase 3 ║ +║ ║ Pb {0.pb} ║ Pb {0.pb} ║ Pb {0.pb} ║ +╠═════╬═════════╬═════════╬═════════╣ +║ {0.name} ║ Def ║ Rap ║ Opp ║ +╠═════╬═════════╬═════════╬═════════╣ +║ {1.name} ║ Opp ║ Def ║ Rap ║ +╠═════╬═════════╬═════════╬═════════╣ +║ {2.name} ║ Rap ║ Opp ║ Def ║ +╚═════╩═════════╩═════════╩═════════╝```""" + elif len(teams) == 4: table = """``` +-----+---------+---------+---------+---------+ | | Phase 1 | Phase 2 | Phase 3 | Phase 4 | @@ -297,6 +297,26 @@ class DiscordTirage(BaseTirage): +-----+---------+---------+---------+---------+ | {3.name} | | Rap | Opp | Déf | +-----+---------+---------+---------+---------+```""" + elif len(teams) == 5: + table = """``` +╔═════╦═══════════════════╦═══════════════════╦═════════╗ +║ ║ Phase 1 ║ Phase 2 ║ Phase 3 ║ +╠═════╬═════════╦═════════╬═════════╦═════════╬═════════╣ +║ ║ Salle 1 ║ Salle 2 ║ Salle 1 ║ Salle 2 ║ Salle 1 ║ +║ ║ Pb {0.pb} ║ Pb {1.pb} ║ Pb {2.pb} ║ Pb {3.pb} ║ Pb {4.pb} ║ +╠═════╬═════════╬═════════╬═════════╬═════════╬═════════╣ +║ {0.name} ║ Def ║ ║ Opp ║ Rap ║ ║ +╠═════╬═════════╬═════════╬═════════╬═════════╬═════════╣ +║ {1.name} ║ ║ Def ║ Rap ║ ║ Opp ║ +╠═════╬═════════╬═════════╬═════════╬═════════╬═════════╣ +║ {2.name} ║ Opp ║ ║ Def ║ ║ Rap ║ +╠═════╬═════════╬═════════╬═════════╬═════════╬═════════╣ +║ {3.name} ║ Rap ║ Opp ║ ║ Def ║ ║ +╠═════╬═════════╬═════════╬═════════╬═════════╬═════════╣ +║ {4.name} ║ ║ Rap ║ ║ Opp ║ Def ║ +╚═════╩═════════╩═════════╩═════════╩═════════╩═════════╝```""" + else: + table = "WTF il n'y a pas 3,4 ou 5 equipes ici." embed = discord.Embed( title=f"Résumé du tirage entre {french_join([t.name for t in teams])}", @@ -389,12 +409,18 @@ class DiscordTirage(BaseTirage): ) @safe - async def info_accepted(self, team, pb): - await self.ctx.send( - f"L'équipe {team.mention} a accepté " - f"**{pb}** ! Les autres équipes " - f"ne peuvent plus l'accepter." - ) + async def info_accepted(self, team, pb, still_available): + if still_available: + await self.ctx.send( + f"L'équipe {team.mention} a accepté " + f"**{pb}** ! Une autre équipe peut encore l'accepter." + ) + else: + await self.ctx.send( + f"L'équipe {team.mention} a accepté " + f"**{pb}** ! Les autres équipes " + f"ne peuvent plus l'accepter." + ) @safe async def info_rejected(self, team, pb, rnd): @@ -574,8 +600,8 @@ class TirageCog(Cog, name="Tirages"): "par exemple `3+3` pour deux poules à trois équipes" ) - if not set(fmt).issubset({3, 4}): - raise TfjmError("Seuls les poules à 3 ou 4 équipes sont suportées.") + if not set(fmt).issubset({3, 4, 5}): + raise TfjmError("Seuls les poules à 3, 4 ou 5 équipes sont suportées.") # Here all data should be valid diff --git a/src/constants.py b/src/constants.py index bf299ec..997fe01 100644 --- a/src/constants.py +++ b/src/constants.py @@ -34,9 +34,9 @@ if TOKEN is None: GUILD = "690934836696973404" DIEGO = 430566197868625920 # Mon id BOT = 703305132300959754 -TEAMS_CHANNEL_CATEGORY = "Channels d'équipes" +TEAMS_CHANNEL_CATEGORY = "Channels d'équipes 2" EMBED_COLOR = 0xFFA500 -FRACTAL_URL = "https://thefractal.space/img/{seed}.png?size=1500" +FRACTAL_URL = "https://thefractal.space/img/{seed}.png?size=1000" FRACTAL_COOLDOWN = 30 # seconds ROUND_NAMES = ["premier tour", "deuxième tour"]