diff --git a/src/werewolf_bot.py b/src/werewolf_bot.py index 3ef0819..7ee894e 100644 --- a/src/werewolf_bot.py +++ b/src/werewolf_bot.py @@ -70,43 +70,23 @@ async def setup(ctx): await send_friendly(ctx, "This channel can now play Werewolf") -def channel_setup(command): - @functools.wraps(command) - async def wrapper(ctx, *args, **kwargs): - if ctx.channel not in game_instances: - await send_wrong(ctx, f"No game setup yet. Use {PREFIX}game setup") - else: - await command(ctx, *args, **kwargs) - return wrapper - - -def game_not_running(command): - @functools.wraps(command) - @channel_setup - async def wrapper(ctx, *args, **kwargs): - if game_instances[ctx.channel].running: - await send_wrong(ctx, "Sorry! A game is already running") - else: - await command(ctx, *args, **kwargs) - return wrapper - - def game_running(command): @functools.wraps(command) - @channel_setup - async def wrapper(ctx, *args, **kwargs): - if not game_instances[ctx.channel].running: - await send_wrong(ctx, "No game is running") + async def wrapper(ctx): + if ctx.channel not in game_instances: + await send_wrong(ctx, f"No game setup yet. Use {PREFIX}game setup") + elif game_instances[ctx.channel].running: + await send_wrong(ctx, "Sorry! A game is already running") else: - await command(ctx, *args, **kwargs) + await command(ctx) return wrapper def error_handling(command): @functools.wraps(command) - async def wrapper(ctx, *args, **kwargs): + async def wrapper(ctx): try: - await command(ctx, *args, **kwargs) + await command(ctx) except ValueError as error: await send_wrong(ctx, str(error)) except asyncio.TimeoutError: @@ -115,47 +95,31 @@ def error_handling(command): @game.command() -@game_not_running +@game_running @error_handling async def start(ctx): - game_instances[ctx.channel].game = bot.loop.create_task(game_instances[ctx.channel].round()) - await game_instances[ctx.channel].game + await game_instances[ctx.channel].game() @game.command() @game_running -@channel_setup -async def stop(ctx): - game_instances[ctx.channel].game.cancel() - await send_friendly(ctx, "Game canceled") - - -@game.command() -@game_not_running @error_handling async def players(ctx): await game_instances[ctx.channel].set_players(ctx.message) @game.command() -@game_not_running +@game_running @error_handling -async def roles(ctx, *args): - await game_instances[ctx.channel].set_roles(args) - - -@game.command() -@game_not_running -@error_handling -async def minutes(ctx, i): - await game_instances[ctx.channel].set_time(i) - +async def roles(ctx): + await game_instances[ctx.channel].set_roles(ctx.message.content.split()[3:]) # exclude commands +# ONLY FOR TESTING @game.command() @game_running @error_handling -async def time(ctx): - await send_friendly(ctx, game_instances[ctx.channel].remaining_time_string()) +async def vote(ctx): + await game_instances[ctx.channel].vote() # smaller commands @@ -176,10 +140,10 @@ async def ping(ctx): def developer(command): @functools.wraps(command) - async def wrapper(ctx, *args, **kwargs): + async def wrapper(ctx): DEV_ID = 461892912821698562 if ctx.author.id == DEV_ID: - await command(ctx, *args, **kwargs) + await command(ctx) else: await send_wrong(ctx, "This command is not for you!") return wrapper @@ -193,9 +157,8 @@ async def logout(ctx): @bot.command() @developer -async def debug(ctx, *args): +async def debug(ctx): print("DEBUG") - print(ctx.args) - print(ctx.kwargs) + print(ctx.message.author.id) bot.run(TOKEN) diff --git a/src/werewolf_game.py b/src/werewolf_game.py index 255bef0..8abf581 100644 --- a/src/werewolf_game.py +++ b/src/werewolf_game.py @@ -1,5 +1,4 @@ from random import shuffle -import time import asyncio import discord from werewolf_roles import Role, Doppelganger, Werewolf, Minion, Mason, Seer, Robber, Troublemaker, Drunk, Insomniac, Tanner, Hunter, No_role @@ -14,7 +13,6 @@ class Game: self.channel = channel self.player_list = [] self.role_list = [] - self.discussion_time = 300 # seconds async def send(self, message): await self.channel.send(embed=discord.Embed(description=message, color=0x00ffff)) @@ -30,10 +28,6 @@ class Game: self.role_list = [Role.match(r) for r in suggestions] # raises ValueError await self.send(f"Roles: {', '.join(r.name() for r in self.role_list)}") # send confirmation - async def set_time(self, msg): - self.discussion_time = int(msg) * 60 + 1 - await self.send(f"You have set the discussion time to {self.discussion_time//60} minutes") # send confirmation - def check(self): if not 0 <= len(self.player_list) <= 10: raise ValueError(f"Invalid number of players: {len(self.player_list)}") @@ -42,25 +36,21 @@ class Game: def setup(self): self.role = dict() - # setting default value for r in Role.__subclasses__(): - if r == No_role: - continue - if r in [Werewolf, Mason]: - self.role[r] = [] - else: - r(self).add_yourself() + if r not in [Werewolf, Mason, No_role]: + r(self) + self.role[Werewolf] = [] + self.role[Mason] = [] self.voting_list = self.player_list + [No_player()] for c in self.voting_list: - c.reset() + c.tally = 0 + c.won = c.dead = False def distribute_roles(self): shuffle(self.role_list) for i in range(len(self.player_list)): - role_obj = self.role_list[i](self, self.player_list[i]) - self.player_list[i].setRole(role_obj) - role_obj.add_yourself() - self.middle_card = [r(self) for r in self.role_list[-3:]] + self.role_list[i](self, self.player_list[i]) + self.middle_card = self.role_list[-3:] async def start_night(self): await self.for_all_player(lambda p: p.send_normal("*The night has begun*")) @@ -84,32 +74,12 @@ class Game: async def start_day(self): await self.send("The day has started") - def remaining_time_string(self): - t = int(self.start_time + self.discussion_time - time.time()) - return f"{t//60} minute(s) and {t%60} second(s)" - - async def discussion_timer(self): - self.start_time = time.time() - await self.send(f"You have {self.remaining_time_string()} to discuss") - await asyncio.sleep(self.discussion_time / 2) - await self.send(f"{self.remaining_time_string()} remaining") - await asyncio.sleep(self.discussion_time / 2 - 60) - await self.send(f"{self.remaining_time_string()} remaining") - await asyncio.sleep(30) - await self.send(f"{self.remaining_time_string()} remaining") - await asyncio.sleep(30) - - async def early_vote(self): - await self.for_all_player(lambda p: p.ready_to_vote()) - - async def discussion_finished(self): - done, pending = await asyncio.wait([self.discussion_timer(), self.early_vote()], return_when=asyncio.FIRST_COMPLETED) - for p in pending: - p.cancel() - await asyncio.wait(pending) - async def vote(self): + + # vote + # replace with dm: await self.receive('$vote') await self.send("Vote in DM") + await self.for_all_player(lambda p: p.cast_vote("Who do you want to kill?", self.voting_list)) def tally(self): @@ -175,17 +145,15 @@ class Game: return werewolf_won, village_won, tanner_won, dead async def result(self, werewolf_won, village_won, tanner_won, dead): - winnning = [] - if werewolf_won: - winnning.append("Werewolves") - if village_won: - winnning.append("Village") - if tanner_won: - winnning.append(f"{sum(1 for d in dead if d.day_role.is_role(Tanner))} tanner") - if len(winnning) == 0: - winnning = ["No one"] - embed = discord.Embed(title=f"{' and '.join(winnning)} won!", color=0x00ffff) + if werewolf_won: + winnning = ["Werewolves won!"] + if village_won: + winnning = ["Village won!"] + if tanner_won: + winnning.append(f"{sum(1 for d in dead if d.day_role.is_role(Tanner))} tanner won") + + embed = discord.Embed(title=' and '.join(winnning), color=0x00ffff) for p in self.player_list: won_emoji = ":trophy:" if p.won else ":frowning2:" dead_emoji = ":skull:" if p.dead else ":no_mouth:" @@ -197,20 +165,24 @@ class Game: def end(self): self.running = False - async def round(self): + async def game(self): try: self.check() - self.setup() self.running = True + self.setup() self.distribute_roles() await self.start_night() await self.send_role() + await self.night_phases() + await self.start_day() - await self.discussion_finished() + # discussion timer + await self.vote() self.tally() await self.result(*self.who_won(self.who_dead())) + await self.send("Round ended") finally: self.end() diff --git a/src/werewolf_players.py b/src/werewolf_players.py index 13af25b..5c7f89e 100644 --- a/src/werewolf_players.py +++ b/src/werewolf_players.py @@ -14,10 +14,6 @@ class Player: def setRole(self, role): self.day_role = self.night_role = role - def reset(self): - self.tally = 0 - self.won = self.dead = False - def name(self): return self.member.name @@ -85,12 +81,6 @@ class Player: async def cast_vote(self, question, options): self.vote = options[await self.get_choice(question, options)] - async def ready_to_vote(self): - def check(msg): - return msg.channel == self.dm and msg.author == self.member and msg.content.casefold() == "vote" - await self.game.bot.wait_for('message', check=check) - await self.send_confirmation("You are ready to vote") - class No_player(Player): def name(self): diff --git a/src/werewolf_roles.py b/src/werewolf_roles.py index ce14fb1..0c5b213 100644 --- a/src/werewolf_roles.py +++ b/src/werewolf_roles.py @@ -1,5 +1,4 @@ import functools -from fuzzywuzzy import fuzz from werewolf_players import No_player @@ -7,6 +6,8 @@ class Role: def __init__(self, game, player=No_player()): self.game = game self.player = player + self.player.setRole(self) + self.add_yourself() def add_yourself(self): self.game.role[type(self)] = self @@ -19,7 +20,10 @@ class Role: @staticmethod def match(message): - return max(Role.__subclasses__(), key=lambda role_class: fuzz.ratio(message, role_class.name())) + for role_class in Role.__subclasses__(): + if message.casefold() == role_class.name(): + return role_class + raise ValueError(f"Invalid role: {message}") @staticmethod def no_player(func): @@ -125,7 +129,7 @@ class Robber(Role): self.choice = await self.player.get_choice("Which player do you want to rob?", self.player.other()) @Role.no_player - async def simulate(self): + def simulate(self): self.player.swap(self.player.other()[self.choice]) @Role.no_player @@ -139,7 +143,7 @@ class Troublemaker(Role): self.A, self.B = await self.player.get_double_choice("Who do you want to exchange? (send two numbers)", self.player.other()) @Role.no_player - async def simulate(self): + def simulate(self): self.player.other()[self.A].swap(self.player.other()[self.B]) @@ -149,14 +153,14 @@ class Drunk(Role): self.choice = await self.player.get_choice("Which card from the middle do you want to take?", ["left", "middle", "right"]) @Role.no_player - async def simulate(self): + def simulate(self): self.player.day_role, self.game.middle_card[self.choice] = self.game.middle_card[self.choice], self.player.day_role # swap class Insomniac(Role): @Role.no_player async def send_info(self): - await self.player.send_info(f"You are: {self.player.day_role}") + await self.player.send_info(f"You are now: {self.player.day_role}") class Villager(Role):