From 7b9b79a0d1c233035a5945b88da9058054ffb312 Mon Sep 17 00:00:00 2001 From: skizzerz Date: Fri, 12 Jun 2015 12:29:45 -0500 Subject: [PATCH] Make code a tiny bit more readable --- src/wolfgame.py | 18 +++--------------- 1 file changed, 3 insertions(+), 15 deletions(-) diff --git a/src/wolfgame.py b/src/wolfgame.py index 02e69d2..4d8d9a4 100644 --- a/src/wolfgame.py +++ b/src/wolfgame.py @@ -40,6 +40,7 @@ import signal from src import logger import urllib.request import sqlite3 +import itertools # done this way so that events is accessible in !eval (useful for debugging) from src import events @@ -3609,25 +3610,12 @@ def transition_day(cli, gameid=0): del_player(cli, deadperson, end_game = False, killer_role = "wolf" if deadperson in onlybywolves or deadperson in wolfghostvictims else "villager", deadlist = dead, original = deadperson) message = [] - processed_totems = [] - - for (i, player) in enumerate(havetotem): - ntotems = 1 - - if player in processed_totems: - continue - - for j in range(i + 1, len(havetotem)): - if havetotem[j] == player: - ntotems += 1 - else: - break + for player, tlist in itertools.groupby(havetotem): + ntotems = len(list(tlist)) message.append("\u0002{0}\u0002 seem{1} to be in possession of {2} mysterious totem{3}...".format( player, "ed" if player in dead else "s", "a" if ntotems == 1 else "\u0002{0}\u0002".format(ntotems), "s" if ntotems > 1 else "")) - processed_totems.append(player) - for brokentotem in brokentotem: message.append("Broken totem pieces were found next to \u0002{0}\u0002's body...".format(brokentotem)) cli.msg(chan, "\n".join(message))