update coins.py to use command specs (and not docstrings)

1 files changed, 77 insertions, 106 deletions

diff --git a/modules/coins.py b/modules/coins.py
index aa897948..3a56ea6f 100644
--- a/modules/coins.py
+++ b/modules/coins.py
@@ -8,7 +8,6 @@ SIDES = {"heads": 0, "tails": 1}
 DEFAULT_REDEEM_DELAY = 600 # 600 seconds, 10 minutes
 DEFAULT_REDEEM_AMOUNT = "100.0"
 DEFAULT_INTEREST_RATE = "0.01"
-INTEREST_INTERVAL = 60*60 # 1 hour
 REGEX_FLOAT = re.compile("(?:\d+(?:\.\d{1,2}|$)|\.\d{1,2})")
 
 DECIMAL_ZERO = decimal.Decimal("0")
@@ -40,6 +39,32 @@ class CoinParseException(Exception):
     pass
 
 class Module(ModuleManager.BaseModule):
+    def on_load(self):
+        self.exports.add("command-spec.coins", self._coins_spec)
+        self.exports.add("command-spec.coinsmany", self._coins_many_spec)
+
+    def _coin_spec_parse(self, word):
+        try:
+            out = decimal.Decimal(word)
+        except decimal.InvalidOperation:
+            raise utils.parse.SpecTypeError(
+                "Please provide a valid coin amount")
+        if out >= DECIMAL_ZERO:
+            return out
+        else:
+            raise utils.parse.SpecTypeError(
+                "Please provide a positive coin amount")
+    def _coins_spec(self, server, channel, user, args):
+        if args:
+            return self._coin_spec_parse(args[0]), 1
+        else:
+            raise utils.parse.SpecTypeError("No coin amount provided")
+    def _coins_many_spec(self, server, channel, user, args):
+        out = []
+        for arg in args:
+            out.append(self._coin_spec_parse(arg))
+        return (out or None), 1
+
     def _until_next_hour(self, now=None):
         now = now or datetime.datetime.utcnow()
         until_next_hour = 60-now.second
@@ -108,10 +133,8 @@ class Module(ModuleManager.BaseModule):
                 "Please provide a valid positive coin amount")
 
     @utils.hook("received.command.coins")
+    @utils.kwarg("help", "Show how many coins you have")
     def coins(self, event):
-        """
-        :help: Show how many coins you have
-        """
         if event["args_split"]:
             target = event["server"].get_user(event["args_split"][0])
         else:
@@ -120,40 +143,30 @@ class Module(ModuleManager.BaseModule):
         event["stdout"].write("%s has %s coin%s" % (target.nickname,
             self._coin_str_human(coins), "" if coins == 1 else "s"))
 
-    @utils.hook("received.command.resetcoins", min_args=1)
+    @utils.hook("received.command.resetcoins")
+    @utils.kwarg("help", "Reset a user's coins to 0")
+    @utils.kwarg("permission", "reset-coins")
+    @utils.spec("!<nickname>ouser")
     def reset_coins(self, event):
-        """
-        :help: Reset a user's coins to 0
-        :usage: <target>
-        :permission: resetcoins
-        """
         target = event["server"].get_user(event["args_split"][0])
         coins = self._get_user_coins(target)
         self._take(event["server"], target, coins)
         event["stdout"].write("Reset coins for %s" % target.nickname)
 
-    @utils.hook("received.command.givecoins", min_args=1)
+    @utils.hook("received.command.givecoins")
+    @utils.kwarg("help", "Create coins and give them to a user")
+    @utils.kwarg("permission", "give-coins")
+    @utils.spec("!<nickname>ouser !<amount>coins")
     def give_coins(self, event):
-        """
-        :help: Give coins to a user
-        :usage: <nickname> <coins>
-        :permission: givecoins
-        """
-        target = event["server"].get_user(event["args_split"][0])
-        try:
-            coins = self._parse_coins(event["args_split"][1], DECIMAL_ZERO)
-        except CoinParseException as e:
-            raise utils.EventError("%s: %s" % (event["user"].nickname, str(e)))
-
+        target = event["spec"][0]
+        coins = event["spec"][1]
         self._give(event["server"], target, coins)
         event["stdout"].write("Gave '%s' %s coins" % (target.nickname,
             self._coin_str(coins)))
 
     @utils.hook("received.command.richest")
+    @utils.kwarg("help", "Show the top 10 richest users")
     def richest(self, event):
-        """
-        :help: Show the top 10 richest users
-        """
         top_10 = utils.top_10(self._all_coins(event["server"]),
             convert_key=lambda nickname:
             event["server"].get_user(nickname).nickname,
@@ -164,10 +177,8 @@ class Module(ModuleManager.BaseModule):
         return "redeem|%s|%s@%s" % (server.id, user.username, user.hostname)
 
     @utils.hook("received.command.redeemcoins")
+    @utils.kwarg("help", "Redeem your free coins")
     def redeem_coins(self, event):
-        """
-        :help: Redeem your free coins
-        """
         user_coins = self._get_user_coins(event["user"])
         if user_coins == DECIMAL_ZERO:
             cache = self._redeem_cache(event["server"], event["user"])
@@ -190,29 +201,19 @@ class Module(ModuleManager.BaseModule):
                 "%s: You can only redeem coins when you have none" %
                 event["user"].nickname)
 
-    @utils.hook("received.command.flip", min_args=2, authenticated=True)
+    @utils.hook("received.command.flip")
+    @utils.kwarg("help", "Bet on a coin flip")
+    @utils.kwarg("authenticated", True)
+    @utils.spec("!'heads,tails !'all")
+    @utils.spec("!'heads,tails !<amount>coins")
     def flip(self, event):
-        """
-        :help: Bet on a coin flip
-        :usage: heads|tails <coin amount>
-        """
-        side_name = event["args_split"][0].lower()
-        coin_bet = event["args_split"][1].lower()
+        side_name = event["spec"][0]
+        coin_bet = event["spec"][1]
         if coin_bet == "all":
             coin_bet = self._get_user_coins(event["user"])
             if coin_bet <= DECIMAL_ZERO:
                 raise utils.EventError("%s: You have no coins to bet" %
                     event["user"].nickname)
-        else:
-            try:
-                coin_bet = self._parse_coins(coin_bet, DECIMAL_BET_MINIMUM)
-            except CoinParseException as e:
-                raise utils.EventError("%s: %s" % (event["user"].nickname,
-                    str(e)))
-
-        if not side_name in SIDES:
-             raise utils.EventError("%s: Please provide 'heads' or 'tails'" %
-                event["user"].nickname)
 
         user_coins = self._get_user_coins(event["user"])
         if coin_bet > user_coins:
@@ -242,23 +243,18 @@ class Module(ModuleManager.BaseModule):
                 )
             )
 
-    @utils.hook("received.command.sendcoins", min_args=2, authenticated=True)
+    @utils.hook("received.command.sendcoins")
+    @utils.kwarg("help", "Send coins to another user")
+    @utils.kwarg("authenticated", True)
+    @utils.spec("!<nickname>ouser !<amount>coins")
     def send(self, event):
-        """
-        :help: Send coins to another user
-        :usage: <nickname> <amount>
-        """
-        target_user = event["server"].get_user(event["args_split"][0])
+        target_user = event["spec"][0]
 
         if event["user"].get_id() == target_user.get_id():
             raise utils.EventError("%s: You can't send coins to yourself" %
                 event["user"].nickname)
 
-        send_amount = event["args_split"][1]
-        try:
-            send_amount = self._parse_coins(send_amount, DECIMAL_ZERO)
-        except CoinParseException as e:
-            raise utils.EventError("%s: %s" % (event["user"].nickname, str(e)))
+        send_amount = event["spec"][1]
 
         user_coins = self._get_user_coins(event["user"])
         redeem_amount = self._redeem_amount(event["server"])
@@ -288,39 +284,29 @@ class Module(ModuleManager.BaseModule):
     def _double_street(self, i):
         return (i*3)-2, (i*3)+3
 
-    @utils.hook("received.command.roulette", min_args=2, authenticated=True)
+    @utils.hook("received.command.roulette")
+    @utils.kwarg("help", "Spin a roulette wheel")
+    @utils.kwarg("authenticated", True)
+    @utils.spec("!<types>word !'all")
+    @utils.spec("!<types>word !<amounts>coinsmany")
     def roulette(self, event):
-        """
-        :help: Spin a roulette wheel
-        :usage: <type> <amount>
-        """
-        expected_args = 1
-        expected_args += len(event["args_split"][0].split(","))
+        bets = event["spec"][0].lower().split(",")
+        if not len(bets) <= len(event["spec"][1]):
+            raise utils.EventError("%s: Please provide an amount for each bet" %
+                event["user"].nickname)
 
