diff --git a/modules/wolfgame.py b/modules/wolfgame.py index c5090ba..2407276 100644 --- a/modules/wolfgame.py +++ b/modules/wolfgame.py @@ -1239,6 +1239,11 @@ def stop_game(cli, winner = ""): if acc != "*": var.update_role_stats(acc, orol, won, iwon) + for role in var.TEMPLATE_RESTRICTIONS.keys(): + if plr in var.ORIGINAL_ROLES[role]: + var.update_role_stats(acc, role, won, iwon) + if splr in var.LOVERS: + var.update_role_stats(acc, "lover", won, iwon) if won or iwon: winners.append(splr) diff --git a/settings/wolfgame.py b/settings/wolfgame.py index c13c5ef..eb95484 100644 --- a/settings/wolfgame.py +++ b/settings/wolfgame.py @@ -803,7 +803,7 @@ def update_game_stats(roleset, size, winner): (roleset, size, vwins, wwins, mwins, fwins, total)) def get_player_stats(acc, role): - if role.lower() not in [k.lower() for k in ROLE_GUIDE.keys()]: + if role.lower() not in [k.lower() for k in ROLE_GUIDE.keys()] and role != "lover": return "No such role: {0}".format(role) with conn: c.execute("SELECT player FROM rolestats WHERE player=? COLLATE NOCASE", (acc,))