From ab0920f824e58238d72420493f342db87770101a Mon Sep 17 00:00:00 2001
From: "Vgr E. Barry" <vgr255@live.ca>
Date: Fri, 8 Dec 2017 09:39:33 -0500
Subject: [PATCH] Make harlot listen to swap_player

---
 src/roles/harlot.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/roles/harlot.py b/src/roles/harlot.py
index 8c8620e..2c69a74 100644
--- a/src/roles/harlot.py
+++ b/src/roles/harlot.py
@@ -136,6 +136,14 @@ def on_begin_day(evt, var):
     VISITED.clear()
     PASSED.clear()
 
+@event_listener("swap_player")
+def on_swap(evt, var, old_user, user):
+    for actor, target in set(VISITED.items()):
+        if target is old_user:
+            VISITED[actor] = user
+        if actor is old_user:
+            VISITED[user] = VISITED.pop(actor)
+
 @event_listener("get_special")
 def on_get_special(evt, var):
     evt.data["special"].update(get_players(("harlot",)))