65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
import secrets
|
|
import string
|
|
import uuid
|
|
|
|
from core.builtins import Bot
|
|
from core.component import module
|
|
|
|
r = module('random', alias=['rand', 'rng'],
|
|
developers=['Dianliang233', 'DoroWolf'], desc='{random.help.desc}', )
|
|
|
|
|
|
@r.handle('number <min> <max> {{random.help.number}}', )
|
|
async def _(msg: Bot.MessageSession, min: int, max: int):
|
|
if min > max:
|
|
return await msg.finish(msg.locale.t('error.range.invalid'))
|
|
|
|
random = secrets.randbelow(max - min + 1) + min
|
|
await msg.finish('' + str(random))
|
|
|
|
|
|
@r.handle('choice ... {{random.help.choice}}', )
|
|
async def _(msg: Bot.MessageSession):
|
|
choices = msg.parsed_msg['...']
|
|
await msg.finish(secrets.choice(choices))
|
|
|
|
|
|
@r.handle('shuffle ... {{random.help.shuffle}}', )
|
|
async def _(msg: Bot.MessageSession):
|
|
cards: list = msg.parsed_msg['...']
|
|
x = cards.copy()
|
|
for i in reversed(range(1, len(x))):
|
|
# pick an element in x[:i+1] with which to exchange x[i]
|
|
j = secrets.randbelow(i + 1)
|
|
x[i], x[j] = x[j], x[i]
|
|
await msg.finish(', '.join(x))
|
|
|
|
|
|
@r.handle('string <count> [-u] [-l] [-n] [-s] {{random.help.string}}',
|
|
options_desc={'-u': '{random.help.option.string.u}',
|
|
'-l': '{random.help.option.string.l}',
|
|
'-n': '{random.help.option.string.n}',
|
|
'-s': '{random.help.option.string.s}'})
|
|
async def _(msg: Bot.MessageSession, count: int):
|
|
if count < 1 or count > 100:
|
|
return await msg.finish(msg.locale.t('random.message.string.error.invalid'))
|
|
characters = ""
|
|
if msg.parsed_msg.get('-u', False):
|
|
characters += string.ascii_uppercase
|
|
if msg.parsed_msg.get('-l', False):
|
|
characters += string.ascii_lowercase
|
|
if msg.parsed_msg.get('-n', False):
|
|
characters += string.digits
|
|
if msg.parsed_msg.get('-s', False):
|
|
characters += "!@#$%^&*-_+=?"
|
|
|
|
if not characters:
|
|
characters = string.ascii_letters + string.digits
|
|
|
|
random = ''.join(secrets.choice(characters) for _ in range(count))
|
|
await msg.finish(random)
|
|
|
|
|
|
@r.handle('uuid {{random.help.uuid}}', )
|
|
async def _(msg: Bot.MessageSession):
|
|
await msg.finish(str(uuid.uuid4()))
|