ynerant/tfjm-discord-bot
1
0
Fork 0
mirror of https://gitlab.com/ddorn/tfjm-discord-bot.git synced 2025-11-04 02:12:06 +01:00
Code Issues Releases Wiki Activity

✨ don't draw problems that can't be accepted

Browse Source
This commit is contained in:
ddorn
2020-04-27 23:11:49 +02:00
parent c7ed5599b4
commit eb8b884d47
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/cogs/tirage_logic.py
View File

@@ -222,7 +222,7 @@ class Phase:
async def dice(self, ctx: Context, author, dice): async def dice(self, ctx: Context, author, dice):
raise UnwantedCommand() raise UnwantedCommand()
async def choose_problem(self, ctx: Context, author, problem): async def choose_problem(self, ctx: Context, author):
raise UnwantedCommand() raise UnwantedCommand()
async def accept(self, ctx: Context, author, yes): async def accept(self, ctx: Context, author, yes):
@@ -319,7 +319,7 @@ class TiragePhase(Phase):
def available(self, problem): def available(self, problem):
return all(team.accepted_problems[self.round] != problem for team in self.teams) return all(team.accepted_problems[self.round] != problem for team in self.teams)
async def choose_problem(self, ctx: Context, author, problem): async def choose_problem(self, ctx: Context, author):
team = self.current_team team = self.current_team
if self.team_for(author) != team: if self.team_for(author) != team:
raise UnwantedCommand( raise UnwantedCommand(
@@ -337,6 +337,12 @@ class TiragePhase(Phase):
"ou de le refuser (`!no)`." "ou de le refuser (`!no)`."
) )
# Choose an *available* problem
problems = [
p for p in PROBLEMS if self.available(p) and not p in team.accepted_problems
]
problem = random.choice(problems)
await ctx.send(f"{team.mention} a tiré **{problem}** !") await ctx.send(f"{team.mention} a tiré **{problem}** !")
if not self.available(problem): if not self.available(problem):
await ctx.send( await ctx.send(