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

This commit is contained in:
jesopo 2020-02-14 22:47:01 +00:00
parent 34c7e3d9d9
commit 6e9c303fcd

View file

@ -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["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):
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)
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:
if "0" in bets:
raise utils.EventError("%s: You can't bet on 0" %
event["user"].nickname)
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)