From bd66621a433141713865377c5a7c43a5625060ba Mon Sep 17 00:00:00 2001 From: jacob1 Date: Sat, 16 Dec 2017 21:11:56 -0500 Subject: [PATCH] fix ORIGINAL_ROLES not being updated on player swaps --- src/wolfgame.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/wolfgame.py b/src/wolfgame.py index 043f816..72567bc 100644 --- a/src/wolfgame.py +++ b/src/wolfgame.py @@ -72,7 +72,7 @@ var.LAST_GOAT = {} var.USERS = {} var.ADMIN_PINGING = False -var.ORIGINAL_ROLES = {} +var.ORIGINAL_ROLES = {} # type: Dict[str, Set[users.User]] var.DCED_LOSERS = set() # type: Set[users.User] var.PLAYERS = {} var.DCED_PLAYERS = {} @@ -602,6 +602,10 @@ def swap_player(evt, var, old_user, user): if old_user in players: players.remove(old_user) players.add(user) + for role, players in var.ORIGINAL_ROLES.items(): + if old_user in players: + players.remove(old_user) + players.add(user) @command("pingif", "pingme", "pingat", "pingpref", pm=True)