From 9c36602e21017800564bb9871ecf5a019bf77520 Mon Sep 17 00:00:00 2001
From: jacob1 <jfu614@gmail.com>
Date: Sat, 3 Oct 2015 16:12:45 -0400
Subject: [PATCH] fix error when lover changes nickname and then is matched
 again

---
 src/wolfgame.py | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/src/wolfgame.py b/src/wolfgame.py
index 2993c41..5fa381f 100644
--- a/src/wolfgame.py
+++ b/src/wolfgame.py
@@ -3251,7 +3251,8 @@ def rename_player(cli, prefix, nick):
                             var.DOCTORS, var.BITTEN_ROLES, var.LYCAN_ROLES, var.AMNESIAC_ROLES):
                 if prefix in dictvar.keys():
                     dictvar[nick] = dictvar.pop(prefix)
-            for dictvar in (var.KILLS, var.LOVERS, var.ORIGINAL_LOVERS):
+            # Looks like {'jacob2': ['5'], '7': ['3']}
+            for dictvar in (var.KILLS,):
                 kvp = []
                 for a,b in dictvar.items():
                     nl = []
@@ -3265,6 +3266,21 @@ def rename_player(cli, prefix, nick):
                 dictvar.update(kvp)
                 if prefix in dictvar.keys():
                     del dictvar[prefix]
+            # Looks like {'6': {'jacob3'}, 'jacob3': {'6'}}
+            for dictvar in (var.LOVERS, var.ORIGINAL_LOVERS):
+                kvp = []
+                for a,b in dictvar.items():
+                    nl = set()
+                    for n in b:
+                        if n == prefix:
+                            n = nick
+                        nl.add(n)
+                    if a == prefix:
+                        a = nick
+                    kvp.append((a,nl))
+                dictvar.update(kvp)
+                if prefix in dictvar.keys():
+                    del dictvar[prefix]
             for idx, tup in enumerate(var.EXCHANGED_ROLES):
                 a, b = tup
                 if a == prefix: