From 9c0f96decc32fe669b353679a0ca7f5853fbd05f Mon Sep 17 00:00:00 2001
From: jacob1 <jfu614@gmail.com>
Date: Sun, 27 Aug 2017 00:17:22 -0400
Subject: [PATCH] fix issue where night wouldn't end if wolves were diseased

---
 src/roles/wolf.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/roles/wolf.py b/src/roles/wolf.py
index fbf45c1..ba1b0bd 100644
--- a/src/roles/wolf.py
+++ b/src/roles/wolf.py
@@ -433,8 +433,6 @@ def on_get_role_metadata(evt, var, kind):
         nevt = Event("wolf_numkills", {"numkills": 1})
         nevt.dispatch(var)
         evt.data["wolf"] = nevt.data["numkills"]
-        if var.DISEASED_WOLVES:
-            evt.data["wolf"] = 0
         # TODO: split into alpha
         if var.ALPHA_ENABLED:
             # alpha wolf gives an extra kill; note that we consider someone being
@@ -444,4 +442,10 @@ def on_get_role_metadata(evt, var, kind):
             # as any other wolf kill).
             evt.data["wolf"] += 1
 
+@event_listener("wolf_numkills", priority=10)
+def on_wolf_numkills(evt, var):
+    if var.DISEASED_WOLVES:
+        evt.data["numkills"] = 0
+        evt.stop_processing = True
+
 # vim: set sw=4 expandtab: