From 49303b62eb65cc4efa95ae77527b940876d79f8e Mon Sep 17 00:00:00 2001
From: nyuszika7h <nyuszika7h@cadoth.net>
Date: Wed, 19 Feb 2014 20:26:09 +0100
Subject: [PATCH] More fixes to stasis.

---
 modules/wolfgame.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/modules/wolfgame.py b/modules/wolfgame.py
index 4997316..a731628 100644
--- a/modules/wolfgame.py
+++ b/modules/wolfgame.py
@@ -2527,8 +2527,14 @@ def fstasis(cli, nick, *rest):
             cloak = None
         amt = data[1]
         if cloak is not None:
-            var.STASISED[cloak] = int(amt)
-            cli.msg(nick, "{0} ({1}) is now in stasis for {2} games.".format(data[0], cloak, amt))
+            if amt < 0 and cloak in var.STASISED:
+                var.STASISED[cloak] -= amt
+                cli.msg(nick, "{0} ({1}) is now in stasis for {2} games.".format(data[0], cloak, var.STASISED[cloak]))
+            elif amt <= 0:
+                var.STASISED[cloak].pop(int(amt))
+                cli.msg(nick, "{0} ({1}) is no longer in stasis.".format(data[0], cloak))
+            else:
+                cli.msg(nick, "{0} ({1}) is now in stasis for {2} games.".format(data[0], cloak, amt))
         else:
             cli.msg(nick, "Sorry, that user cannot be found.")
     else: