Added coins.py

1 files changed, 68 insertions, 0 deletions

diff --git a/modules/coins.py b/modules/coins.py
new file mode 100644
index 00000000..ca98448c
--- /dev/null
+++ b/modules/coins.py
@@ -0,0 +1,68 @@
+import random, time
+import Utils
+
+SIDES = {"heads": 0, "tails": 1}
+DEFAULT_REDEEM_DELAY = 600 # 600 seconds, 10 minutes
+
+class Module(object):
+    def __init__(self, bot):
+        bot.events.on("received.command.coins").hook(self.coins,
+            help="Show how many coins you have")
+        bot.events.on("received.command.redeemcoins").hook(
+            self.redeem_coins, help="Redeem free coins")
+        bot.events.on("received.command.flip").hook(self.flip,
+            help="Bet coins on a coin flip", usage=
+            "heads|tails <coin amount>", min_args=2)
+
+    def coins(self, event):
+        coins = event["user"].get_setting("coins", 0)
+        event["stdout"].write("%s has %d coin%s" % (
+            event["user"].nickname, coins,
+            "" if coins == 1 else "s"))
+
+    def redeem_coins(self, event):
+        last_redeem = event["user"].get_setting("last-redeem", None)
+        redeem_delay = event["server"].get_setting("redeem-delay",
+            DEFAULT_REDEEM_DELAY)
+
+        if last_redeem == None or (time.time()-last_redeem) >= redeem_delay:
+            user_coins = event["user"].get_setting("coins", 0)
+            redeem_amount = event["server"].get_setting("redeem-amount", 100)
+            event["user"].set_setting("coins", user_coins+redeem_amount)
+            event["stdout"].write("Redeemed %d coins" % redeem_amount)
+            event["user"].set_setting("last-redeem", time.time())
+        else:
+            time_left = (last_redeem+redeem_delay)-time.time()
+            event["stdout"].write("Please wait %s before redeeming" %
+                Utils.to_pretty_time(time_left))
+
+    def flip(self, event):
+        side_name = event["args_split"][0].lower()
+        coin_bet = event["args_split"][1]
+
+        if not coin_bet.isdigit():
+            event["stderr"].write("Please provide a number of coins to bet")
+            return
+        coin_bet = int(coin_bet)
+        if not side_name in SIDES:
+            event["stderr"].write("Please provide 'heads' or 'tails'")
+            return
+
+        user_coins = event["user"].get_setting("coins", 0)
+        if coin_bet > user_coins:
+            event["stderr"].write("You don't have enough coins to bet")
+            return
+
+        chosen_side = random.choice(list(SIDES.keys()))
+        win = side_name == chosen_side
+        print(chosen_side)
+        print(side_name)
+
+        if win:
+            event["user"].set_setting("coins", user_coins+coin_bet)
+            event["stdout"].write("%s flipped %s and wins %d!" % (
+                event["user"].nickname, side_name, coin_bet))
+        else:
+            event["user"].set_setting("coins", user_coins-coin_bet)
+            event["stdout"].write("%s flipped %s and loses %d!" % (
+                event["user"].nickname, side_name, coin_bet))