52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
from nonebot import on_command
|
|
from nonebot.rule import Rule
|
|
from nonebot.adapters.onebot.v11.adapter import Message, MessageSegment
|
|
from nonebot.adapters.onebot.v11 import Message, MessageSegment, Bot, MessageEvent
|
|
from nonebot.matcher import Matcher
|
|
from nonebot.params import CommandArg
|
|
|
|
from pathlib import Path
|
|
import random
|
|
import os
|
|
|
|
fabing = on_command("fabing", rule=Rule(), aliases={'发病', '发病文学'}, priority=5)
|
|
PACKAGE_PATH = os.path.dirname(os.path.abspath(__file__))
|
|
TEMPLATE_PATH = str(Path(PACKAGE_PATH) / "templates.txt")
|
|
fabing_tmpl = []
|
|
random_index = 0
|
|
|
|
|
|
def load_fabing_tmpl():
|
|
global fabing_tmpl
|
|
global random_index
|
|
|
|
fabing_tmpl = []
|
|
with open(TEMPLATE_PATH, encoding='utf-8') as fp:
|
|
contents = fp.readlines()
|
|
for line in contents:
|
|
fabing_tmpl.append(line.strip().replace('\\n', '\n\n'))
|
|
|
|
random.shuffle(fabing_tmpl)
|
|
random_index = 0
|
|
|
|
load_fabing_tmpl()
|
|
|
|
@fabing.handle()
|
|
async def handle_fabing(matcher: Matcher, event: MessageEvent, arg: Message = CommandArg()):
|
|
global random_index
|
|
|
|
if len(arg) == 0:
|
|
await matcher.finish('缺少必要的参数：必须指定一个对象')
|
|
|
|
args = arg.extract_plain_text().rsplit()
|
|
if args[0] == '重载' or args[0] == 'reload':
|
|
load_fabing_tmpl()
|
|
await matcher.finish('重载成功')
|
|
|
|
target = args[0]
|
|
if random_index >= len(fabing_tmpl):
|
|
random_index = 0
|
|
tmpl = fabing_tmpl[random_index]
|
|
random_index += 1
|
|
await matcher.finish(tmpl.format(target))
|