1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
import math, random, time
import Utils
SIDES = {"heads": 0, "tails": 1}
DEFAULT_REDEEM_DELAY = 600 # 600 seconds, 10 minutes
DEFAULT_REDEEM_AMOUNT = 100
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.richest").hook(
self.richest, help="Show the top 10 richest users")
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)
bot.events.on("received.command.sendcoins").hook(
self.send, min_args=2, help="Send coins to a user",
usage="<nickname> <amount>")
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 richest(self, event):
all_coins = event["server"].get_all_user_settings("coins", [])
items = [(coin[0], coin[1]) for coin in all_coins]
all_coins = dict(items)
top_10 = sorted(all_coins.keys())
top_10 = sorted(top_10, key=all_coins.get, reverse=True)[:10]
top_10 = ", ".join("%s (%d)" % (event["server"].get_user(
nickname).nickname, all_coins[nickname]
) for nickname in top_10)
event["stdout"].write("Richest users: %s" % top_10)
def redeem_coins(self, event):
user_coins = event["user"].get_setting("coins", 0)
if user_coins == 0:
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", DEFAULT_REDEEM_AMOUNT)
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(math.ceil(time_left)))
else:
event["stderr"].write(
"You can only redeem coins when you have none")
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
if win:
event["user"].set_setting("coins", user_coins+coin_bet)
event["stdout"].write("%s flips %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 flips %s and loses %d!" % (
event["user"].nickname, side_name, coin_bet))
def send(self, event):
send_amount = event["args_split"][1]
if not send_amount.isdigit() or int(send_amount) <= 0:
event["stderr"].write(
"Please provide a positive number of coins to send")
return
send_amount = int(send_amount)
user_coins = event["user"].get_setting("coins")
redeem_amount = event["server"].get_setting(
"redeem-amount", DEFAULT_REDEEM_AMOUNT)
new_user_coins = user_coins - send_amount
if new_user_coins == 0:
event["stderr"].write("You have no coins")
return
elif new_user_coins < redeem_amount:
event["stderr"].write(
"You cannot send an amount of money that puts"
" you below %d coins" % redeem_amount)
return
target_user = event["server"].get_user(event["args_split"][0])
target_user_coins = target_user.get_setting("coins", 0)
event["user"].set_setting("coins", new_user_coins)
target_user.set_setting("coins", target_user_coins+send_amount)
event["stdout"].write("%s sent %d coins to %s" % (
event["user"].nickname, send_amount, target_user.nickname))
|
