From 05f048c6c9a6d135e7f8b12893e1123f09a59b5d Mon Sep 17 00:00:00 2001 From: nyuszika7h Date: Wed, 19 Feb 2014 21:02:37 +0100 Subject: [PATCH] Even more fixes for stasis. --- modules/wolfgame.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/modules/wolfgame.py b/modules/wolfgame.py index 6571493..8f55369 100644 --- a/modules/wolfgame.py +++ b/modules/wolfgame.py @@ -2518,15 +2518,18 @@ def on_error(cli, pfx, msg): @pmcmd("fstasis", admin_only=True) def fstasis(cli, nick, *rest): data = rest[0].split() - if len(data) == 2: + if data and len(data) < 3: lusers = {k.lower(): v for k, v in var.USERS.items()} user = data[0].lower() if user in lusers: cloak = lusers[user]['cloak'] else: cloak = None - amt = int(data[1]) - if cloak is not None: + if cloak is None: + cli.msg(nick, "Sorry, that user cannot be found.") + return + if len(data) == 2: + amt = int(data[1]) 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])) @@ -2539,8 +2542,12 @@ def fstasis(cli, nick, *rest): else: var.STASISED[cloak] = amt 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: + elif len(data) == 1: + if cloak in var.STASISED: + cli.msg(nick, "{0} ({1}) is in stasis for {2} games.".format(data[0], cloak, var.STASISED[cloak])) + else: + cli.msg(nick, "{0} ({1}) is not in stasis.".format(data[0], cloak)) elif not data: if var.STASISED: cli.msg(nick, "Currently stasised: {0}".format(