49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
|
# coding=utf-8
|
||
|
|
"""
|
||
|
|
rand.py - Rand Module
|
||
|
|
Copyright 2013, Ari Koivula, <ari@koivu.la>
|
||
|
|
Licensed under the Eiffel Forum License 2.
|
||
|
|
|
||
|
|
http://sopel.chat
|
||
|
|
"""
|
||
|
|
|
||
|
|
from module import commands, example
|
||
|
|
import random
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
@commands('rand')
|
||
|
|
@example('.rand 2', r'random\(0, 2\) = (0|1|2)', re=True, repeat=10)
|
||
|
|
@example('.rand -1 -1', 'random(-1, -1) = -1')
|
||
|
|
@example('.rand', r'random\(0, \d+\) = \d+', re=True)
|
||
|
|
@example('.rand 99 10', r'random\(10, 99\) = \d\d', re=True, repeat=10)
|
||
|
|
@example('.rand 10 99', r'random\(10, 99\) = \d\d', re=True, repeat=10)
|
||
|
|
def rand(bot, trigger):
|
||
|
|
"""Replies with a random number between first and second argument."""
|
||
|
|
arg1 = trigger.group(3)
|
||
|
|
arg2 = trigger.group(4)
|
||
|
|
|
||
|
|
try:
|
||
|
|
if arg2 is not None:
|
||
|
|
low = int(arg1)
|
||
|
|
high = int(arg2)
|
||
|
|
elif arg1 is not None:
|
||
|
|
low = 0
|
||
|
|
high = int(arg1)
|
||
|
|
else:
|
||
|
|
low = 0
|
||
|
|
high = sys.maxsize
|
||
|
|
except (ValueError, TypeError):
|
||
|
|
return bot.reply("Arguments must be of integer type")
|
||
|
|
|
||
|
|
if low > high:
|
||
|
|
low, high = high, low
|
||
|
|
|
||
|
|
number = random.randint(low, high)
|
||
|
|
bot.reply("random(%d, %d) = %d" % (low, high, number))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
from test_tools import run_example_tests
|
||
|
|
run_example_tests(__file__)
|