-        bets = event["args_split"][0].lower().split(",")
         if "0" in bets:
             raise utils.EventError("%s: You can't bet on 0" %
                 event["user"].nickname)
-        bet_amounts = [amount.lower() for amount in event["args_split"][
-            1:expected_args]]
-        if len(bet_amounts) < len(bets):
-            raise utils.EventError("%s: Please provide an amount for each bet" %
-                event["user"].nickname)
-
-        if len(bet_amounts) == 1 and bet_amounts[0] == "all":
-            bet_amounts[0] = self._get_user_coins(event["user"])
-            if bet_amounts[0] <= DECIMAL_ZERO:
+        bet_amounts = []
+        if event["spec"][1] == "all":
+            all_coins = self._get_user_coins(event["user"])
+            if all_coins <= DECIMAL_ZERO:
                 raise utils.EventError("%s: You have no coins to bet" %
                     event["user"].nickname)
-            bet_amounts[0] = self._coin_str(bet_amounts[0])
-
-        for i, bet_amount in enumerate(bet_amounts):
-            try:
-                bet_amounts[i] = self._parse_coins(bet_amount,
-                    DECIMAL_BET_MINIMUM)
-            except CoinParseException as e:
-                raise utils.EventError("%s: %s" % (event["user"].nickname,
-                    str(e)))
+            bet_amounts = [all_coins]
+        else:
+            bet_amounts = event["spec"][1]
 
         bet_amount_total = sum(bet_amounts)
 
@@ -452,19 +438,12 @@ class Module(ModuleManager.BaseModule):
                     server.set_user_setting(nickname, "coins",
                         self._coin_str(coins+interest))
 
-    @utils.hook("received.command.lotterybuy", authenticated=True)
+    @utils.hook("received.command.lotterybuy")
+    @utils.kwarg("help", "Buy ticket(s) for the lottery")
+    @utils.kwarg("authenticated", True)
+    @utils.spec("?<count>int")
     def lottery_buy(self, event):
-        """
-        :help: Buy ticket(s) for the lottery
-        :usage: [amount]
-        """
-        amount = "1"
-        if event["args_split"]:
-            amount = event["args_split"][0]
-        if not amount.isdigit():
-            raise utils.EventError("%s: Please provide a positive number "
-                "of tickets to buy" % event["user"].nickname)
-        amount = int(amount)
+        amount = event["spec"][0] or 1
 
         user_coins = self._get_user_coins(event["user"])
         coin_amount = decimal.Decimal(LOTTERY_BUYIN)*amount
@@ -494,39 +473,31 @@ class Module(ModuleManager.BaseModule):
             self._coin_str(coin_amount)))
 
     @utils.hook("received.command.mylottery")
+    @utils.kwarg("help", "Show how many lottery tickets you currently have")
     def my_lottery(self, event):
-        """
-        :help: Show how many lottery tickets you currently have
-        """
         lottery = event["server"].get_setting("lottery", {})
         count = lottery.get(event["user"].nickname_lower, 0)
         event["stdout"].write("%s: You have %d lottery ticket%s" % (
             event["user"].nickname, count, "" if count == 1 else "s"))
 
     @utils.hook("received.command.jackpot")
+    @utils.kwarg("help", "Show the current lottery jackpot")
     def jackpot(self, event):
-        """
-        :help: Show the current lottery jackpot
-        """
         lottery = event["server"].get_setting("lottery", {})
         count = sum(value for nickname, value in lottery.items())
         event["stdout"].write("%s: The current jackpot is %s" % (
             event["user"].nickname, decimal.Decimal(LOTTERY_BUYIN)*count))
 
     @utils.hook("received.command.nextlottery")
+    @utils.kwarg("help", "Show time until the next lottery draw")
     def next_lottery(self, event):
-        """
-        :help: Show time until the next lottery draw
-        """
         until = self._until_next_6_hour()
         event["stdout"].write("Next lottery is in: %s" %
             utils.datetime.format.to_pretty_time(until))
 
     @utils.hook("received.command.lotterywinner")
+    @utils.kwarg("help", "Show who last won the lottery")
     def lottery_winner(self, event):
-        """
-        :help: Show who last won the lottery
-        """
         winner = event["server"].get_setting("lottery-winner", None)
         if winner:
             event["stdout"].write("Last lottery winner: %s" % winner)