mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-09-21 05:12:34 +00:00
83 lines
2.5 KiB
Python
83 lines
2.5 KiB
Python
import re
|
|
import random
|
|
import string
|
|
|
|
from command import CommandRegistry, split_arguments
|
|
from commands import core
|
|
|
|
__registry__ = cr = CommandRegistry()
|
|
|
|
|
|
@cr.register('随机数')
|
|
@cr.register('number', hidden=True)
|
|
@split_arguments()
|
|
def number(_, ctx_msg, internal=False, argv=None):
|
|
if len(argv) == 0 or not re.match('x\d+', argv[-1]):
|
|
n = 1
|
|
else:
|
|
n = max(1, int(argv[-1][1:]))
|
|
argv = argv[:-1]
|
|
|
|
if len(argv) > 2 or any((not re.match('-?\d+', num) for num in argv)):
|
|
core.echo('参数有错误哦～\n正确的使用方法（主要看参数，你的命令是对的～）：\n\n'
|
|
'/random.number\n'
|
|
'/random.number x5\n'
|
|
'/random.number 100\n'
|
|
'/random.number 100 x10\n'
|
|
'/random.number 50 100\n'
|
|
'/random.number 50 100 x3',
|
|
ctx_msg, internal)
|
|
return
|
|
|
|
if len(argv) == 1:
|
|
argv.append(1)
|
|
|
|
start, end = (int(argv[0]), int(argv[1])) if len(argv) == 2 else (None, None)
|
|
start, end = (min(start, end), max(start, end)) if start is not None else (start, end)
|
|
|
|
result = []
|
|
|
|
for _ in range(n):
|
|
result.append(random.randint(start, end) if start is not None else random.random())
|
|
|
|
core.echo(', '.join(str(num) for num in result), ctx_msg, internal)
|
|
return result
|
|
|
|
|
|
@cr.register('随机字符')
|
|
@cr.register('char', hidden=True)
|
|
@split_arguments()
|
|
def char(_, ctx_msg, internal=False, argv=None):
|
|
if len(argv) > 2 or (len(argv) == 2 and not re.match('x\d+', argv[-1])):
|
|
core.echo('参数有错误哦～\n正确的使用方法（主要看参数，你的命令是对的～）：\n\n'
|
|
'/random.char\n'
|
|
'/random.char x5\n'
|
|
'/random.char ABCDEFG\n'
|
|
'/random.char ABCDEFG x10\n',
|
|
ctx_msg, internal)
|
|
return
|
|
|
|
chars = string.ascii_letters + string.digits
|
|
size = 1
|
|
if len(argv) and re.match('x\d+', argv[-1]):
|
|
size = max(1, int(argv[-1][1:]))
|
|
argv = argv[:-1]
|
|
if len(argv):
|
|
chars = argv[0]
|
|
|
|
result = ''.join(random.choice(chars) for _ in range(size))
|
|
core.echo(result, ctx_msg, internal)
|
|
return result
|
|
|
|
|
|
@cr.register('随机化')
|
|
@cr.register('shuffle', hidden=True)
|
|
@split_arguments()
|
|
def char(_, ctx_msg, internal=False, argv=None):
|
|
if len(argv) == 0:
|
|
core.echo('请传入正确的参数哦～', ctx_msg, internal)
|
|
return argv
|
|
random.shuffle(argv)
|
|
core.echo(', '.join(argv), ctx_msg, internal)
|
|
return argv
|